db_config.go 676 B

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  1. package config
  2. import (
  3. "eta_mini_ht_api/common/contants"
  4. "sync"
  5. )
  6. var (
  7. once sync.Once
  8. config *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) GetConfig() interface{} {
  19. return r.opts
  20. }
  21. func (r *DBConfig) InitConfig() {
  22. opts := DBOpts{
  23. Url: r.GetString("url"),
  24. Driver: r.GetString("driver"),
  25. }
  26. r.opts = opts
  27. }
  28. func NewDBConfig() Config {
  29. if config == nil {
  30. once.Do(func() {
  31. config = &DBConfig{
  32. BaseConfig: BaseConfig{prefix: contants.DATABASE},
  33. opts: DBOpts{},
  34. }
  35. })
  36. }
  37. return config
  38. }
  39. func init() {
  40. Register(contants.DATABASE, NewDBConfig)
  41. }