mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-17 04:24:35 +03:00
137 lines
3.3 KiB
Go
137 lines
3.3 KiB
Go
package settings
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/hasura/graphql-engine/cli/internal/hasura"
|
|
)
|
|
|
|
type StateStoreHdbTable struct {
|
|
client hasura.PGSourceOps
|
|
schema, table string
|
|
}
|
|
|
|
func NewStateStoreHdbTable(client hasura.PGSourceOps, schema, table string) *StateStoreHdbTable {
|
|
return &StateStoreHdbTable{client, schema, table}
|
|
}
|
|
|
|
func (s *StateStoreHdbTable) GetSetting(name string) (value string, err error) {
|
|
query := hasura.PGRunSQLInput{
|
|
SQL: `SELECT value from ` + fmt.Sprintf("%s.%s", s.schema, s.table) + ` where setting='` + name + `'`,
|
|
}
|
|
|
|
resp, err := s.client.PGRunSQL(query)
|
|
if err != nil {
|
|
return value, err
|
|
}
|
|
|
|
if resp.ResultType != hasura.TuplesOK {
|
|
return value, fmt.Errorf("invalid result Type %s", resp.ResultType)
|
|
}
|
|
|
|
if len(resp.Result) < 2 {
|
|
for _, setting := range Settings {
|
|
if setting.GetName() == name {
|
|
return setting.GetDefaultValue(), nil
|
|
}
|
|
}
|
|
return value, fmt.Errorf("invalid setting name: %s", name)
|
|
}
|
|
|
|
return resp.Result[1][0], nil
|
|
}
|
|
|
|
func (s *StateStoreHdbTable) GetAllSettings() (map[string]string, error) {
|
|
query := hasura.PGRunSQLInput{
|
|
SQL: `SELECT setting, value from ` + fmt.Sprintf("%s.%s", s.schema, s.table) + `;`,
|
|
}
|
|
|
|
resp, err := s.client.PGRunSQL(query)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if resp.ResultType != hasura.TuplesOK {
|
|
return nil, fmt.Errorf("invalid result Type %s", resp.ResultType)
|
|
}
|
|
|
|
var settings = map[string]string{}
|
|
for idx, row := range resp.Result {
|
|
if idx == 0 {
|
|
continue
|
|
}
|
|
if len(row) == 2 {
|
|
settings[row[0]] = row[1]
|
|
}
|
|
}
|
|
return settings, nil
|
|
}
|
|
|
|
func (s *StateStoreHdbTable) UpdateSetting(name string, value string) error {
|
|
query := hasura.PGRunSQLInput{
|
|
SQL: `INSERT INTO ` + fmt.Sprintf("%s.%s", s.schema, s.table) + ` (setting, value) VALUES ('` + name + `', '` + value + `') ON CONFLICT (setting) DO UPDATE SET value='` + value + `'`,
|
|
}
|
|
|
|
resp, err := s.client.PGRunSQL(query)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if resp.ResultType != hasura.CommandOK {
|
|
return fmt.Errorf("cannot set setting %s to %s", name, value)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *StateStoreHdbTable) PrepareSettingsDriver() error {
|
|
// check if migration table exists
|
|
query := hasura.PGRunSQLInput{
|
|
SQL: `SELECT COUNT(1) FROM information_schema.tables WHERE table_name = '` + s.table + `' AND table_schema = '` + s.schema + `' LIMIT 1`,
|
|
}
|
|
|
|
resp, err := s.client.PGRunSQL(query)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if resp.ResultType != hasura.TuplesOK {
|
|
return fmt.Errorf("invalid result Type %s", resp.ResultType)
|
|
}
|
|
|
|
if resp.Result[1][0] != "0" {
|
|
return nil
|
|
}
|
|
|
|
// Now Create the table
|
|
query = hasura.PGRunSQLInput{
|
|
SQL: `CREATE TABLE ` + fmt.Sprintf("%s.%s", s.schema, s.table) + ` (setting text not null primary key, value text not null)`,
|
|
}
|
|
|
|
resp, err = s.client.PGRunSQL(query)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if resp.ResultType != hasura.CommandOK {
|
|
return fmt.Errorf("creating Version table failed %s", resp.ResultType)
|
|
}
|
|
return s.setDefaults()
|
|
}
|
|
|
|
func (s *StateStoreHdbTable) setDefaults() error {
|
|
var sql string
|
|
for _, setting := range Settings {
|
|
sql += `INSERT INTO ` + fmt.Sprintf("%s.%s", s.schema, s.table) + ` (setting, value) VALUES ('` + fmt.Sprintf("%s", setting.GetName()) + `', '` + fmt.Sprintf("%s", setting.GetDefaultValue()) + `');`
|
|
}
|
|
|
|
query := hasura.PGRunSQLInput{
|
|
SQL: sql,
|
|
}
|
|
_, err := s.client.PGRunSQL(query)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|