@@ -79,14 +79,14 @@ func (o *User) FindOneAndSave(ctx context.Context, query interface{}) (*mongo.Si
7979 col := UserMgr .GetCol ()
8080 opts := options .FindOneAndUpdate ().SetUpsert (true )
8181 opts .SetReturnDocument (options .After )
82+ setFields := bson.M {}
83+ setFields [UserMgoFieldUserId ] = o .UserId
84+ setFields [UserMgoFieldUsername ] = o .Username
85+ setFields [UserMgoFieldAge ] = o .Age
86+ setFields [UserMgoFieldBlogs ] = o .Blogs
87+ setFields [UserMgoFieldRegisterDate ] = o .RegisterDate
8288 update := bson.M {
83- "$set" : bson.M {
84- UserMgoFieldUserId : o .UserId ,
85- UserMgoFieldUsername : o .Username ,
86- UserMgoFieldAge : o .Age ,
87- UserMgoFieldBlogs : o .Blogs ,
88- UserMgoFieldRegisterDate : o .RegisterDate ,
89- },
89+ "$set" : setFields ,
9090 }
9191 ret := col .FindOneAndUpdate (ctx , query , update , opts )
9292 if ret .Err () != nil {
@@ -103,14 +103,14 @@ func (o *User) Save(ctx context.Context) (*mongo.UpdateResult, error) {
103103 }
104104
105105 filter := bson.M {"_id" : o .ID }
106+ setFields := bson.M {}
107+ setFields [UserMgoFieldUserId ] = o .UserId
108+ setFields [UserMgoFieldUsername ] = o .Username
109+ setFields [UserMgoFieldAge ] = o .Age
110+ setFields [UserMgoFieldBlogs ] = o .Blogs
111+ setFields [UserMgoFieldRegisterDate ] = o .RegisterDate
106112 update := bson.M {
107- "$set" : bson.M {
108- UserMgoFieldUserId : o .UserId ,
109- UserMgoFieldUsername : o .Username ,
110- UserMgoFieldAge : o .Age ,
111- UserMgoFieldBlogs : o .Blogs ,
112- UserMgoFieldRegisterDate : o .RegisterDate ,
113- },
113+ "$set" : setFields ,
114114 }
115115
116116 opts := options .Update ().SetUpsert (true )
@@ -135,15 +135,15 @@ func (o *User) Save(ctx context.Context) (*mongo.UpdateResult, error) {
135135}
136136
137137func (o * User ) InsertUnique (ctx context.Context , query interface {}) (saved bool , err error ) {
138+ setFields := bson.M {}
139+ setFields [UserMgoFieldID ] = o .ID
140+ setFields [UserMgoFieldUserId ] = o .UserId
141+ setFields [UserMgoFieldUsername ] = o .Username
142+ setFields [UserMgoFieldAge ] = o .Age
143+ setFields [UserMgoFieldBlogs ] = o .Blogs
144+ setFields [UserMgoFieldRegisterDate ] = o .RegisterDate
138145 update := bson.M {
139- "$setOnInsert" : bson.M {
140- UserMgoFieldID : o .ID ,
141- UserMgoFieldUserId : o .UserId ,
142- UserMgoFieldUsername : o .Username ,
143- UserMgoFieldAge : o .Age ,
144- UserMgoFieldBlogs : o .Blogs ,
145- UserMgoFieldRegisterDate : o .RegisterDate ,
146- },
146+ "$setOnInsert" : setFields ,
147147 }
148148
149149 opts := options .Update ().SetUpsert (true )
0 commit comments