jiazhizhong/polaris
1
0
Fork 0
mirror of https://github.com/simon-ding/polaris.git synced 2026-04-21 11:17:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: serval bugs

Browse Source
This commit is contained in:
Simon Ding
2024-07-12 13:35:37 +08:00
parent 76bf358de7
commit cfa6d2d54c
7 changed files with 79 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
db/db.go
View File

@@ -245,12 +245,19 @@ func (c *Client) AddStorage(s StorageInfo) error {
SetImplementation(s.Implementation).
SetPath(s.Path).
SetUser(s.User).
SetDefault(s.Default).
SetPassword(s.Password).Exec(context.TODO())
}
countAll := c.ent.Storage.Query().CountX(context.TODO())
if countAll == 0 {
log.Infof("first storage, make it default: %s", s.Name)
s.Default = true
}
_, err := c.ent.Storage.Create().SetName(s.Name).
SetImplementation(s.Implementation).
SetPath(s.Path).
SetUser(s.User).
SetDefault(s.Default).
SetPassword(s.Password).Save(context.TODO())
if err != nil {
return err