瀏覽代碼

新增创建表并生成索引CreateIndex

maxiaoshan 3 年之前
父節點
當前提交
aac84733d9
共有 1 個文件被更改,包括 15 次插入0 次删除
  1. 15 0
      src/mongodb/mongodb.go

+ 15 - 0
src/mongodb/mongodb.go

@@ -377,6 +377,21 @@ func (m *MongodbSim) Close() {
 	<-m.pool
 }
 
+//新建表并生成索引
+func (m *MongodbSim) CreateIndex(c string, models []mongo.IndexModel) bool {
+	defer catch()
+	m.Open()
+	defer m.Close()
+	coll := m.C.Database(m.DbName).Collection(c)
+	names, err := coll.Indexes().CreateMany(m.Ctx, models)
+	if err == nil && len(names) > 0 {
+		return true
+	} else {
+		log.Println("CreateIndex Error:", err)
+		return false
+	}
+}
+
 func (m *MongodbSim) Save(c string, doc interface{}) string {
 	defer catch()
 	m.Open()