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 }