From 34e6252ef2627367db5e3b0f964e73b1e4522227 Mon Sep 17 00:00:00 2001 From: manoj Date: Wed, 2 Nov 2022 23:00:03 +0530 Subject: [PATCH] user crud added --- server/db/providers/couchbase/user.go | 135 +++++++++++++++++++++++++- 1 file changed, 133 insertions(+), 2 deletions(-) diff --git a/server/db/providers/couchbase/user.go b/server/db/providers/couchbase/user.go index 67c4395..90659db 100644 --- a/server/db/providers/couchbase/user.go +++ b/server/db/providers/couchbase/user.go @@ -2,12 +2,16 @@ package couchbase import ( "context" + "fmt" + "log" + "reflect" "time" "github.com/authorizerdev/authorizer/server/constants" "github.com/authorizerdev/authorizer/server/db/models" "github.com/authorizerdev/authorizer/server/graph/model" "github.com/authorizerdev/authorizer/server/memorystore" + "github.com/couchbase/gocb/v2" "github.com/google/uuid" ) @@ -27,29 +31,108 @@ func (p *provider) AddUser(ctx context.Context, user models.User) (models.User, user.CreatedAt = time.Now().Unix() user.UpdatedAt = time.Now().Unix() - + insertOpt := gocb.InsertOptions{ + Context: ctx, + } + _, err := p.db.Collection(models.Collections.User).Insert(user.ID, user, &insertOpt) + if err != nil { + return user, err + } return user, nil } // UpdateUser to update user information in database func (p *provider) UpdateUser(ctx context.Context, user models.User) (models.User, error) { user.UpdatedAt = time.Now().Unix() + unsertOpt := gocb.UpsertOptions{ + Context: ctx, + } + _, err := p.db.Collection(models.Collections.User).Upsert(user.ID, user, &unsertOpt) + if err != nil { + return user, err + } return user, nil } // DeleteUser to delete user information from database func (p *provider) DeleteUser(ctx context.Context, user models.User) error { + removeOpt := gocb.RemoveOptions{ + Context: ctx, + } + _, err := p.db.Collection(models.Collections.User).Remove(user.ID, &removeOpt) + // query := fmt.Sprintf("INSERT INTO %s %s VALUES %s IF NOT EXISTS", KeySpace+"."+models.Collections.User, fields, values) + + // sessionCollection := p.db.Collection(models.Collections.Session).Queue() + // _, err = sessionCollection.DeleteMany(ctx, bson.M{"user_id": user.ID}, options.Delete()) + // if err != nil { + // return err + // } + if err != nil { + return err + } return nil } // ListUsers to get list of users from database func (p *provider) ListUsers(ctx context.Context, pagination model.Pagination) (*model.Users, error) { - return nil, nil + users := []*model.User{} + // r := p.db.Collection(models.Collections.User). + paginationClone := pagination + + inventoryScope := p.db.Scope("_default") + userQuery := fmt.Sprintf("SELECT * FROM auth._default.%s ORDER BY id OFFSET %d LIMIT %d", models.Collections.User, paginationClone.Offset, paginationClone.Limit) + + queryResult, err := inventoryScope.Query(userQuery, &gocb.QueryOptions{}) + + if err != nil { + return nil, err + } + + for queryResult.Next() { + var user models.User + err := queryResult.Row(&user) + if err != nil { + log.Fatal(err) + } + users = append(users, user.AsAPIUser()) + } + + if err := queryResult.Err(); err != nil { + return nil, err + + } + + return &model.Users{ + Pagination: &paginationClone, + Users: users, + }, nil } // GetUserByEmail to get user information from database using email address func (p *provider) GetUserByEmail(ctx context.Context, email string) (models.User, error) { var user models.User + // inventoryScope := p.db.Scope("_default") + // userQuery := fmt.Sprintf("SELECT * FROM auth._default.%s WHERE email = '%s' LIMIT %d", models.Collections.User, email, 1) + queryResult, err := p.db.Collection(models.Collections.User).Get("email:"+email, &gocb.GetOptions{}) + if err != nil { + return user, err + } + + // queryResult, err := inventoryScope.Query(userQuery, &gocb.QueryOptions{}) + err = queryResult.Content(&user) + + if err != nil { + return user, err + } + + // for queryResult.Next() { + // var user models.User + // err := queryResult.Row(&user) + // if err != nil { + // log.Fatal(err) + // } + // return user, nil + // } return user, nil } @@ -57,7 +140,15 @@ func (p *provider) GetUserByEmail(ctx context.Context, email string) (models.Use // GetUserByID to get user information from database using user ID func (p *provider) GetUserByID(ctx context.Context, id string) (models.User, error) { var user models.User + queryResult, err := p.db.Collection(models.Collections.User).Get("id:"+id, &gocb.GetOptions{}) + if err != nil { + return user, err + } + err = queryResult.Content(&user) + if err != nil { + return user, err + } return user, nil } @@ -66,6 +157,46 @@ func (p *provider) GetUserByID(ctx context.Context, id string) (models.User, err func (p *provider) UpdateUsers(ctx context.Context, data map[string]interface{}, ids []string) error { // set updated_at time for all users data["updated_at"] = time.Now().Unix() + inventoryScope := p.db.Scope("_default") + + updateFields := "" + for key, value := range data { + if key == "_id" { + continue + } + + if key == "_key" { + continue + } + + if value == nil { + updateFields += fmt.Sprintf("%s = null,", key) + continue + } + + valueType := reflect.TypeOf(value) + if valueType.Name() == "string" { + updateFields += fmt.Sprintf("%s = '%s', ", key, value.(string)) + } else { + updateFields += fmt.Sprintf("%s = %v, ", key, value) + } + } + + if ids != nil && len(ids) > 0 { + for _, v := range ids { + userQuery := fmt.Sprintf("UPDATE * FROM auth._default.%s SET %s WHERE id = '%s'", models.Collections.User, updateFields, v) + _, err := inventoryScope.Query(userQuery, &gocb.QueryOptions{}) + if err != nil { + return err + } + } + } else { + userQuery := fmt.Sprintf("UPDATE * FROM auth._default.%s SET %s WHERE id IS NOT NULL", models.Collections.User, updateFields) + _, err := inventoryScope.Query(userQuery, &gocb.QueryOptions{}) + if err != nil { + return err + } + } return nil }