74 lines
1.8 KiB
Go
74 lines
1.8 KiB
Go
package drawyoo
|
|
|
|
import (
|
|
"context"
|
|
"net/url"
|
|
"strconv"
|
|
|
|
"go-common/app/interface/main/reply/conf"
|
|
"go-common/app/interface/main/reply/model/drawyoo"
|
|
"go-common/library/log"
|
|
httpx "go-common/library/net/http/blademaster"
|
|
"go-common/library/xstr"
|
|
)
|
|
|
|
// Dao Dao
|
|
type Dao struct {
|
|
url string
|
|
http *httpx.Client
|
|
}
|
|
|
|
// New New
|
|
func New(c *conf.Config) *Dao {
|
|
d := &Dao{
|
|
url: "http://h.bilibili.com/api/pushS",
|
|
http: httpx.NewClient(c.DrawyooHTTPClient),
|
|
}
|
|
return d
|
|
}
|
|
|
|
// Info Info
|
|
func (dao *Dao) Info(c context.Context, hid int64) (info *drawyoo.Drawyoo, err error) {
|
|
params := url.Values{}
|
|
params.Set("act", "getHidInfo")
|
|
params.Set("hid", strconv.FormatInt(hid, 10))
|
|
var res struct {
|
|
State int `json:"state"`
|
|
Data []*drawyoo.Drawyoo `json:"data"`
|
|
}
|
|
if err = dao.http.Post(c, dao.url, "", params, &res); err != nil {
|
|
log.Error("drawyoo url(%v),err (%v)", dao.url+"?"+params.Encode(), err)
|
|
return
|
|
}
|
|
if res.State != 200 || len(res.Data) == 0 {
|
|
log.Error("drawyoo url (%v),err (%v)", dao.url+"?"+params.Encode(), err)
|
|
return
|
|
}
|
|
info = res.Data[0]
|
|
return
|
|
}
|
|
|
|
// Infos Infos
|
|
func (dao *Dao) Infos(c context.Context, hids []int64) (info map[int64]interface{}, err error) {
|
|
params := url.Values{}
|
|
params.Set("act", "getHidInfo")
|
|
params.Set("hid", xstr.JoinInts(hids))
|
|
var res struct {
|
|
State int `json:"state"`
|
|
Data []*drawyoo.Drawyoo `json:"data"`
|
|
}
|
|
if err = dao.http.Post(c, dao.url, "", params, &res); err != nil {
|
|
log.Error("drawyoo url(%v),err (%v)", dao.url+"?"+params.Encode(), err)
|
|
return
|
|
}
|
|
if res.State != 200 || len(res.Data) == 0 {
|
|
log.Error("drawyoo url (%v),err (%v)", dao.url+"?"+params.Encode(), err)
|
|
return
|
|
}
|
|
info = make(map[int64]interface{}, len(res.Data))
|
|
for _, r := range res.Data {
|
|
info[r.Hid] = r
|
|
}
|
|
return
|
|
}
|