Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 7 additions & 7 deletions airecord/recordcfg.go
Original file line number Diff line number Diff line change
Expand Up @@ -30,24 +30,24 @@ func init() {
if err := sdb.db.Open(time.Hour * 24); err != nil {
panic(err)
}
if err := sdb.db.Create("config", &recordconfig{}); err != nil {
if err := sdb.db.Create("config", &RecordConfig{}); err != nil {
panic(err)
}
}

// recordconfig 存储语音记录相关配置
type recordconfig struct {
// RecordConfig 存储语音记录相关配置
type RecordConfig struct {
ID int64 `db:"id"` // 主键ID
ModelName string `db:"modelName"` // 语音模型名称
ModelID string `db:"modelID"` // 语音模型ID
Customgid int64 `db:"customgid"` // 自定义群ID
}

// GetConfig 获取当前配置
func GetConfig() recordconfig {
func GetConfig() RecordConfig {
sdb.RLock()
defer sdb.RUnlock()
cfg := recordconfig{}
cfg := RecordConfig{}
_ = sdb.db.Find("config", &cfg, "WHERE id = 1")
return cfg
}
Expand All @@ -57,7 +57,7 @@ func SetRecordModel(modelName, modelID string) error {
cfg := GetConfig()
sdb.Lock()
defer sdb.Unlock()
return sdb.db.Insert("config", &recordconfig{
return sdb.db.Insert("config", &RecordConfig{
ID: 1,
ModelName: modelName,
ModelID: modelID,
Expand All @@ -70,7 +70,7 @@ func SetCustomGID(gid int64) error {
cfg := GetConfig()
sdb.Lock()
defer sdb.Unlock()
return sdb.db.Insert("config", &recordconfig{
return sdb.db.Insert("config", &RecordConfig{
ID: 1,
ModelName: cfg.ModelName,
ModelID: cfg.ModelID,
Expand Down
Loading