66 lines
1.4 KiB
Go
66 lines
1.4 KiB
Go
package rest
|
||
|
||
import (
|
||
"github.com/eyebluecn/tank/code/core"
|
||
"github.com/eyebluecn/tank/code/tool/result"
|
||
"github.com/nu7hatch/gouuid"
|
||
"time"
|
||
)
|
||
|
||
type PreferenceDao struct {
|
||
BaseDao
|
||
}
|
||
|
||
//按照Id查询偏好设置
|
||
func (this *PreferenceDao) Fetch() *Preference {
|
||
|
||
// Read
|
||
var preference = &Preference{}
|
||
db := core.CONTEXT.GetDB().First(preference)
|
||
if db.Error != nil {
|
||
|
||
if db.Error.Error() == result.DB_ERROR_NOT_FOUND {
|
||
preference.Name = "蓝眼云盘"
|
||
|
||
this.Create(preference)
|
||
return preference
|
||
} else {
|
||
return nil
|
||
}
|
||
}
|
||
|
||
return preference
|
||
}
|
||
|
||
//创建
|
||
func (this *PreferenceDao) Create(preference *Preference) *Preference {
|
||
|
||
timeUUID, _ := uuid.NewV4()
|
||
preference.Uuid = string(timeUUID.String())
|
||
preference.CreateTime = time.Now()
|
||
preference.UpdateTime = time.Now()
|
||
preference.Sort = time.Now().UnixNano() / 1e6
|
||
db := core.CONTEXT.GetDB().Create(preference)
|
||
this.PanicError(db.Error)
|
||
|
||
return preference
|
||
}
|
||
|
||
//修改一个偏好设置
|
||
func (this *PreferenceDao) Save(preference *Preference) *Preference {
|
||
|
||
preference.UpdateTime = time.Now()
|
||
db := core.CONTEXT.GetDB().Save(preference)
|
||
this.PanicError(db.Error)
|
||
|
||
return preference
|
||
}
|
||
|
||
//执行清理操作
|
||
func (this *PreferenceDao) Cleanup() {
|
||
|
||
this.logger.Info("[PreferenceDao]执行清理：清除数据库中所有Preference记录。")
|
||
db := core.CONTEXT.GetDB().Where("uuid is not null").Delete(Preference{})
|
||
this.PanicError(db.Error)
|
||
}
|