tservicemodel_gen.go 3.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495
  1. // Code generated by goctl. DO NOT EDIT.
  2. // versions:
  3. // goctl version: 1.10.2
  4. package model
  5. import (
  6. "context"
  7. "database/sql"
  8. "fmt"
  9. "strings"
  10. "time"
  11. "github.com/zeromicro/go-zero/core/stores/builder"
  12. "github.com/zeromicro/go-zero/core/stores/sqlx"
  13. "github.com/zeromicro/go-zero/core/stringx"
  14. )
  15. var (
  16. tServiceFieldNames = builder.RawFieldNames(&TService{})
  17. tServiceRows = strings.Join(tServiceFieldNames, ",")
  18. tServiceRowsExpectAutoSet = strings.Join(stringx.Remove(tServiceFieldNames, "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), ",")
  19. tServiceRowsWithPlaceHolder = strings.Join(stringx.Remove(tServiceFieldNames, "`id`", "`create_at`", "`create_time`", "`created_at`", "`update_at`", "`update_time`", "`updated_at`"), "=?,") + "=?"
  20. )
  21. type (
  22. tServiceModel interface {
  23. Insert(ctx context.Context, data *TService) (sql.Result, error)
  24. FindOne(ctx context.Context, id int64) (*TService, error)
  25. Update(ctx context.Context, data *TService) error
  26. Delete(ctx context.Context, id int64) error
  27. }
  28. defaultTServiceModel struct {
  29. conn sqlx.SqlConn
  30. table string
  31. }
  32. TService struct {
  33. Id int64 `db:"id"`
  34. Name sql.NullString `db:"name"` // 服务名称
  35. Ip sql.NullString `db:"ip"` // 外挂服务ip
  36. Port int64 `db:"port"` // 外挂服务端口(DB 列已改 int,int64 直传;NOT NULL DEFAULT 0,0=未配置)
  37. Iden sql.NullString `db:"iden"` // 唯一标识
  38. KeyIden sql.NullString `db:"key_iden"` // 密钥标识
  39. KeyAlgo sql.NullString `db:"key_algo"` // 密钥算法
  40. Status sql.NullString `db:"status"` // 服务状态
  41. MountStatus sql.NullString `db:"mount_status"` // 挂载状态:mounted=已挂载,unmounted=未挂载,failed=挂载失败
  42. Brush int64 `db:"brush"` // 一键刷库:0=否(默认),1=创建后自动下发刷库
  43. CreateAt time.Time `db:"create_at"` // 创建时间
  44. }
  45. )
  46. func newTServiceModel(conn sqlx.SqlConn) *defaultTServiceModel {
  47. return &defaultTServiceModel{
  48. conn: conn,
  49. table: "`t_service`",
  50. }
  51. }
  52. func (m *defaultTServiceModel) Delete(ctx context.Context, id int64) error {
  53. query := fmt.Sprintf("delete from %s where `id` = ?", m.table)
  54. _, err := m.conn.ExecCtx(ctx, query, id)
  55. return err
  56. }
  57. func (m *defaultTServiceModel) FindOne(ctx context.Context, id int64) (*TService, error) {
  58. query := fmt.Sprintf("select %s from %s where `id` = ? limit 1", tServiceRows, m.table)
  59. var resp TService
  60. err := m.conn.QueryRowCtx(ctx, &resp, query, id)
  61. switch err {
  62. case nil:
  63. return &resp, nil
  64. case sqlx.ErrNotFound:
  65. return nil, ErrNotFound
  66. default:
  67. return nil, err
  68. }
  69. }
  70. func (m *defaultTServiceModel) Insert(ctx context.Context, data *TService) (sql.Result, error) {
  71. query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, tServiceRowsExpectAutoSet)
  72. ret, err := m.conn.ExecCtx(ctx, query, data.Id, data.Name, data.Ip, data.Port, data.Iden, data.KeyIden, data.KeyAlgo, data.Status, data.MountStatus, data.Brush)
  73. return ret, err
  74. }
  75. func (m *defaultTServiceModel) Update(ctx context.Context, data *TService) error {
  76. query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, tServiceRowsWithPlaceHolder)
  77. _, err := m.conn.ExecCtx(ctx, query, data.Name, data.Ip, data.Port, data.Iden, data.KeyIden, data.KeyAlgo, data.Status, data.MountStatus, data.Brush, data.Id)
  78. return err
  79. }
  80. func (m *defaultTServiceModel) tableName() string {
  81. return m.table
  82. }