#coding=utf-8
from elasticsearch import Elasticsearch
from elasticsearch import helpers

import time

es = Elasticsearch("http://localhost:9200")

index = "myindex"

#create index
es.indices.create(index=index, ignore=400)

actions = []
f = open('user.csv', 'r', encoding='utf-8')
data_list = f.readlines()


i = 0
start_time = time.time()
for data in data_list:
    data = data.strip().split(',')
    #print(data)

    action = {
        "_index":index,
        "_type":"usertest",
        "_id":None,
        "_source":{
            "id":data[0],
            "username":data[1],
            "password":data[2],
            "email":data[3]
        }
    }
    #print(action)
    actions.append(action)
    i += 1
    if len(action) == 1000:
        helpers.bulk(es, actions)
        del actions[0:len(action)]

if i > 0:
    helpers.bulk(es, actions)

end_time = time.time()
t = end_time - start_time
print('本次共寫入{}條資料，用時{}s'.format(i, t))

通過elasticsearch-head檢視結果如下：

相關推薦