Unique Identity API update (#10)

Co-authored-by: badhri85 <[email protected]>
omec-project · Jul 26, 2022 · 2b5a8f2 · 2b5a8f2
1 parent 19942de
commit 2b5a8f2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/api_mongoDB.go b/api_mongoDB.go
@@ -87,23 +87,25 @@ func RestfulAPIGetMany(collName string, filter bson.M) []map[string]interface{}
 }
 
 /* Get unique identity from counter collection. */
-func GetUniqueIdentity() int32 {
+func GetUniqueIdentity(idName string) int32 {
 	counterCollection := Client.Database(dbName).Collection("counter")
 
 	counterFilter := bson.M{}
-	counterFilter["_id"] = "uniqueIdentity"
+	counterFilter["_id"] = idName
 
 	for {
 		count := counterCollection.FindOneAndUpdate(context.TODO(), counterFilter, bson.M{"$inc": bson.M{"count": 1}})
 
 		if count.Err() != nil {
+			logger.MongoDBLog.Println("FindOneAndUpdate error. Create entry for field  ")
 			counterData := bson.M{}
 			counterData["count"] = 1
-			counterData["_id"] = "uniqueIdentity"
+			counterData["_id"] = idName
 			counterCollection.InsertOne(context.TODO(), counterData)
 
 			continue
 		} else {
+			logger.MongoDBLog.Println("found entry. inc and return")
 			data := bson.M{}
 			count.Decode(&data)
 			decodedCount := data["count"].(int32)

diff --git a/dbtestapp/dbtestapp.go b/dbtestapp/dbtestapp.go
@@ -126,10 +126,10 @@ func main() {
 	//time.Sleep(120 * time.Second)
 	//updateDocumentWithTimeout("timeout", "yak1", 200, "createdAt")
 
-	uniqueId := MongoDBLibrary.GetUniqueIdentity()
+	uniqueId := MongoDBLibrary.GetUniqueIdentity("tmsi")
 	log.Println(uniqueId)
 
-	uniqueId = MongoDBLibrary.GetUniqueIdentity()
+	uniqueId = MongoDBLibrary.GetUniqueIdentity("amfUeNgapId")
 	log.Println(uniqueId)
 
 	uniqueId = MongoDBLibrary.GetUniqueIdentityWithinRange(3, 6)