-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.go
55 lines (46 loc) · 1.41 KB
/
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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
package cfg
import (
"github.com/arimanius/digivision-backend/internal/bootstrap"
validation "github.com/go-ozzo/ozzo-validation/v4"
"github.com/milvus-io/milvus-sdk-go/v2/entity"
)
type Config struct {
Env string
Log struct {
Level string
}
bootstrap.GrpcServerRunnerConfig `mapstructure:",squash" yaml:",inline"`
HttpServer struct {
Port int
}
Img2Vec struct {
Addr string
}
Milvus struct {
Addr string
VectorDim int
MetricType entity.MetricType
NProbe int
CollectionName string
}
ObjectDetector struct {
Addr string
}
Redis struct {
Addr string
}
}
func (c *Config) Validate() error {
return validation.Errors{
"env": validation.Validate(c.Env, validation.Required),
"log.level": validation.Validate(c.Log.Level, validation.Required, validation.In(
"panic", "fatal", "error", "warn", "info", "debug", "trace",
)),
"server.host": validation.Validate(c.Server.Host, validation.Required),
"server.port": validation.Validate(c.Server.Port, validation.Required),
"milvus.vectorDim": validation.Validate(c.Milvus.VectorDim, validation.Required),
"milvus.metricType": validation.Validate(c.Milvus.MetricType, validation.Required),
"milvus.nProbe": validation.Validate(c.Milvus.NProbe, validation.Required),
"milvus.collectionName": validation.Validate(c.Milvus.CollectionName, validation.Required),
}.Filter()
}