1
0
mirror of https://github.com/fumiama/imago.git synced 2026-06-07 17:30:26 +08:00

add funcs in storage instance

This commit is contained in:
fumiama
2022-01-06 16:24:07 +08:00
parent 8ce09fa76d
commit d32e0e45f6
3 changed files with 23 additions and 0 deletions

View File

@@ -17,6 +17,8 @@ type StorageInstance interface {
IsImgExsits(name string) bool
AddImage(name string)
Pick(exclude []string) string
SaveConf(data []byte) error
GetConf() ([]byte, error)
}
type storage struct {

View File

@@ -120,3 +120,13 @@ func (ns *NativeStorage) ScanImgs(imgdir string) error {
}
return nil
}
// SaveConf Save config file into storage
func (ns *NativeStorage) SaveConf(data []byte) error {
return os.WriteFile("conf.pb", data, 0644)
}
// SaveConf Save config file into storage
func (ns *NativeStorage) GetConf() ([]byte, error) {
return os.ReadFile("conf.pb")
}

View File

@@ -123,3 +123,14 @@ func (remo *RemoteStorage) ScanImgs(imgdir string) error {
}
return nil
}
// SaveConf Save config file into storage
func (remo *RemoteStorage) SaveConf(data []byte) error {
return remo.cli.SetFile("cfg", "conf.pb", data)
}
// SaveConf Save config file into storage
func (remo *RemoteStorage) GetConf() (data []byte, err error) {
data, _, err = remo.cli.GetFile("cfg", "conf.pb")
return
}