github.com/hungdoo/bot@v0.0.0-20240325145135-dd1f386f7b81/src/services/telecommands/helper.go (about)

     1  package telecommands
     2  
     3  import (
     4  	command "github.com/hungdoo/bot/src/packages/command/common"
     5  	"github.com/hungdoo/bot/src/packages/db"
     6  	"github.com/hungdoo/bot/src/packages/log"
     7  	"go.mongodb.org/mongo-driver/bson"
     8  )
     9  
    10  func StoreCmd(cmd command.ICommand) error {
    11  	if err := db.GetDb().Insert("commands", cmd); err != nil {
    12  		log.GeneralLogger.Printf("Job [%s] insert db failed: [%s]", cmd.GetName(), err)
    13  		return err
    14  	}
    15  	return nil
    16  }
    17  
    18  func UpdateCmd(cmd command.ICommand) error {
    19  	filter := bson.M{"_id": cmd.GetName()}
    20  	update := bson.M{"$set": cmd}
    21  	if err := db.GetDb().Update("commands", filter, update); err != nil {
    22  		log.GeneralLogger.Printf("Job [%s] update db failed: [%s]", cmd.GetName(), err)
    23  		return err
    24  	}
    25  	return nil
    26  }
    27  
    28  func UpdateMultiCmd(cmds []command.ICommand) {
    29  	for _, c := range cmds {
    30  		if err := UpdateCmd(c); err != nil {
    31  			log.GeneralLogger.Printf("Job [%s] update db failed: [%s]", c.GetName(), err)
    32  		}
    33  	}
    34  }