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)