databag/net/server/internal/api_setNodeConfig.go

70 lines
1.9 KiB
Go
Raw Normal View History

2022-01-15 21:45:37 +00:00
package databag
import (
"net/http"
"gorm.io/gorm"
2022-01-17 05:11:24 +00:00
"gorm.io/gorm/clause"
2022-01-15 21:45:37 +00:00
"databag/internal/store"
)
func SetNodeConfig(w http.ResponseWriter, r *http.Request) {
// validate login
if code, err := ParamAdminToken(r); err != nil {
ErrResponse(w, code, err)
2022-01-15 21:45:37 +00:00
return
}
// parse node config
2022-01-17 21:42:17 +00:00
var config NodeConfig
2022-01-19 20:07:57 +00:00
if err := ParseRequest(r, w, &config); err != nil {
ErrResponse(w, http.StatusBadRequest, err)
2022-01-15 21:45:37 +00:00
return
}
// store credentials
err := store.DB.Transaction(func(tx *gorm.DB) error {
2022-01-17 05:11:24 +00:00
// upsert domain config
if res := tx.Clauses(clause.OnConflict{
Columns: []clause.Column{{Name: "config_id"}},
DoUpdates: clause.AssignmentColumns([]string{"str_value"}),
2022-07-22 18:01:29 +00:00
}).Create(&store.Config{ConfigID: CNFDomain, StrValue: config.Domain}).Error; res != nil {
2022-01-15 21:45:37 +00:00
return res
}
2022-01-17 05:11:24 +00:00
2022-03-11 05:26:17 +00:00
// upsert account limit config
2022-01-17 05:11:24 +00:00
if res := tx.Clauses(clause.OnConflict{
Columns: []clause.Column{{Name: "config_id"}},
DoUpdates: clause.AssignmentColumns([]string{"num_value"}),
2022-07-22 18:01:29 +00:00
}).Create(&store.Config{ConfigID: CNFAccountLimit, NumValue: config.AccountLimit}).Error; res != nil {
2022-03-11 05:26:17 +00:00
return res
}
// upsert account open access
if res := tx.Clauses(clause.OnConflict{
Columns: []clause.Column{{Name: "config_id"}},
DoUpdates: clause.AssignmentColumns([]string{"bool_value"}),
2022-07-22 18:01:29 +00:00
}).Create(&store.Config{ConfigID: CNFAccountLimit, BoolValue: config.OpenAccess}).Error; res != nil {
2022-01-15 21:45:37 +00:00
return res
}
2022-01-17 05:11:24 +00:00
// upsert account storage config
if res := tx.Clauses(clause.OnConflict{
Columns: []clause.Column{{Name: "config_id"}},
DoUpdates: clause.AssignmentColumns([]string{"num_value"}),
2022-07-22 18:01:29 +00:00
}).Create(&store.Config{ConfigID: CNFStorage, NumValue: config.AccountStorage}).Error; res != nil {
2022-01-15 21:45:37 +00:00
return res
}
2022-01-17 05:11:24 +00:00
2022-01-15 21:45:37 +00:00
return nil;
})
if(err != nil) {
2022-01-19 20:07:57 +00:00
ErrResponse(w, http.StatusInternalServerError, err)
2022-01-15 21:45:37 +00:00
return
}
w.WriteHeader(http.StatusOK)
}