入esUtil

public class Test {

    private static void test1(TransportClient client, String index, String type) throws UnknownHostException {
        Long count = 100000L;
        long start = System.currentTimeMillis();
        BulkRequestBuilder bulkRequest = client.prepareBulk();
        for (int i = 0; i < count; i++) {
            Map<String, Object> ret = new HashMap<String, Object>();
            ret.put("recordtime", 11);
            ret.put("area", 22);
            ret.put("usertype", 33);
            ret.put("count", 44);
            bulkRequest.add(client.prepareIndex(index, type).setSource(ret));
            // 每10000條提交一次
            if (i % 1000 == 0) {
                bulkRequest.execute().actionGet();
                bulkRequest = client.prepareBulk();
                System.out.println("提交+1=======" + i);
            }
        }
        bulkRequest.execute().actionGet();
        long end = System.currentTimeMillis();
        System.out.println((end - start) / 1000.000);
    }
    @Value("${spring.importModel.threatNum:1}")
    private Integer threatNum;

    public static void main(String[] args) {
        // 配置信息
        Settings settings = Settings.builder()
                //集羣名字
                .put("cluster.name", "idss-logaudit")
                //增加嗅探機制,找到ES集羣
                .put("client.transport.sniff", true)
                //增加線程池個數,暫時設爲5
                .put("thread_pool.search.size", Integer.parseInt("5"))
                .build();
        try {
            TransportClient client = new PreBuiltTransportClient(settings)
                    .addTransportAddress(new InetSocketTransportAddress(InetAddress.getByName("192.168.0.238"), 9300));

            test1(client, "tyx_test_index_2018.9.24", "log");
        } catch (UnknownHostException e) {
            e.printStackTrace();
        }
    }
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章