快捷導(dǎo)航

Python+MongoDB自增鍵值的簡單實(shí)現(xiàn)

更新時(shí)間：2016年11月04日 10:49:27 投稿：jingxian

下面小編就為大家?guī)硪黄狿ython+MongoDB自增鍵值的簡單實(shí)現(xiàn)。小編覺得挺不錯(cuò)的，現(xiàn)在就分享給大家，也給大家做個(gè)參考。一起跟隨小編過來看看吧

背景

最近在寫一個(gè)測試工具箱，里面有一個(gè)bug記錄系統(tǒng)，因?yàn)楹笈_我是用Django和MongoDB來實(shí)現(xiàn)的，就遇到了一個(gè)問題，要如何實(shí)現(xiàn)一個(gè)自增的字段。

傳統(tǒng)的關(guān)系型數(shù)據(jù)庫要實(shí)現(xiàn)起來是非常容易，只要直接設(shè)置一個(gè)自增字段就行了，插入數(shù)據(jù)時(shí)不用管這個(gè)鍵值，只管自己處理的數(shù)據(jù)就行了，會自動(dòng)實(shí)現(xiàn)自增的功能，但是非關(guān)系型數(shù)據(jù)庫好像沒有這個(gè)功能（或者我不知道）。百度之后發(fā)現(xiàn)都是MongoDB的設(shè)置方法，并不是我想要的。

解決思路

百度沒有找到好的思路，那就只能自己解決了，我的想法很簡單，字段不會自增，那么就自己造一個(gè)自增的方案。

我發(fā)現(xiàn)在MongoDB修改的方法有一個(gè)$inc的方法.可以實(shí)現(xiàn)int類型的自增。那么就非常簡單了，自己建一個(gè)collection，然后這個(gè)collection中只有一個(gè)int的字段，每次插入數(shù)據(jù)時(shí)來這個(gè)collection取ID就行了，然后再調(diào)用一下$inc方法，那么就實(shí)現(xiàn)了自動(dòng)自增的方案。

代碼展示

Python實(shí)現(xiàn)起來還是非常簡單的Python和Django非常契合。代碼如下：

def bugPlus(self):
  """
  bugID自增
  :return:True
  """
  db = self.__chooseCollection(config.COLLECTION['bugID'])
  db.update_one({"bugID": self.getBugID()}, {"$inc": {"bugID": 1}})
  return True

每次插入成功后，調(diào)用這個(gè)方法就可以實(shí)現(xiàn)ID的自增。

def getBugID(self):
  """
  獲取當(dāng)前bug的最新編號
  :return:None
  """
  db = self.__chooseCollection(config.COLLECTION['bugID'])
  rst = db.find_one()
  return rst['bugID']

在插入之前，調(diào)用這個(gè)方法，這樣每次插入數(shù)據(jù)時(shí)插入的ID就是實(shí)現(xiàn)自增的ID。

缺點(diǎn)

當(dāng)然，這種方式還是有缺點(diǎn)的，調(diào)用的時(shí)候多使用了方法，因此需要確認(rèn)方法是否執(zhí)行成功，否則會導(dǎo)致下次插入的ID不是自增后的數(shù)據(jù)。每次多調(diào)了一次方法，會導(dǎo)致性能下降。

其他

如果有更好的實(shí)現(xiàn)方式，請告訴我！

以上這篇Python+MongoDB自增鍵值的簡單實(shí)現(xiàn)就是小編分享給大家的全部內(nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: