go mongodb Iter迭代器樣本

來源:互聯網
上載者:User
func (this *Mongodb) StoreRecords(dbName, collection, storeDbName, tmpCollection string, queryMap, selecter bson.M) (err error) {session := this.GetSession()defer session.Close()iter := session.DB(dbName).C(collection).Find(queryMap).Select(selecter).Iter()oneMap := make(bson.M)historyMapSlice := make([]interface{}, 0, MAX_BULK_WRITE_FILE_LEN)flag := falsefor iter.Next(&oneMap) {tmpMap := DeepCopy(oneMap)//go深度拷貝historyMapSlice = append(historyMapSlice, tmpMap.(bson.M))if len(historyMapSlice) >= MAX_BULK_WRITE_FILE_LEN {//寫入最大長度this.BulkInsert(storeDbName, tmpCollection, historyMapSlice)historyMapSlice = make([]interface{}, 0, MAX_BULK_WRITE_FILE_LEN)flag = true}}if err = iter.Close(); err != nil {log.Error("StoreRecords,iter.Close() fail, db:", dbName,", Collection:", collection,", queryCondition:", queryMap,", selector:", selecter,", ErrInfo:", err)}if 0 == len(historyMapSlice) && !flag {err = errors.New("not found StoreRecords data")return}this.BulkInsert(storeDbName, tmpCollection, historyMapSlice)return}func (this *Mongodb) BulkInsert(dataBase, collection string, allRecords []interface{}) (err error) {    if 0 == len(allRecords) {        log.Info("bulkInsert: 0 == len(allRecords)! db:", dataBase, "collection:", collection)        return    }    bulkInsert := func(c *mgo.Collection) error {        bulk := c.Bulk()        bulk.Unordered()        bulk.Insert(allRecords...)        _, err := bulk.Run()        return err    }    err = this.WitchCollection(dataBase, collection, bulkInsert)    if err != nil {        log.Error("mongodb BulkInsert, BulkInsert Failed, db:", dataBase,            ", Collection:", collection,            ", ErrInfo:", err)    }    return}

相關文章

聯繫我們

該頁面正文內容均來源於網絡整理,並不代表阿里雲官方的觀點,該頁面所提到的產品和服務也與阿里云無關,如果該頁面內容對您造成了困擾,歡迎寫郵件給我們,收到郵件我們將在5個工作日內處理。

如果您發現本社區中有涉嫌抄襲的內容,歡迎發送郵件至: info-contact@alibabacloud.com 進行舉報並提供相關證據,工作人員會在 5 個工作天內聯絡您,一經查實,本站將立刻刪除涉嫌侵權內容。

A Free Trial That Lets You Build Big!

Start building with 50+ products and up to 12 months usage for Elastic Compute Service

  • Sales Support

    1 on 1 presale consultation

  • After-Sales Support

    24/7 Technical Support 6 Free Tickets per Quarter Faster Response

  • Alibaba Cloud offers highly flexible support services tailored to meet your exact needs.