mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-12 19:14:23 +00:00
68 lines
1.6 KiB
Go
68 lines
1.6 KiB
Go
package config
|
|
|
|
import (
|
|
"encoding/json"
|
|
|
|
"github.com/caos/zitadel/internal/errors"
|
|
"github.com/caos/zitadel/internal/tracing"
|
|
tracing_g "github.com/caos/zitadel/internal/tracing/google"
|
|
tracing_log "github.com/caos/zitadel/internal/tracing/log"
|
|
)
|
|
|
|
type TracingConfig struct {
|
|
Type string
|
|
Config tracing.Config
|
|
}
|
|
|
|
var tracer = map[string]func() tracing.Config{
|
|
"google": func() tracing.Config { return &tracing_g.Config{} },
|
|
"log": func() tracing.Config { return &tracing_log.Config{} },
|
|
"none": func() tracing.Config { return &NoTracing{} },
|
|
"": func() tracing.Config { return &NoTracing{} },
|
|
}
|
|
|
|
func (c *TracingConfig) UnmarshalJSON(data []byte) error {
|
|
var rc struct {
|
|
Type string
|
|
Config json.RawMessage
|
|
}
|
|
|
|
if err := json.Unmarshal(data, &rc); err != nil {
|
|
return errors.ThrowInternal(err, "TRACE-vmjS", "error parsing config")
|
|
}
|
|
|
|
c.Type = rc.Type
|
|
|
|
var err error
|
|
c.Config, err = newTracingConfig(c.Type, rc.Config)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return c.Config.NewTracer()
|
|
}
|
|
|
|
func newTracingConfig(tracerType string, configData []byte) (tracing.Config, error) {
|
|
t, ok := tracer[tracerType]
|
|
if !ok {
|
|
return nil, errors.ThrowInternalf(nil, "TRACE-HMEJ", "config type %s not supported", tracerType)
|
|
}
|
|
|
|
tracingConfig := t()
|
|
if len(configData) == 0 {
|
|
return tracingConfig, nil
|
|
}
|
|
|
|
if err := json.Unmarshal(configData, tracingConfig); err != nil {
|
|
return nil, errors.ThrowInternal(err, "TRACE-1tSS", "Could not read config: %v")
|
|
}
|
|
|
|
return tracingConfig, nil
|
|
}
|
|
|
|
type NoTracing struct{}
|
|
|
|
func (_ *NoTracing) NewTracer() error {
|
|
return nil
|
|
}
|