RuleConfig.go 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768
  1. // this file is auto create by program, don't edit manually
  2. package data
  3. import (
  4. mhayaError "github.com/mhaya/error"
  5. mhayaLogger "github.com/mhaya/logger"
  6. )
  7. type ruleConfig struct {
  8. maps map[int]*RuleConfigRow
  9. }
  10. type RuleConfigRow struct {
  11. ID int // ID
  12. Enable int // 是否开启规则 0 否 1是
  13. Num int // 注册人数
  14. MaxTime int // 过期时间(按天)
  15. }
  16. func (p *ruleConfig) Name() string {
  17. return "RuleConfig"
  18. }
  19. func (p *ruleConfig) Init() {
  20. p.maps = make(map[int]*RuleConfigRow)
  21. }
  22. func (p *ruleConfig) OnLoad(maps interface{}, _ bool) (int, error) {
  23. list, ok := maps.([]interface{})
  24. if !ok {
  25. return 0, mhayaError.Error("maps convert to []interface{} error.")
  26. }
  27. loadMaps := make(map[int]*RuleConfigRow)
  28. for index, data := range list {
  29. loadConfig := &RuleConfigRow{}
  30. err := DecodeData(data, loadConfig)
  31. if err != nil {
  32. mhayaLogger.Warnf("decode error. [row = %d, %v], err = %s", index+1, loadConfig, err)
  33. continue
  34. }
  35. loadMaps[loadConfig.ID] = loadConfig
  36. }
  37. p.maps = loadMaps
  38. return len(list), nil
  39. }
  40. func (p *ruleConfig) OnAfterLoad(_ bool) {
  41. }
  42. func (p *ruleConfig) Get(pk int) (*RuleConfigRow, bool) {
  43. i, found := p.maps[pk]
  44. if found {
  45. if i.Enable == 0 {
  46. return nil, false
  47. }
  48. }
  49. return i, found
  50. }
  51. func (p *ruleConfig) Contain(pk int) bool {
  52. _, found := p.Get(pk)
  53. return found
  54. }