openbilibili/app/interface/main/app-resource/dao/sidebar/dao.go

package sidebar

import (
	"context"
	"fmt"
	"time"

	"go-common/app/interface/main/app-resource/conf"
	"go-common/app/interface/main/app-resource/model/sidebar"
	xsql "go-common/library/database/sql"
	"go-common/library/log"
)

const (
	_initSidebarKey = "sidebar_%d_%d"
	_selSideSQL     = `SELECT s.id,s.plat,s.module,s.name,s.logo,s.logo_white,s.param,s.rank,l.build,l.conditions,s.tip FROM 
		sidebar AS s,sidebar_limit AS l WHERE s.state=1 AND s.id=l.s_id AND s.online_time<? ORDER BY s.rank DESC,l.id ASC`
)

type Dao struct {
	db  *xsql.DB
	get *xsql.Stmt
}

func New(c *conf.Config) (d *Dao) {
	d = &Dao{
		db: xsql.NewMySQL(c.MySQL.Show),
	}
	d.get = d.db.Prepared(_selSideSQL)
	return
}

// SideBar
func (d *Dao) SideBar(ctx context.Context, now time.Time) (ss map[string][]*sidebar.SideBar, limits map[int64][]*sidebar.Limit, err error) {
	ss = map[string][]*sidebar.SideBar{}
	limits = map[int64][]*sidebar.Limit{}
	rows, err := d.get.Query(ctx, now)
	if err != nil {
		log.Error("mysqlDB.Query error(%v)", err)
		return
	}
	defer rows.Close()
	for rows.Next() {
		s := &sidebar.SideBar{}
		if err = rows.Scan(&s.ID, &s.Plat, &s.Module, &s.Name, &s.Logo, &s.LogoWhite, &s.Param, &s.Rank, &s.Build, &s.Conditions, &s.Tip); err != nil {
			log.Error("row.Scan error(%v)", err)
			return
		}
		key := fmt.Sprintf(_initSidebarKey, s.Plat, s.Module)
		if _, ok := limits[s.ID]; !ok {
			ss[key] = append(ss[key], s)
		}
		limit := &sidebar.Limit{
			ID:        s.ID,
			Build:     s.Build,
			Condition: s.Conditions,
		}
		limits[s.ID] = append(limits[s.ID], limit)
	}
	return
}

func (d *Dao) Close() {
	if d.db != nil {
		d.db.Close()
	}
}