mirror of
https://github.com/luxfi/mpc.git
synced 2026-09-24 00:26:24 +00:00
zeekay
721db3ec4a
- Fix import formatting with goimports - Remove unnecessary type conversions for syscall.Stdin - Format imports with correct grouping
49 lines
1.2 KiB
Go
49 lines
1.2 KiB
Go
package infra
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/hashicorp/consul/api"
|
|
"github.com/spf13/viper"
|
|
|
|
"github.com/luxfi/mpc/pkg/constant"
|
|
"github.com/luxfi/mpc/pkg/logger"
|
|
)
|
|
|
|
type ConsulKV interface {
|
|
Put(kv *api.KVPair, options *api.WriteOptions) (*api.WriteMeta, error)
|
|
Get(key string, options *api.QueryOptions) (*api.KVPair, *api.QueryMeta, error)
|
|
Delete(key string, options *api.WriteOptions) (*api.WriteMeta, error)
|
|
List(prefix string, options *api.QueryOptions) (api.KVPairs, *api.QueryMeta, error)
|
|
}
|
|
|
|
func GetConsulClient(environment string) *api.Client {
|
|
config := api.DefaultConfig()
|
|
if environment != constant.EnvProduction {
|
|
config.Scheme = "http"
|
|
} else {
|
|
config.Scheme = "https"
|
|
config.Token = viper.GetString("consul.token")
|
|
config.HttpAuth = &api.HttpBasicAuth{
|
|
Username: viper.GetString("consul.username"),
|
|
Password: viper.GetString("consul.password"),
|
|
}
|
|
}
|
|
|
|
config.Address = viper.GetString("consul.address")
|
|
config.WaitTime = 10 * time.Second
|
|
// Ping the Consul server to verify connectivity
|
|
|
|
client, err := api.NewClient(config)
|
|
if err != nil {
|
|
logger.Fatal("Failed to create consul client", err)
|
|
}
|
|
|
|
_, err = client.Status().Leader()
|
|
if err != nil {
|
|
logger.Fatal("failed to connect to Consul", err)
|
|
}
|
|
|
|
return client
|
|
}
|