在這里使用python演示
根據(jù)上面的業(yè)務(wù)場景描述,最容易想到的解決方法就是
from pymongo import MongoClient# 連接數(shù)據(jù)庫db = MongoClient('mongodb://127.0.0.1:27017')['my_db']# 簡化的查詢數(shù)據(jù)集A的條件filter = {...}# 查詢Collection Aa_cursor = db.a.find(_filter)a_docs = [x for x in a_cursor]# 變量的初始定義count = 0total = 0# 加入需要用到的元素為第21個index = 20# 查詢Collection B,同時做累加for a_doc in a _docs: b_doc = db.b.find_one({'uid':a_doc['uid'], 'date': a_doc['date']}) # 只有能查到相應(yīng)的結(jié)果時,才可以 if b_doc is not None: total += b_doc['actions'][20]['number'] count += 1 # 求平均數(shù) if count > 0 : avg = total/count實(shí)現(xiàn)難度當(dāng)然是最低的,可是整個任務(wù)在第一步只有1萬條左右的返回時,消耗的時間竟然達(dá)到了驚人38秒。當(dāng)然這是已經(jīng)加了索引的結(jié)果,否則可能都無法得到結(jié)果了。
瓶頸顯而易見,在循環(huán)中查詢Collection B,增加了網(wǎng)絡(luò)開銷,自然也就增加時間,如果一次查詢出所有結(jié)果,自然會大大提高效率。也就是說,我要把第一步的結(jié)果作為條件一次性傳遞,做一個$in操作??墒窃趺床拍茏龅侥兀咳绻趗id和date上分別做$in操作,那么返回的結(jié)果就會是二者單獨(dú)做$操作的合集,很顯然這和要求是不符的。 經(jīng)過上面的分析,似乎進(jìn)入了死胡同。其實(shí)答案也基本顯現(xiàn)了,需要有一個字段可以滿足上面的要求,那么這個字段就是uid和date的合體,就命名為uid_date。uid_date是一個新字段,在B中并不存在,在使用之前需要將數(shù)據(jù)庫現(xiàn)有的數(shù)據(jù)做一下處理。處理完畢改造程序:
# 下面的只體現(xiàn)和本次修改相關(guān)的內(nèi)容uid_date_list = []for a_doc in a_docs: uid_date_list.append(a_doc['uid'] + '_' + a_doc['date'])# 查詢Bb_cursor = db.b.find({'uid_date':{'$in':uid_date_list}})# 下面就是取出結(jié)果,求平均數(shù)...這一番改造頗費(fèi)時間,主要是前期的數(shù)據(jù)處理。代碼改造完畢,執(zhí)行下看看吧。 可是,可是…… 45秒 我做錯了什么?!
我還是堅(jiān)信上面的優(yōu)化思路是對的,現(xiàn)在看看數(shù)據(jù)庫能給一些什么線索吧。 登錄到數(shù)據(jù)庫服務(wù)器,找到MongoDB的日志/data/mongodb/logs/mongod.log。仔細(xì)查找,發(fā)現(xiàn)在查詢數(shù)據(jù)集B時有很多getMore命令。這就奇怪了,我是一次性查詢,為什么還有g(shù)etMore。趕緊查下官方的文檔,然后發(fā)現(xiàn)了下面的內(nèi)容: batcSize參數(shù)指定了每次返回的個數(shù),默認(rèn)的101個。那看來這個應(yīng)該是問題所在。找下pymongo的文檔,也可以設(shè)置這個參數(shù),那就設(shè)個大的吧10000。再次改造程序如下:
這次總該可以了。
嗯,好了一些,降到了20秒左右??墒?,這離1秒只能還差距20倍呢。
當(dāng)日不能放棄,繼續(xù)通過日志查找線索,發(fā)現(xiàn)還是有很多getMore。通過各方查找,發(fā)現(xiàn)mongodb每次最多返回16M的記錄,通過getMore日志的比對,發(fā)現(xiàn)的確如此。由于B中每條記錄的過去龐大,每次只能幾百條記錄,因此要一次多返回,那就必須要減少每次返回的記錄數(shù)。因?yàn)樵谟?jì)算時,只用了特定索引位置上的數(shù)據(jù),所以只返回該條記錄就可以了。
最后的代碼就不再寫了,具體可以參考官方文檔的實(shí)例。
新聞熱點(diǎn)
疑難解答