28 lines
566 B
Go
28 lines
566 B
Go
package impl
|
|
|
|
import (
|
|
"ai_scheduler/internal/data/model"
|
|
"ai_scheduler/tmpl/dataTemp"
|
|
"ai_scheduler/utils"
|
|
"database/sql"
|
|
)
|
|
|
|
type BotUserImpl struct {
|
|
dataTemp.DataTemp
|
|
}
|
|
|
|
func NewBotUserImpl(db *utils.Db) *BotUserImpl {
|
|
return &BotUserImpl{
|
|
DataTemp: *dataTemp.NewDataTemp(db, new(model.AiBotUser)),
|
|
}
|
|
}
|
|
|
|
func (k BotUserImpl) GetByStaffId(staffId string) (*model.AiBotUser, error) {
|
|
var data model.AiBotUser
|
|
err := k.Db.Model(k.Model).Where("staff_id = ?", staffId).Find(&data).Error
|
|
if data.UserID == 0 {
|
|
err = sql.ErrNoRows
|
|
}
|
|
return &data, err
|
|
}
|