MongoDB使用python腳本寫入數據

批量寫：

from pymongo import MongoClient
from pymongo import InsertOne
import time

def insert():
    #連接數據庫
    conn = MongoClient("XXX.XXX.XXX.XXX:XXXX",maxPoolSize=None)
    my_db = conn['test']
    my_collection = my_db['test_1']

    # 批量寫
    i = 0
    t0 = time.time()
    data =[]
    while True:
        #'_id'為主鍵，循環時遞增，全部添加到data列表內
        data.append(InsertOne({"_id":i,"insert_time": int(time.time() * 1000)}))
        i+=1
        #判斷列表長度，達到10000執行插入，后繼續循環
        if len(data) == 10000:
            my_collection.bulk_write(data)
            res = []
            i += 1
            continue
      #判斷i等於1億時停止循環
      elif i == 100000000:
             break

if __name__ == '__main__':
    insert()

 

逐條寫：

def test_insert():
    conn = MongoClient("xxx.xxx.xxx.xxx:xxxx",maxPoolSize=None)
    my_db = conn['test']
    # 逐條寫
    t0 = time.time()
    for i in range(0,10000000):
        my_db['test_1'].insert_one({"_id":i, "insert_time": int(time.time() * 1000)})
    print (time.time() - t0)