dao.go 863 B

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  1. package charge
  2. import (
  3. "context"
  4. "go-common/app/job/main/growup/conf"
  5. "go-common/library/database/sql"
  6. )
  7. // Dao dao
  8. type Dao struct {
  9. c *conf.Config
  10. db *sql.DB
  11. }
  12. // New fn
  13. func New(c *conf.Config) (d *Dao) {
  14. d = &Dao{
  15. c: c,
  16. db: sql.NewMySQL(c.Mysql.Growup),
  17. }
  18. return d
  19. }
  20. // Ping ping health.
  21. func (d *Dao) Ping(c context.Context) (err error) {
  22. return d.db.Ping(c)
  23. }
  24. // Close close connections of mc, redis, db.
  25. func (d *Dao) Close() {
  26. if d.db != nil {
  27. d.db.Close()
  28. }
  29. }
  30. // Exec exec
  31. func (d *Dao) Exec(c context.Context, sql string) (err error) {
  32. _, err = d.db.Exec(c, sql)
  33. return
  34. }
  35. // QueryRow QueryRow
  36. func (d *Dao) QueryRow(c context.Context, sql string) (rows *sql.Row) {
  37. return d.db.QueryRow(c, sql)
  38. }
  39. // Query query
  40. func (d *Dao) Query(c context.Context, sql string) (rows *sql.Rows, err error) {
  41. return d.db.Query(c, sql)
  42. }