pkg/SugarDB
1
0
Fork 0
mirror of https://github.com/EchoVault/SugarDB.git synced 2025-10-05 07:56:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

Updated logic for loading acl config. If the config file does not exist, it will be created.

Browse Source
This commit is contained in:
Kelvin Clement Mwinuka
2024-06-01 22:16:16 +08:00
parent 4d56ee9083
commit bdfaf5446a
3 changed files with 236 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
internal/modules/acl/commands.go
View File

@@ -442,7 +442,7 @@ func handleSave(params internal.HandlerFuncParams) ([]byte, error) {
acl.RLockUsers()
acl.RUnlockUsers()
f, err := os.OpenFile(acl.Config.AclConfig, os.O_WRONLY|os.O_CREATE, os.ModeAppend)
f, err := os.OpenFile(acl.Config.AclConfig, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, os.ModePerm)
if err != nil {
return nil, err
}
@@ -455,32 +455,29 @@ func handleSave(params internal.HandlerFuncParams) ([]byte, error) {
ext := path.Ext(f.Name())
if ext == ".json" {
if strings.ToLower(ext) == ".json" {
// Write to JSON config file
out, err := json.Marshal(acl.Users)
if err != nil {
return nil, err
}
_, err = f.Write(out)
if err != nil {
if _, err = f.Write(out); err != nil {
return nil, err
}
}
if ext == ".yaml" || ext == ".yml" {
if slices.Contains([]string{".yaml", ".yml"}, strings.ToLower(ext)) {
// Write to yaml file
out, err := yaml.Marshal(acl.Users)
if err != nil {
return nil, err
}
_, err = f.Write(out)
if err != nil {
if _, err = f.Write(out); err != nil {
return nil, err
}
}
err = f.Sync()
if err != nil {
if err = f.Sync(); err != nil {
return nil, err
}