kratos-utils/config/source.go

package config

import (
	"fmt"
	"github.com/go-kratos/kratos/v2/config"
	"github.com/go-kratos/kratos/v2/config/file"
)

type Source interface {
	NewSource() (config.Source, error)
	Validate() bool
	String() string
}

type Format string

const (
	Yaml Format = "yaml"
)

var formatMap = map[Format]string{
	Yaml: "yaml",
}

func (f Format) String() string {
	return string(f)
}

func (f Format) Validate() bool {
	_, ok := formatMap[f]
	return ok
}

// =======================================

type FileSource struct {
	Format Format
	Path   string
}

func (s *FileSource) NewSource() (config.Source, error) {
	source := file.NewSource(s.Path)
	return source, nil
}

func (s *FileSource) Validate() bool {
	if !s.Format.Validate() {
		return false
	}
	if s.Path == "" {
		return false
	}
	return true
}

func (s *FileSource) String() string {
	return fmt.Sprintf("file source format:%v, path:%v", s.Format, s.Path)
}