gotta swap to tcp conns
This commit is contained in:
64
src/config.go
Normal file
64
src/config.go
Normal file
@@ -0,0 +1,64 @@
|
||||
package src
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"os"
|
||||
"slices"
|
||||
"strings"
|
||||
|
||||
valkey "github.com/redis/go-redis/v9"
|
||||
)
|
||||
|
||||
type Config struct {
|
||||
Listen string `json:"LISTEN"`
|
||||
Forwards string `json:"FORWARDS"`
|
||||
forwards []*valkey.Client
|
||||
}
|
||||
|
||||
func NewConfig(ctx context.Context) (Config, error) {
|
||||
config := Config{
|
||||
Listen: ":10000",
|
||||
Forwards: "",
|
||||
}
|
||||
b, _ := json.Marshal(config)
|
||||
var m map[string]any
|
||||
if err := json.Unmarshal(b, &m); err != nil {
|
||||
return config, err
|
||||
}
|
||||
for k := range m {
|
||||
if v := os.Getenv(k); v != "" {
|
||||
m[k] = v
|
||||
}
|
||||
}
|
||||
b2, _ := json.Marshal(m)
|
||||
if err := json.Unmarshal(b2, &config); err != nil {
|
||||
return config, err
|
||||
}
|
||||
|
||||
forwards := strings.Split(config.Forwards, ",")
|
||||
forwards = slices.DeleteFunc(forwards, func(s string) bool { return s == "" })
|
||||
config.forwards = make([]*valkey.Client, len(forwards))
|
||||
for i := range forwards {
|
||||
opt, err := valkey.ParseURL(forwards[i])
|
||||
if err != nil {
|
||||
config.Close()
|
||||
return config, err
|
||||
}
|
||||
config.forwards[i] = valkey.NewClient(opt)
|
||||
if err != nil {
|
||||
config.Close()
|
||||
return config, err
|
||||
}
|
||||
}
|
||||
|
||||
return config, nil
|
||||
}
|
||||
|
||||
func (c Config) Close() {
|
||||
for i := range c.forwards {
|
||||
if c.forwards[i] != nil {
|
||||
c.forwards[i].Close()
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user