Python+MongoDB自增鍵值的簡單實現

背景

最近在寫一個測試工具箱，裡面有一個bug記錄系統，因為後台我是用Django和MongoDB來實現的，就遇到了一個問題，要如何實作一個自增的欄位。

傳統的關係型資料庫要實現起來是非常容易，只要直接設定一個自增字段就行了，插入資料時不用管這個鍵值，只管自己處理的資料就行了，會自動實現自增的功能，但是非關係型資料庫好像沒有這個功能（或者我不知道）。百度之後發現都是MongoDB的設定方法，不是我想要的。

解決思路

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

我發現在MongoDB修改的方法有一個$inc的方法.可以實現int類型的自增。那就非常簡單了，自己建一個collection，然後這個collection中只有一個int的字段，每次插入資料時來這個collection取ID就行了，然後再調用一下$inc方法，那麼就實現了自動自增的方案。

程式碼展示

Python實作還是非常簡單的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

# 每次插入成功後，呼叫這個方法就可以實現ID的自增。

def getBugID(self):
  """
  获取当前bug的最新编号
  :return:None
  """
  db = self.__chooseCollection(config.COLLECTION['bugID'])
  rst = db.find_one()
  return rst['bugID']

在插入之前，呼叫這個方法，這樣每次插入資料時插入的ID就是實作自增的ID。

缺點

當然，這種方式還是有缺點的，呼叫的時候多使用了方法，因此需要確認方法是否執行成功，否則會導致下次插入的ID不是自增後的資料。每次多調了一次方法，會導致效能下降。

其他

如果有更好的實作方式，請告訴我！

以上這篇Python+MongoDB自增鍵值的簡單實作就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持PHP中文網。

更多Python+MongoDB自增鍵值的簡單實作相關文章請關注PHP中文網！