go-common/app/job/main/tv/dao/report/style.go

70 lines
1.6 KiB
Go
Raw Normal View History

2019-04-22 10:49:16 +00:00
package report
import (
"context"
mdlpgc "go-common/app/job/main/tv/model/pgc"
"go-common/library/database/sql"
"go-common/library/log"
)
const (
_styleSQL = "SELECT id,style,category FROM tv_ep_season WHERE is_deleted=0 AND `check`=1 AND valid=1"
_labelSQL = `SELECT name,value,category FROM tv_label WHERE deleted=0 AND param="style_id" AND valid=1`
)
// FindStyle style all .
func (d *Dao) FindStyle(ctx context.Context) (res []*mdlpgc.StyleRes, err error) {
var (
rows *sql.Rows
)
if rows, err = d.DB.Query(ctx, _styleSQL); err != nil {
log.Error("d.DB.Query sql(%s) error(%v)", _styleSQL, err)
return
}
defer rows.Close()
for rows.Next() {
r := &mdlpgc.StyleRes{}
if err = rows.Scan(&r.ID, &r.Style, &r.Category); err != nil {
log.Error("d.DB.QueryRow error(%v)", err)
return
}
res = append(res, r)
}
if err = rows.Err(); err != nil {
log.Error("rows.Err() error(%v)", err)
}
return
}
// FindLabelID label all .
func (d *Dao) FindLabelID(ctx context.Context) (res map[int]map[string]int, err error) {
var (
rows *sql.Rows
m map[string]int
)
res = make(map[int]map[string]int)
if rows, err = d.DB.Query(ctx, _labelSQL); err != nil {
log.Error("d.DB.Query error(%v)", err)
return
}
defer rows.Close()
for rows.Next() {
r := &mdlpgc.LabelRes{}
if err = rows.Scan(&r.Name, &r.Value, &r.Category); err != nil {
log.Error("d.DB.Query Scan error(%v)", err)
return
}
if _, ok := res[r.Category]; ok {
m[r.Name] = r.Value
} else {
m = make(map[string]int)
}
res[r.Category] = m
}
if err = rows.Err(); err != nil {
log.Error("rows.Err() error(%v)", err)
}
return
}