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  }