84 lines
2.9 KiB
Go
84 lines
2.9 KiB
Go
package dao
|
||
|
||
import (
|
||
"bytes"
|
||
"context"
|
||
"fmt"
|
||
"mime/multipart"
|
||
"net/http"
|
||
"strconv"
|
||
"strings"
|
||
"time"
|
||
|
||
"go-common/app/interface/main/push-archive/model"
|
||
"go-common/library/log"
|
||
"go-common/library/xstr"
|
||
)
|
||
|
||
type _response struct {
|
||
Code int `json:"code"`
|
||
Data int `json:"data"`
|
||
}
|
||
|
||
// NoticeFans pushs the notification to fans.
|
||
func (d *Dao) NoticeFans(fans *[]int64, params map[string]interface{}) (err error) {
|
||
arc := params["archive"].(*model.Archive)
|
||
group := strings.TrimSpace(params["group"].(string))
|
||
msgTemplate := params["msgTemplate"].(string)
|
||
uuid := params["uuid"].(string)
|
||
relationType := params["relationType"].(int)
|
||
author := "UP主"
|
||
if arc.Author != "" {
|
||
author = fmt.Sprintf(`“%s”`, arc.Author)
|
||
}
|
||
// 普通关注和特殊关注用不同的业务组推
|
||
businessID := d.c.Push.BusinessID
|
||
businessToken := d.c.Push.BusinessToken
|
||
if relationType == model.RelationSpecial {
|
||
businessID = d.c.Push.BusinessSpecialID
|
||
businessToken = d.c.Push.BusinessSpecialToken
|
||
}
|
||
msg := fmt.Sprintf(msgTemplate, author, arc.Title)
|
||
sp := strings.SplitN(msg, "\r\n", 2)
|
||
buf := new(bytes.Buffer)
|
||
w := multipart.NewWriter(buf)
|
||
w.WriteField("group", group) // 实验组名，值为实验组数据表名
|
||
w.WriteField("app_id", "1") // 1表示哔哩哔哩动画
|
||
w.WriteField("business_id", strconv.Itoa(businessID))
|
||
w.WriteField("alert_title", sp[0])
|
||
w.WriteField("alert_body", sp[1])
|
||
w.WriteField("mids", xstr.JoinInts(*fans))
|
||
w.WriteField("link_type", "2") // 2代表视频稿件播放页
|
||
w.WriteField("link_value", strconv.FormatInt(arc.ID, 10))
|
||
w.WriteField("uuid", uuid)
|
||
// 1、v5.20.0 后客户端才接特殊关注 2、 iPad版本没更新，不推
|
||
w.WriteField("builds", `{"2":{"Build":6500,"Condition":"gte"}, "3":{"Build":0,"Condition":"lt"}, "1":{"Build":519010,"Condition":"gte"}}`)
|
||
w.Close()
|
||
query := map[string]string{
|
||
"ts": strconv.FormatInt(time.Now().Unix(), 10),
|
||
"appkey": d.c.HTTPClient.Key,
|
||
}
|
||
query["sign"] = d.signature(query, d.c.HTTPClient.Secret)
|
||
url := fmt.Sprintf("%s?ts=%s&appkey=%s&sign=%s", d.c.Push.AddAPI, query["ts"], query["appkey"], query["sign"])
|
||
req, err := http.NewRequest(http.MethodPost, url, buf)
|
||
if err != nil {
|
||
log.Error("http.NewRequest(%s) error(%v) uuid(%s)", url, err, uuid)
|
||
PromError("http:NewRequest")
|
||
return
|
||
}
|
||
req.Header.Set("Content-Type", w.FormDataContentType())
|
||
req.Header.Set("Authorization", fmt.Sprintf("token=%s", businessToken))
|
||
res := &_response{}
|
||
if err = d.httpClient.Do(context.TODO(), req, &res); err != nil {
|
||
log.Error("httpClient.Do() error(%v)", err)
|
||
PromError("http:Do")
|
||
return
|
||
}
|
||
if res.Code != 0 || res.Data == 0 {
|
||
log.Error("push failed archive(%d) upper(%d) fans_total(%d) group(%s) response(%+v)", arc.ID, arc.Mid, len(*fans), group, res)
|
||
} else {
|
||
log.Info("push success archive(%d) upper(%d) fans_total(%d) group(%s) response(%+v)", arc.ID, arc.Mid, len(*fans), group, res)
|
||
}
|
||
return
|
||
}
|