97 lines
2.3 KiB
Go
97 lines
2.3 KiB
Go
package zlimit
|
|
|
|
import (
|
|
"context"
|
|
|
|
"go-common/library/database/sql"
|
|
"go-common/library/log"
|
|
"go-common/library/xstr"
|
|
)
|
|
|
|
const (
|
|
_getPolicySQL = "SELECT id, play_auth, down_auth, zone_id FROM policy_item WHERE zone_id <> '' AND state=1"
|
|
_getRelationSQL = "SELECT policy_id FROM archive_relation WHERE aid=?"
|
|
_getGolbalPolicySQL = "select group_id,group_concat(id) from policy_item WHERE zone_id <> '' AND state=1 GROUP BY group_id"
|
|
)
|
|
|
|
// policies get policy data from db
|
|
func (s *Service) policies(c context.Context) (res map[int64]map[int64]int64, err error) {
|
|
var (
|
|
tmpres map[int64]int64
|
|
ok bool
|
|
)
|
|
rows, err := s.db.Query(c, _getPolicySQL)
|
|
if err != nil {
|
|
log.Error("db.Query error(%v)", err)
|
|
return
|
|
}
|
|
defer rows.Close()
|
|
res = make(map[int64]map[int64]int64)
|
|
for rows.Next() {
|
|
var (
|
|
pid, playAuth, downAuth int64
|
|
zoneID string
|
|
zoneIDs []int64
|
|
)
|
|
if err = rows.Scan(&pid, &playAuth, &downAuth, &zoneID); err != nil {
|
|
log.Error("rows.Scan error(%v)", err)
|
|
return
|
|
}
|
|
if zoneIDs, err = xstr.SplitInts(zoneID); err != nil {
|
|
log.Error("xstr.SplitInts(%s) error(%v)", zoneID, err)
|
|
continue
|
|
}
|
|
for _, zoneid := range zoneIDs {
|
|
if tmpres, ok = res[pid]; !ok {
|
|
tmpres = make(map[int64]int64)
|
|
res[pid] = tmpres
|
|
}
|
|
resCode := playAuth<<8 | downAuth
|
|
tmpres[zoneid] = resCode
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
// groupPolicies get policy data from db group by group_id
|
|
func (s *Service) groupPolicies(c context.Context) (res map[int64][]int64, err error) {
|
|
rows, err := s.db.Query(c, _getGolbalPolicySQL)
|
|
if err != nil {
|
|
log.Error("db.Query error(%v)", err)
|
|
return
|
|
}
|
|
defer rows.Close()
|
|
res = make(map[int64][]int64)
|
|
for rows.Next() {
|
|
var (
|
|
groupID int64
|
|
pids string
|
|
zoneIDs []int64
|
|
)
|
|
if err = rows.Scan(&groupID, &pids); err != nil {
|
|
log.Error("rows.Scan error(%v)", err)
|
|
return
|
|
}
|
|
if zoneIDs, err = xstr.SplitInts(pids); err != nil {
|
|
log.Error("xstr.SplitInts(%s) error(%v)", pids, err)
|
|
continue
|
|
}
|
|
res[groupID] = zoneIDs
|
|
}
|
|
return
|
|
}
|
|
|
|
// policy get pids from db by aid
|
|
func (s *Service) groupid(c context.Context, aid int64) (gid int64, err error) {
|
|
row := s.getRelationStmt.QueryRow(c, aid)
|
|
if err = row.Scan(&gid); err != nil {
|
|
if err == sql.ErrNoRows {
|
|
gid = 0
|
|
err = nil
|
|
} else {
|
|
log.Error("rows.Scan error(%v)", err)
|
|
}
|
|
}
|
|
return
|
|
}
|