-
Notifications
You must be signed in to change notification settings - Fork 1
/
config.go
32 lines (27 loc) · 977 Bytes
/
config.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
package gospice
import "os"
type ClientConfig struct {
HttpUrl string `json:"http_url,omitempty"`
FlightUrl string `json:"flight_url,omitempty"`
FirecacheUrl string `json:"firecache_url,omitempty"`
}
func LoadConfig() ClientConfig {
return ClientConfig{
HttpUrl: getEnvOrDefault("SPICE_HTTP_URL", "https://data.spiceai.io"),
FirecacheUrl: getEnvOrDefault("SPICE_FIRECACHE_URL", "firecache.spiceai.io:443"),
FlightUrl: getEnvOrDefault("SPICE_FLIGHT_URL", "flight.spiceai.io:443"),
}
}
func LoadLocalConfig() ClientConfig {
return ClientConfig{
HttpUrl: getEnvOrDefault("SPICE_LOCAL_HTTP_URL", "http://localhost:3000"),
FlightUrl: getEnvOrDefault("SPICE_LOCAL_FLIGHT_URL", "grpc://localhost:50051"),
FirecacheUrl: getEnvOrDefault("SPICE_FIRECACHE_URL", "firecache.spiceai.io:443"),
}
}
func getEnvOrDefault(key string, defaultValue string) string {
if v, exists := os.LookupEnv(key); exists {
return v
}
return defaultValue
}