* Copyright 2011 Peter Karich, peat_hal 'at' users 'dot' sourceforge 'dot' net.

*

* Licensed under the Apache License, Version 2.0 (the "License");

* you may not use this file except in compliance with the License.

* You may obtain a copy of the License at

*

*      http://www.apache.org/licenses/LICENSE-2.0

*

* Unless required by applicable law or agreed to in writing, software

* distributed under the License is distributed on an "AS IS" BASIS,

* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

* See the License for the specific language governing permissions and

* limitations under the License.

*/

*

* @author Peter Karich, peat_hal 'at' users 'dot' sourceforge 'dot' net

*/

    private Logger logger = LoggerFactory.getLogger(getClass());

    protected Client client;

    AbstractElasticSearch() {

    }

    public AbstractElasticSearch(Client client) {

        this.client = client;

    }

    public AbstractElasticSearch(String url, int port) {

        client = createClient(ElasticNode.CLUSTER, url, port);

    }

    public static Client createClient(String cluster, String url, int port) {

        Settings s = ImmutableSettings.settingsBuilder().put("cluster.name", cluster).build();

        TransportClient tmp = new TransportClient(s);

        tmp.addTransportAddress(new InetSocketTransportAddress(url, port));

        return tmp;

    }

   

    public abstract String getIndexName();

    public abstract String getIndexType();

    public void nodeInfo() {

        NodesInfoResponse rsp = client.admin().cluster().nodesInfo(new NodesInfoRequest()).actionGet();

        String str = "Cluster:" + rsp.getClusterName() + ". Active nodes:";

        str += rsp.getNodesMap().keySet();

        logger.info(str);

    }

    public boolean indexExists(String indexName) {

        // make sure node is up to create the index otherwise we get: blocked by: [1/not recovered from gateway];

        // waitForYellow();

//        Map map = client.admin().cluster().health(new ClusterHealthRequest(indexName)).actionGet().getIndices();

        Map map = client.admin().cluster().prepareState().execute().actionGet().getState().getMetaData().getIndices();

        return map.containsKey(indexName);

    }

    public void createIndex(String indexName) {

        // no need for the following because of _default mapping under config

        // String fileAsString = Helper.readInputStream(getClass().getResourceAsStream("tweet.json"));

        // new CreateIndexRequest(indexName).mapping(indexType, fileAsString)

        client.admin().indices().create(new CreateIndexRequest(indexName)).actionGet();

    }

    public void saveCreateIndex() {

        saveCreateIndex(getIndexName(), true);

    }

    public void saveCreateIndex(String name, boolean log) {

        try {           

            createIndex(name);

            if (log)

                logger.info("Created index: " + name);

        } catch (Exception ex) {

            if (log)

                logger.info("Index " + getIndexName() + " already exists");

        }

    }

//        waitForYellow();

//        client.admin().cluster().nodesInfo(new NodesInfoRequest()).actionGet();

//        System.out.println("health:"+client.admin().cluster().health(new ClusterHealthRequest(getIndexName())).actionGet().getStatus().name());

    // hmmh here we need indexName again ... but in createIndex it does not exist when calling ping ...

//        client.admin().cluster().ping(new SinglePingRequest(getIndexName(), getIndexType(), "1")).actionGet();

//    }

    void waitForYellow() {

        waitForYellow(getIndexName());

    }

    void waitForYellow(String name) {

        client.admin().cluster().health(new ClusterHealthRequest(name).waitForYellowStatus()).actionGet();

    }

    void waitForGreen(String name) {

        client.admin().cluster().health(new ClusterHealthRequest(name).waitForGreenStatus()).actionGet();

    }

    public void refresh() {

        refresh(getIndexName());

    }

    public void refresh(Collection<String> indices) {

        refresh(Helper.toStringArray(indices));

    }

    public void refresh(String... indices) {

        RefreshResponse rsp = client.admin().indices().refresh(new RefreshRequest(indices)).actionGet();

        //assertEquals(1, rsp.getFailedShards());

    }

    public long countAll() {

        CountResponse response = client.prepareCount(getIndexName()).

                setQuery(QueryBuilders.matchAllQuery()).

                execute().actionGet();

        return response.getCount();

    }

    public void feedDoc(String twitterId, XContentBuilder b) {

        IndexRequestBuilder irb = client.prepareIndex(getIndexName(), getIndexType(), twitterId).

                setConsistencyLevel(WriteConsistencyLevel.DEFAULT).

                setSource(b);

        irb.execute().actionGet();

    }

    public void deleteById(String id) {

        DeleteResponse response = client.prepareDelete(getIndexName(), getIndexType(), id).

                execute().

                actionGet();

    }

    public void deleteAll() {

        //client.prepareIndex().setOpType(OpType.)

        //there is an index delete operation

        // http://www.elasticsearch.com/docs/elasticsearch/rest_api/admin/indices/delete_index/

        client.prepareDeleteByQuery(getIndexName()).

                setQuery(QueryBuilders.matchAllQuery()).

                execute().

                actionGet();

        refresh();

    }

    public OptimizeResponse optimize(int optimizeToSegmentsAfterUpdate) {

        return client.admin().indices().optimize(new OptimizeRequest(getIndexName()).maxNumSegments(optimizeToSegmentsAfterUpdate)).actionGet();

    }

}