mirror of
https://github.com/simon-ding/polaris.git
synced 2026-06-10 03:57:39 +08:00
chore: updates
This commit is contained in:
22
db/db.go
22
db/db.go
@@ -42,11 +42,22 @@ func Open() (*Client, error) {
|
||||
ent: client,
|
||||
}
|
||||
|
||||
c.generateJwtSerectIfNotExist()
|
||||
c.generateDefaultLocalStorage()
|
||||
return c, nil
|
||||
}
|
||||
|
||||
func (c *Client) init() {
|
||||
c.generateJwtSerectIfNotExist()
|
||||
if err := c.generateDefaultLocalStorage(); err != nil {
|
||||
log.Errorf("generate default storage: %v", err)
|
||||
}
|
||||
|
||||
downloadDir := c.GetSetting(SettingDownloadDir)
|
||||
if downloadDir == "" {
|
||||
log.Infof("set default download dir")
|
||||
c.SetSetting(downloadDir, "/downloads")
|
||||
}
|
||||
}
|
||||
|
||||
func (c *Client) generateJwtSerectIfNotExist() {
|
||||
v := c.GetSetting(JwtSerectKey)
|
||||
if v == "" {
|
||||
@@ -56,12 +67,13 @@ func (c *Client) generateJwtSerectIfNotExist() {
|
||||
}
|
||||
}
|
||||
|
||||
func (c *Client) generateDefaultLocalStorage() {
|
||||
func (c *Client) generateDefaultLocalStorage() error {
|
||||
n, _ := c.ent.Storage.Query().Count(context.TODO())
|
||||
if n != 0 {
|
||||
return
|
||||
return nil
|
||||
}
|
||||
c.AddStorage(&StorageInfo{
|
||||
log.Infof("add default storage")
|
||||
return c.AddStorage(&StorageInfo{
|
||||
Name: "local",
|
||||
Implementation: "local",
|
||||
Default: true,
|
||||
|
||||
Reference in New Issue
Block a user