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();
}
}
}