i_application.go 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. package config
  2. import (
  3. _ "embed"
  4. "fmt"
  5. "gopkg.in/yaml.v2"
  6. "os"
  7. )
  8. type ApplicationYamlStruct struct {
  9. Cname string `yaml:"cname"`
  10. Endpoint string `yaml:"endpoint"`
  11. AccessKeyId string `yaml:"accessKeyId"`
  12. AccessKeySecret string `yaml:"accessKeySecret"`
  13. BucketName string `yaml:"bucketName"`
  14. Cname2 string `yaml:"cname2"`
  15. Endpoint2 string `yaml:"endpoint2"`
  16. AccessKeyId2 string `yaml:"accessKeyId2"`
  17. AccessKeySecret2 string `yaml:"accessKeySecret2"`
  18. BucketName2 string `yaml:"bucketName2"`
  19. CodeDir string `yaml:"codeDir"`
  20. VaildateCmd string `yaml:"vaildateCmd"`
  21. ServicePort string `yaml:"servicePort"`
  22. ServiceName string `yaml:"serviceName"`
  23. LogDir string `yaml:"logDir"`
  24. }
  25. var (
  26. // 使用外部配置文件 go:embed application.yaml
  27. applicationYamlBytes []byte
  28. ApplicationYaml ApplicationYamlStruct
  29. )
  30. //// 使用embed加载
  31. //func InitApplication() {
  32. // _ = yaml.Unmarshal(applicationYamlBytes, &ApplicationYaml)
  33. // fmt.Println("加载配置文件内容为:", ApplicationYaml)
  34. // // 创建镜像下载目录
  35. // util.CreateDir(ApplicationYaml.K8s.AlgorithmTarTempDir)
  36. //}
  37. // 使用外部文件加载
  38. func InitApplication(applicationYamlPath string) {
  39. var err error
  40. applicationYamlBytes, err = os.ReadFile(applicationYamlPath) // 在哪里执行命令,配置文件就放哪里
  41. if err != nil {
  42. fmt.Println("Error reading file:", err)
  43. return
  44. }
  45. _ = yaml.Unmarshal(applicationYamlBytes, &ApplicationYaml)
  46. fmt.Println("加载配置文件内容为:", ApplicationYaml)
  47. }