i_application.go 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  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. Cname3 string `yaml:"cname3"`
  20. Endpoint3 string `yaml:"endpoint3"`
  21. AccessKeyId3 string `yaml:"accessKeyId3"`
  22. AccessKeySecret3 string `yaml:"accessKeySecret3"`
  23. BucketName3 string `yaml:"bucketName3"`
  24. CodeDir string `yaml:"codeDir"`
  25. VaildateCmd string `yaml:"vaildateCmd"`
  26. ServicePort string `yaml:"servicePort"`
  27. ServiceName string `yaml:"serviceName"`
  28. LogDir string `yaml:"logDir"`
  29. }
  30. var (
  31. // 使用外部配置文件 go:embed application.yaml
  32. applicationYamlBytes []byte
  33. ApplicationYaml ApplicationYamlStruct
  34. )
  35. //// 使用embed加载
  36. //func InitApplication() {
  37. // _ = yaml.Unmarshal(applicationYamlBytes, &ApplicationYaml)
  38. // fmt.Println("加载配置文件内容为:", ApplicationYaml)
  39. // // 创建镜像下载目录
  40. // util.CreateDir(ApplicationYaml.K8s.AlgorithmTarTempDir)
  41. //}
  42. // 使用外部文件加载
  43. func InitApplication(applicationYamlPath string) {
  44. var err error
  45. applicationYamlBytes, err = os.ReadFile(applicationYamlPath) // 在哪里执行命令,配置文件就放哪里
  46. if err != nil {
  47. fmt.Println("Error reading file:", err)
  48. return
  49. }
  50. _ = yaml.Unmarshal(applicationYamlBytes, &ApplicationYaml)
  51. fmt.Println("加载配置文件内容为:", ApplicationYaml)
  52. }