openbilibili/app/interface/main/app-feed/dao/converge/dao.go

package converge

import (
	"context"

	"go-common/app/interface/main/app-card/model/card/operate"
	"go-common/app/interface/main/app-feed/conf"
	"go-common/library/database/sql"
)

const (
	_getSQL = "SELECT id,re_type,re_value,title,cover,content FROM content_card"
)

type Dao struct {
	db      *sql.DB
	cardGet *sql.Stmt
}

func New(c *conf.Config) (d *Dao) {
	d = &Dao{
		db: sql.NewMySQL(c.MySQL.Manager),
	}
	// prepare
	d.cardGet = d.db.Prepared(_getSQL)
	return
}

func (d *Dao) Cards(c context.Context) (cm map[int64]*operate.Converge, err error) {
	rows, err := d.cardGet.Query(c)
	if err != nil {
		return
	}
	defer rows.Close()
	cm = map[int64]*operate.Converge{}
	for rows.Next() {
		c := &operate.Converge{}
		if err = rows.Scan(&c.ID, &c.ReType, &c.ReValue, &c.Title, &c.Cover, &c.Content); err != nil {
			return
		}
		c.Change()
		cm[c.ID] = c
	}
	return
}

// Close close memcache resource.
func (dao *Dao) Close() {
	if dao.db != nil {
		dao.db.Close()
	}
}