syscfg.go 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137
  1. package model
  2. import (
  3. "database/sql"
  4. "errors"
  5. )
  6. type SysCfg struct {
  7. Model
  8. }
  9. type SysCfgRow struct {
  10. Id int64
  11. CfgKey string
  12. CfgValue string
  13. }
  14. func NewSysCfg(db *sql.DB) *SysCfg {
  15. return &SysCfg{
  16. Model{
  17. db: db,
  18. },
  19. }
  20. }
  21. func (m *SysCfg) Insert(key, value string) (int64, error) {
  22. if key == "" {
  23. return 0, errors.New("KEY不能为空")
  24. }
  25. ret, err := m.db.Exec("insert into syscfg(cfg_key,cfg_value) values(?,?)", key, value)
  26. if err != nil {
  27. return 0, err
  28. } else {
  29. return ret.LastInsertId()
  30. }
  31. }
  32. func (m *SysCfg) Delete(id int64) (int64, error) {
  33. if id <= 0 {
  34. return 0, errors.New("ID必须是正整数")
  35. }
  36. ret, err := m.db.Exec("delete from syscfg where id=?", id)
  37. if err != nil {
  38. return 0, err
  39. } else {
  40. return ret.RowsAffected()
  41. }
  42. }
  43. func (m *SysCfg) DeleteByKey(key string) (int64, error) {
  44. if key == "" {
  45. return 0, errors.New("KEY不能为空")
  46. }
  47. ret, err := m.db.Exec("delete from syscfg where cfg_key=?", key)
  48. if err != nil {
  49. return 0, err
  50. } else {
  51. return ret.RowsAffected()
  52. }
  53. }
  54. func (m *SysCfg) Update(id int64, value string) (int64, error) {
  55. if id <= 0 {
  56. return 0, errors.New("ID必须是正整数")
  57. }
  58. ret, err := m.db.Exec("update syscfg set cfg_value=? where id=?", value, id)
  59. if err != nil {
  60. return 0, err
  61. } else {
  62. return ret.RowsAffected()
  63. }
  64. }
  65. func (m *SysCfg) UpdateByKey(key, value string) (int64, error) {
  66. if key == "" {
  67. return 0, errors.New("KEY不能为空")
  68. }
  69. ret, err := m.db.Exec("update syscfg set cfg_value=? where cfg_key=?", value, key)
  70. if err != nil {
  71. return 0, err
  72. } else {
  73. return ret.RowsAffected()
  74. }
  75. }
  76. func (m *SysCfg) Get(id int64) (*SysCfgRow, error) {
  77. if id <= 0 {
  78. return nil, errors.New("ID必须是正整数")
  79. }
  80. result := m.db.QueryRow("select id,cfg_key,cfg_value from syscfg where id=?", id)
  81. row := SysCfgRow{}
  82. err := result.Scan(&row.Id, &row.CfgKey, &row.CfgValue)
  83. if err != nil {
  84. if err == sql.ErrNoRows {
  85. return nil, errors.New("记录不存在")
  86. } else {
  87. return nil, err
  88. }
  89. } else {
  90. return &row, nil
  91. }
  92. }
  93. func (m *SysCfg) GetByKey(key string) (*SysCfgRow, error) {
  94. if key == "" {
  95. return nil, errors.New("KEY不能为空")
  96. }
  97. result := m.db.QueryRow("select id,cfg_key,cfg_value from syscfg where cfg_key=?", key)
  98. row := SysCfgRow{}
  99. err := result.Scan(&row.Id, &row.CfgKey, &row.CfgValue)
  100. if err != nil {
  101. if err == sql.ErrNoRows {
  102. return nil, errors.New("记录不存在")
  103. } else {
  104. return nil, err
  105. }
  106. } else {
  107. return &row, nil
  108. }
  109. }
  110. func (m *SysCfg) GetAll() ([]*SysCfgRow, error) {
  111. result, err := m.db.Query("select id,cfg_key,cfg_value from syscfg order by id")
  112. if err != nil {
  113. return nil, err
  114. }
  115. rows := make([]*SysCfgRow, 0)
  116. for result.Next() {
  117. row := SysCfgRow{}
  118. err = result.Scan(&row.Id, &row.CfgKey, &row.CfgValue)
  119. if err != nil {
  120. continue
  121. }
  122. rows = append(rows, &row)
  123. }
  124. return rows, nil
  125. }