add code grpc to manage new collection db groupMember

storage/groups.go

@@ -6,3 +6,10 @@ type Group struct {
 	Members   []string       `json:"members"`
 	Data      map[string]any `json:"data"`
 }
+
+///////////////////code
+type GroupMember struct {
+	ID      string         `json:"id" bson:"_id"`
+	Groupid string         `json:"grouid" bson:"_groupid"`
+	Data    map[string]any `json:"data"`
+}

storage/mongodb.go

@@ -23,6 +23,8 @@ func NewMongoDBStorage(cfg *viper.Viper) (MongoDBStorage, error) {
 		mongodb_port   = cfg.GetString("storage.db.mongodb.port")
 		mongodb_dbname = cfg.GetString("storage.db.mongodb.db_name")
 		mongodb_groups = cfg.GetString("storage.db.mongodb.collections.groups")
+		////////////////////////code
+		mongodb_groups_members = cfg.GetString("storage.db.mongodb.collections.groups_member")
 	)
 
 	client, err := mongo.NewClient(options.Client().ApplyURI("mongodb://" + mongodb_host + ":" + mongodb_port))
@@ -41,6 +43,8 @@ func NewMongoDBStorage(cfg *viper.Viper) (MongoDBStorage, error) {
 		DbName: mongodb_dbname,
 		Collections: map[string]string{
 			"groups": mongodb_groups,
+			/////////////////////////////code
+			"groups_member": mongodb_groups_members,
 		},
 	}
 	//TODO Indexes
@@ -147,3 +151,113 @@ func (s MongoDBStorage) UpdateGroup(group Group) error {
 
 	return nil
 }
+
+/*********************************************************************************************************************************/
+/*********************************************************Code********************************************************************/
+/*********************************************************************************************************************************/
+/*********************************************************************************************************************************/
+
+func (s MongoDBStorage) CreateGroupMember(groupMember GroupMember) error {
+	collection := s.Client.Database(s.DbName).Collection(s.Collections["groups_member"])
+	if _, err := collection.InsertOne(context.TODO(), groupMember); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (s MongoDBStorage) GetGroupMember(id string) (*GroupMember, error) {
+	collection := s.Client.Database(s.DbName).Collection(s.Collections["groups_member"])
+
+	groupMember := &GroupMember{}
+	if err := collection.FindOne(context.TODO(), bson.M{"_id": id}).Decode(groupMember); err != nil {
+		return nil, err
+	}
+
+	return groupMember, nil
+}
+
+func (s MongoDBStorage) GetGroupsMember(namespaces []string) (groupsMember []GroupMember, err error) {
+	collection := s.Client.Database(s.DbName).Collection(s.Collections["groups_member"])
+
+	var cur *mongo.Cursor
+
+	findOptions := options.Find()
+
+	cur, err = collection.Find(context.TODO(), bson.D{}, findOptions)
+	if err != nil {
+		return groupsMember, err
+	}
+	// if len(namespaces) == 0 {
+	// 	cur, err = collection.Find(context.TODO(), bson.D{}, findOptions)
+	// 	if err != nil {
+	// 		return groupsMember, err
+	// 	}
+	// } else {
+	// 	cur, err = collection.Find(context.TODO(), bson.M{"namespace": bson.M{"$in": namespaces}}, findOptions)
+	// 	if err != nil {
+	// 		return groupsMember, err
+	// 	}
+	// }
+
+	for cur.Next(context.TODO()) {
+		var group GroupMember
+		var elem bson.M
+
+		if err := cur.Decode(&elem); err != nil {
+			return groupsMember, err
+		}
+
+		bsonBytes, _ := bson.Marshal(elem)
+		bson.Unmarshal(bsonBytes, &group)
+
+		groupsMember = append(groupsMember, group)
+
+	}
+
+	return
+}
+
+func (s MongoDBStorage) GetGroupsMemberByIds(groupids []string) (groupsMember []GroupMember, err error) {
+	collection := s.Client.Database(s.DbName).Collection(s.Collections["groups_member"])
+
+	var cur *mongo.Cursor
+
+	findOptions := options.Find()
+
+	if len(groupids) == 0 {
+		return groupsMember, errors.New("no group id provided")
+	} else {
+		cur, err = collection.Find(context.TODO(), bson.M{"_id": bson.M{"$in": groupids}}, findOptions)
+		if err != nil {
+			return groupsMember, err
+		}
+	}
+
+	for cur.Next(context.TODO()) {
+		var group GroupMember
+		var elem bson.M
+
+		if err := cur.Decode(&elem); err != nil {
+			return groupsMember, err
+		}
+
+		bsonBytes, _ := bson.Marshal(elem)
+		bson.Unmarshal(bsonBytes, &group)
+
+		groupsMember = append(groupsMember, group)
+
+	}
+
+	return
+}
+
+func (s MongoDBStorage) UpdateGroupMember(group GroupMember) error {
+	collection := s.Client.Database(s.DbName).Collection(s.Collections["groups_member"])
+	if _, err := collection.ReplaceOne(context.TODO(), bson.M{"_id": group.ID}, group); err != nil {
+		fmt.Println(err)
+		return err
+	}
+
+	return nil
+}

storage/storage.go

@@ -12,6 +12,12 @@ type Storage interface {
 	GetGroups([]string) ([]Group, error)
 	GetGroupsByIds([]string) ([]Group, error)
 	UpdateGroup(Group) error
+	/*******************************************************/
+	CreateGroupMember(GroupMember) error
+	GetGroupMember(string) (*GroupMember, error)
+	GetGroupsMember([]string) ([]GroupMember, error)
+	GetGroupsMemberByIds([]string) ([]GroupMember, error)
+	UpdateGroupMember(GroupMember) error
 }
 
 func NewStorage(cfg *viper.Viper) (Storage, error) {