github.com/infraboard/keyauth@v0.8.1/apps/wxwork/impl/dao.go (about) 1 package impl 2 3 import ( 4 "context" 5 6 "github.com/infraboard/keyauth/apps/wxwork" 7 8 "github.com/infraboard/mcube/exception" 9 "github.com/infraboard/mcube/types/ftime" 10 "go.mongodb.org/mongo-driver/bson" 11 ) 12 13 func (s *service) save(ins *wxwork.WechatWorkConfig) error { 14 if _, err := s.col.InsertOne(context.TODO(), ins); err != nil { 15 return exception.NewInternalServerError("inserted wechat work config(%s) document error, %s", 16 ins.Domain, err) 17 } 18 return nil 19 } 20 21 func (s *service) update(ins *wxwork.WechatWorkConfig) error { 22 ins.UpdateAt = ftime.Now() 23 _, err := s.col.UpdateOne(context.TODO(), bson.M{"_id": ins.Domain}, bson.M{"$set": ins}) 24 if err != nil { 25 return exception.NewInternalServerError("update domain(%s) error, %s", ins.Domain, err) 26 } 27 28 return nil 29 }