如何在循环中使用批量更新插入？

Question

我在 Mongoldb 中的字段是； id、website_url、状态。

我需要找到 website_url 并将其状态更新为 3 并添加一个名为 err_desc 的新字段。

我有一个 website_urls 列表，它的状态和它的 err_desc。

下面是我的代码。

client = MongoClient('localhost', 9000)
db1 = client['Company_Website_Crawl']
collection1 = db1['All']
posts1 = collection1.posts
bulk = posts1.initialize_ordered_bulk_op()
website_url = ["http://www.example.com","http://example2.com/"]
err_desc = ["error1","error2"]
for i in website_url:
    parsed_uri = urlparse(i)
    domain = '{uri.scheme}://{uri.netloc}/'.format(uri=parsed_uri)
    final_url = domain
    final_url_strip = domain.rstrip("/")
    print i,final_url,final_url_strip,"\n"
    try:
        k = bulk.find({'website_url':i}).upsert().update({'$push':{'err_desc':err_desc,'status':3}})
        k = bulk.execute()
        print k
    except Exception as e:
        print "fail"
        print e

错误

发生失败批处理操作错误

fail 批量操作只能执行一次。

一开始我用的是

k =  posts1.update({'website_url':final_url_strip},{'$set':{'err_desc':err_desc,'status':3}},multi=True)

对于 500 万条记录来说太慢了。所以我想使用批量更新选项。请帮助我在这种情况下使用批量更新插入。

Answer 1

错误信息告诉你调用execute()后需要重新初始化批量写入操作。但问题是，你做错了。在您的情况下，您需要在 for 循环的末尾调用 execute，如下所示：

from itertools import count


ct = count()

for url in website_url:
    ...
    try:
        bulk.find({'website_url':i}).upsert().update({'$push':{'err_desc':err_desc,'status':3}})
        val = next(ct)
    except Exception as e:
        ...
if val > 0:
    bulk.execute()     

另请注意，Bulk() 现在已弃用并替换为 bulkwrite