Added Insert API for docdb

CHESSComputing · Dec 6, 2024 · ff2aef4 · ff2aef4
1 parent 10101e2
commit ff2aef4
Show file tree

Hide file tree

Showing 4 changed files with 29 additions and 1 deletion.
diff --git a/docdb/apis.go b/docdb/apis.go
@@ -1,6 +1,8 @@
 package docdb
 
 import (
+	"log"
+
 	srvConfig "github.com/CHESSComputing/golib/config"
 	embed "github.com/CHESSComputing/golib/embed/badger"
 	"github.com/CHESSComputing/golib/mongo"
@@ -12,12 +14,23 @@ func InitDocDB(uri string) {
 		srvConfig.Init()
 	}
 	if srvConfig.Config.Embed.DocDb != "" {
-		embed.InitDB(uri)
+		log.Printf("Use embed db %s", srvConfig.Config.Embed.DocDb)
+		embed.InitDB(srvConfig.Config.Embed.DocDb)
 		return
 	}
+	log.Printf("Use mongo db %s", uri)
 	mongo.InitMongoDB(uri)
 }
 
+// Insert record into document-oriented db
+func Insert(dbname, collname string, records []map[string]any) {
+	if srvConfig.Config.Embed.DocDb != "" {
+		embed.Insert(dbname, collname, records)
+		return
+	}
+	mongo.Insert(dbname, collname, records)
+}
+
 // Upsert records into document-oriented db
 func Upsert(dbname, collname, attr string, records []map[string]any) error {
 	if srvConfig.Config.Embed.DocDb != "" {

diff --git a/embed/badger/badger.go b/embed/badger/badger.go
@@ -36,6 +36,11 @@ func ensureDirExists(dir string) error {
 	return nil
 }
 
+// Insert records into BadgerDB
+func Insert(dbname, collname string, records []map[string]any) {
+	upsert(collname, records)
+}
+
 // Upsert records into BadgerDB
 func Upsert(dbname, collname, attr string, records []map[string]any) error {
 	return upsert(collname, records)

diff --git a/embed/clover/embed.go b/embed/clover/embed.go
@@ -33,6 +33,11 @@ func InitDB(dbDir string) {
 	log.Printf("Clover database initialized at %s", dbDir)
 }
 
+// Insert records into document-oriented db
+func Insert(dbname, collname string, records []map[string]any) {
+	Upsert(dbname, collname, "", records)
+}
+
 // Upsert records into document-oriented db
 func Upsert(dbname, collname, attr string, records []map[string]any) error {
 	if err := db.CreateCollection(collname); err != nil && err != clover.ErrCollectionExist {

diff --git a/embed/tiedot/main.go b/embed/tiedot/main.go
@@ -18,6 +18,11 @@ func InitDB(uri string) {
 	}
 }
 
+// Insert records into document-oriented db
+func Insert(dbname, collname string, records []map[string]any) {
+	Upsert(dbname, collname, "", records)
+}
+
 // Upsert records into document-oriented db
 func Upsert(dbname, collname, attr string, records []map[string]any) error {
 	var err error