update

Author: wpcodevo

models/user.model.go

@@ -12,6 +12,7 @@ type SignUpInput struct {
 	Password        string    `json:"password" bson:"password" binding:"required,min=8"`
 	PasswordConfirm string    `json:"passwordConfirm" bson:"passwordConfirm,omitempty" binding:"required"`
 	Role            string    `json:"role" bson:"role"`
+	Provider        string    `json:"provider" bson:"provider"`
 	Verified        bool      `json:"verified" bson:"verified"`
 	CreatedAt       time.Time `json:"created_at" bson:"created_at"`
 	UpdatedAt       time.Time `json:"updated_at" bson:"updated_at"`
@@ -28,6 +29,7 @@ type DBResponse struct {
 	Email           string             `json:"email" bson:"email"`
 	Password        string             `json:"password" bson:"password"`
 	PasswordConfirm string             `json:"passwordConfirm,omitempty" bson:"passwordConfirm,omitempty"`
+	Provider        string             `json:"provider" bson:"provider"`
 	Role            string             `json:"role" bson:"role"`
 	Verified        bool               `json:"verified" bson:"verified"`
 	CreatedAt       time.Time          `json:"created_at" bson:"created_at"`
@@ -39,16 +41,31 @@ type UserResponse struct {
 	Name      string             `json:"name,omitempty" bson:"name,omitempty"`
 	Email     string             `json:"email,omitempty" bson:"email,omitempty"`
 	Role      string             `json:"role,omitempty" bson:"role,omitempty"`
+	Provider  string             `json:"provider" bson:"provider"`
 	CreatedAt time.Time          `json:"created_at" bson:"created_at"`
 	UpdatedAt time.Time          `json:"updated_at" bson:"updated_at"`
 }
 
+type UpdateDBUser struct {
+	ID              primitive.ObjectID `json:"id,omitempty" bson:"_id,omitempty"`
+	Name            string             `json:"name,omitempty" bson:"name,omitempty"`
+	Email           string             `json:"email,omitempty" bson:"email,omitempty"`
+	Password        string             `json:"password,omitempty" bson:"password,omitempty"`
+	PasswordConfirm string             `json:"passwordConfirm,omitempty" bson:"passwordConfirm,omitempty"`
+	Role            string             `json:"role,omitempty" bson:"role,omitempty"`
+	Provider        string             `json:"provider" bson:"provider"`
+	Verified        bool               `json:"verified,omitempty" bson:"verified,omitempty"`
+	CreatedAt       time.Time          `json:"created_at,omitempty" bson:"created_at,omitempty"`
+	UpdatedAt       time.Time          `json:"updated_at,omitempty" bson:"updated_at,omitempty"`
+}
+
 func FilteredResponse(user *DBResponse) UserResponse {
 	return UserResponse{
 		ID:        user.ID,
 		Email:     user.Email,
 		Name:      user.Name,
 		Role:      user.Role,
+		Provider:  user.Provider,
 		CreatedAt: user.CreatedAt,
 		UpdatedAt: user.UpdatedAt,
 	}

services/user.service.impl.go

@@ -2,12 +2,14 @@ package services
 
 import (
 	"context"
+	"errors"
 	"strings"
 
 	"github.com/wpcodevo/golang-mongodb/models"
 	"go.mongodb.org/mongo-driver/bson"
 	"go.mongodb.org/mongo-driver/bson/primitive"
 	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
 )
 
 type UserServiceImpl struct {
@@ -52,3 +54,23 @@ func (us *UserServiceImpl) FindUserByEmail(email string) (*models.DBResponse, er
 
 	return user, nil
 }
+
+func (uc *UserServiceImpl) UpdateUser(id string, data *models.UpdateDBUser) (*models.DBResponse, error) {
+	doc, err := utils.ToDoc(data)
+	if err != nil {
+		return nil, err
+	}
+
+	obId, _ := primitive.ObjectIDFromHex(id)
+	query := bson.D{{Key: "_id", Value: obId}}
+	update := bson.D{{Key: "$set", Value: doc}}
+	res := uc.collection.FindOneAndUpdate(uc.ctx, query, update, options.FindOneAndUpdate().SetReturnDocument(1))
+
+	var updatedPost *models.DBResponse
+
+	if err := res.Decode(&updatedPost); err != nil {
+		return nil, errors.New("no post with that Id exists")
+	}
+
+	return updatedPost, nil
+}