query-orchestration/internal/server/server.go

package server

import (
	"context"

	"queryorchestration/internal/serviceconfig/observability"

	"queryorchestration/internal/database/migrations"
	"queryorchestration/internal/serviceconfig"

	"github.com/go-playground/validator/v10"
)

const DEFAULT_SECRET_PREFIX = "secret"

type Config interface {
	serviceconfig.ConfigProvider
	SetValidator()
	GetValidator() *validator.Validate
}

type BaseConfig struct {
	serviceconfig.BaseConfig
	Validator *validator.Validate
}

func (c *BaseConfig) SetValidator() {
	valid := validator.New()

	c.Validator = valid
}
func (c *BaseConfig) GetValidator() *validator.Validate {
	return c.Validator
}

func New(ctx context.Context, cfg Config) (func() error, error) {
	// init the config in case it has not already been called.
	errInitializingConfig := serviceconfig.InitializeConfig(cfg)
	if errInitializingConfig != nil {
		return func() error { return nil }, errInitializingConfig
	}

	cfg.PrintConfig(DEFAULT_SECRET_PREFIX)
	closeTracer := cfg.SetOtel(ctx)

	version := observability.GetVersion()
	cfg.GetLogger().Info("Starting", "version", version)

	err := migrations.Run(ctx, cfg)
	if err != nil {
		return func() error { return nil }, err
	}

	cfg.SetValidator()

	err = cfg.SetDBPool(ctx)
	if err != nil {
		return func() error { return nil }, err
	}

	err = cfg.SetQueueClient(ctx)
	if err != nil {
		return nil, err
	}

	return closeTracer, nil
}