db_config.go 751 B

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. package config
  2. import (
  3. "eta/eta_mini_ht_api/common/contants"
  4. "sync"
  5. )
  6. var (
  7. once sync.Once
  8. dbConfig *DBConfig
  9. )
  10. type DBOpts struct {
  11. Url string
  12. Driver string
  13. }
  14. type DBConfig struct {
  15. BaseConfig
  16. opts DBOpts
  17. }
  18. func (r *DBConfig) GetUrl() string {
  19. return r.opts.Url
  20. }
  21. func (r *DBConfig) GetDriver() string {
  22. return r.opts.Driver
  23. }
  24. func (r *DBConfig) InitConfig() {
  25. opts := DBOpts{
  26. Url: r.GetString("url"),
  27. Driver: r.GetString("driver"),
  28. }
  29. r.opts = opts
  30. }
  31. func NewDBConfig() Config {
  32. if dbConfig == nil {
  33. once.Do(func() {
  34. dbConfig = &DBConfig{
  35. BaseConfig: BaseConfig{prefix: contants.DATABASE},
  36. opts: DBOpts{},
  37. }
  38. })
  39. }
  40. return dbConfig
  41. }
  42. func init() {
  43. Register(contants.DATABASE, NewDBConfig)
  44. }