77 lines
1.6 KiB
Go
77 lines
1.6 KiB
Go
|
package main
|
||
|
|
|
||
|
|
func createMapping(s []Mapping) error {
|
||
|
|
tx := orm.DB.Begin()
|
||
|
|
if tx.Error != nil {
|
||
|
|
return tx.Error
|
||
|
|
}
|
||
|
|
|
||
|
|
defer func() {
|
||
|
|
if r := recover(); r != nil {
|
||
|
|
logger.Warning(r)
|
||
|
|
tx.Rollback()
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
|
||
|
|
for _, val := range s {
|
||
|
|
if err := tx.Create(&val).Error; err != nil {
|
||
|
|
tx.Rollback()
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return tx.Commit().Error
|
||
|
|
}
|
||
|
|
|
||
|
|
func getConnection(uid string) (*Connection, error) {
|
||
|
|
var connection Connection
|
||
|
|
orm.DB.First(&connection, "client_id = ?", uid)
|
||
|
|
|
||
|
|
return &connection, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func getConnectionByURL(urlCrm string) (*Connection, error) {
|
||
|
|
var connection Connection
|
||
|
|
orm.DB.First(&connection, "api_url = ?", urlCrm)
|
||
|
|
|
||
|
|
return &connection, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *Connection) setConnectionActivity() error {
|
||
|
|
return orm.DB.Model(c).Where("client_id = ?", c.ClientID).Update("Active", c.Active).Error
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *Connection) createConnection() error {
|
||
|
|
return orm.DB.Create(c).Error
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *Connection) saveConnection() error {
|
||
|
|
return orm.DB.Model(c).Where("client_id = ?", c.ClientID).Update(c).Error
|
||
|
|
}
|
||
|
|
|
||
|
|
func getBotByToken(token string) (*Bot, error) {
|
||
|
|
var bot Bot
|
||
|
|
orm.DB.First(&bot, "token = ?", token)
|
||
|
|
|
||
|
|
return &bot, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *Bot) createBot() error {
|
||
|
|
return orm.DB.Create(b).Error
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *Bot) setBotActivity() error {
|
||
|
|
return orm.DB.Model(b).Where("token = ?", b.Token).Update("Active", !b.Active).Error
|
||
|
|
}
|
||
|
|
|
||
|
|
func getBotChannelByToken(token string) uint64 {
|
||
|
|
var b Bot
|
||
|
|
orm.DB.First(&b, "token = ?", token)
|
||
|
|
|
||
|
|
return b.Channel
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *Bots) getBotsByClientID(uid string) error {
|
||
|
|
return orm.DB.Where("client_id = ?", uid).Find(b).Error
|
||
|
|
}
|