ElasticSearch工具类

  1. import org.elasticsearch.action.admin.indices.create.CreateIndexResponse;
  2. import org.elasticsearch.action.admin.indices.exists.indices.IndicesExistsResponse;
  3. import org.elasticsearch.action.delete.DeleteResponse;
  4. import org.elasticsearch.action.admin.indices.delete.DeleteIndexRequest;
  5. import org.elasticsearch.action.support.master.AcknowledgedResponse;
  6. import org.elasticsearch.client.IndicesAdminClient;
  7. import org.elasticsearch.client.transport.TransportClient;
  8. import org.elasticsearch.common.settings.Settings;
  9. import org.elasticsearch.common.transport.TransportAddress;
  10. import org.elasticsearch.common.xcontent.XContentType;
  11. import org.elasticsearch.transport.client.PreBuiltTransportClient;
  13. import java.net.InetAddress;
  14. import java.net.UnknownHostException;
  16. public class ESUtil {
  17.     //集群名,默认值elasticsearch
  18.     private static final String CLUSTER_NAME = "elasticsearch";
  19.     //ES集群中某个节点
  20.     private static final String HOSTNAME = "node1";
  21.     //连接端口号
  22.     private static final int TCP_PORT = 9300;
  23.     //构建Settings对象
  24.     private static Settings settings = Settings.builder().put("cluster.name", CLUSTER_NAME).build();
  25.     //TransportClient对象，用于连接ES集群
  26.     private static volatile TransportClient client;
  28.     /**
  29.      * 同步synchronized(*.class)代码块的作用和synchronized static方法作用一样,
  30.      * 对当前对应的*.class进行持锁,static方法和.class一样都是锁的该类本身,同一个监听器
  31.      * @return
  32.      * @throws UnknownHostException
  33.      */
  34.     public static TransportClient getClient(){
  35.         if(client==null){
  36.             synchronized (TransportClient.class){
  37.                 try {
  38.                     client=new PreBuiltTransportClient(settings)
  39.                             .addTransportAddress(new TransportAddress(InetAddress.getByName(HOSTNAME), TCP_PORT));
  40.                 } catch (UnknownHostException e) {
  41.                     e.printStackTrace();
  42.                 }
  43.             }
  44.         }
  45.         return client;
  46.     }
  48.     /**
  49.      * 获取索引管理的IndicesAdminClient
  50.      */
  51.     public static IndicesAdminClient getAdminClient() {
  53.         return getClient().admin().indices();
  54.     }
  56.     /**
  57.      * 判定索引是否存在
  58.      * @param indexName
  59.      * @return
  60.      */
  61.     public static boolean isExists(String indexName){
  62.         IndicesExistsResponse response=getAdminClient().prepareExists(indexName).get();
  63.         return response.isExists()?true:false;
  64.     }
  65.     /**
  66.      * 创建索引
  67.      * @param indexName
  68.      * @return
  69.      */
  70.     public static boolean createIndex(String indexName){
  71.         CreateIndexResponse createIndexResponse = getAdminClient()
  72.                 .prepareCreate(indexName.toLowerCase())
  73.                 .get();
  74.         return createIndexResponse.isAcknowledged()?true:false;
  75.     }
  77.     /**
  78.      * 创建索引
  79.      * @param indexName 索引名
  80.      * @param shards   分片数
  81.      * @param replicas  副本数
  82.      * @return
  83.      */
  84.     public static boolean createIndex(String indexName, int shards, int replicas) {
  85.         Settings settings = Settings.builder()
  86.                 .put("index.number_of_shards", shards)
  87.                 .put("index.number_of_replicas", replicas)
  88.                 .build();
  89.         CreateIndexResponse createIndexResponse = getAdminClient()
  90.                 .prepareCreate(indexName.toLowerCase())
  91.                 .setSettings(settings)
  92.                 .execute().actionGet();
  93.         return createIndexResponse.isAcknowledged()?true:false;
  94.     }
  96.     /**
  97.      * 位索引indexName设置mapping
  98.      * @param indexName
  99.      * @param typeName
  100.      * @param mapping
  101.      */
  102.     public static void setMapping(String indexName, String typeName, String mapping) {
  103.         getAdminClient().preparePutMapping(indexName)
  104.                 .setType(typeName)
  105.                 .setSource(mapping, XContentType.JSON)
  106.                 .get();
  107.     }
  109.     /**
  110.      * 删除索引
  111.      * @param indexName
  112.      * @return
  113.      */
  114.     public static boolean deleteIndex(String indexName) {
  115.         // DeleteResponse response = client.prepareDelete("twitter", "_doc", "1").get();
  116. //        DeleteResponse deleteResponse = getAdminClient()
  117. //                .prepareDelete(indexName.toLowerCase())
  118. //                .get();
  119. //            DeleteResponse response = getAdminClient().prepareDelete(indexName).get();
  120.         AcknowledgedResponse response = getAdminClient().prepareDelete(indexName).get();
  121. //        DeleteResponse response =  getClient().prepareDelete().get();
  123.         return response.isAcknowledged()?true:false;
  124.     }
  125. }