package services import ( "encoding/json" "errors" "fmt" "hongze/hongze_mfyx/models" "hongze/hongze_mfyx/utils" "io/ioutil" "net/http" "strconv" "strings" ) var ( ServerUrl = "http://127.0.0.1:5008/" Cygx_activity_sigin_html = "cygx_activity_sigin_html" Cygx_mp3_html = "cygx_mp3_html" Cygx_mp4_html = "cygx_mp4_html" ) type Html2ImgResp struct { Code int `json:"code"` Msg string `json:"msg"` Data string `json:"data"` } // postHtml2Img 请求htm2img接口 func postHtml2Img(param map[string]interface{}) (resp *Html2ImgResp, err error) { // 目前仅此处调用该接口,暂不加授权、校验等 postUrl := ServerUrl + "htm2img" postData, err := json.Marshal(param) if err != nil { return } result, err := Html2ImgHttpPost(postUrl, string(postData), "application/json") if err != nil { return } if err = json.Unmarshal(result, &resp); err != nil { return } return resp, nil } // Html2ImgHttpPost post请求 func Html2ImgHttpPost(url, postData string, params ...string) ([]byte, error) { body := ioutil.NopCloser(strings.NewReader(postData)) client := &http.Client{} req, err := http.NewRequest("POST", url, body) if err != nil { return nil, err } contentType := "application/x-www-form-urlencoded;charset=utf-8" if len(params) > 0 && params[0] != "" { contentType = params[0] } req.Header.Set("Content-Type", contentType) resp, err := client.Do(req) if err != nil { return nil, err } defer resp.Body.Close() b, err := ioutil.ReadAll(resp.Body) fmt.Println("HttpPost:" + string(b)) return b, err } //func init() { // GetYanxuanSpecialAuthoMomentsImg(53559) //} // 生成研选专栏分享到朋友圈的图片 func GetYanxuanSpecialAuthoMomentsImg(userId int) (imgUrl string) { var err error //time.Sleep(3*time.Second) // 有时候同时添加多个活动,延迟三秒 defer func() { if err != nil { fmt.Println("err:", err) go utils.SendAlarmMsg("生成研选专栏分享到朋友圈的图片,MakeArticleMomentsImg Err:"+err.Error()+"用户ID"+strconv.Itoa(userId), 3) } }() articleInfo, e := models.GetYanxuanSpecialAuthor(userId, 0, "") if e != nil { err = errors.New("GetYanxuanSpecialAuthor, Err: " + e.Error()) return } configCode := "special_author_moments_img_html" detailConfig, e := models.GetConfigByCode(configCode) if e != nil { err = errors.New("GetCygxConfigDetailByCode 获取生成研选专栏分享到朋友圈的图片格式信息失败, Err: " + e.Error()) return } configValue := detailConfig.ConfigValue configValue = strings.Replace(configValue, "{{HeadImg}}", articleInfo.HeadImg, -1) configValue = strings.Replace(configValue, "{{SpecialName}}", articleInfo.SpecialName, -1) configValue = strings.Replace(configValue, "{{NickName}}", articleInfo.NickName, -1) configValue = strings.Replace(configValue, "{{SpecialArticleNum}}", strconv.Itoa(articleInfo.SpecialArticleNum), -1) configValue = strings.Replace(configValue, "{{CollectNum}}", strconv.Itoa(articleInfo.CollectNum), -1) configValue = strings.Replace(configValue, "{{FollowNum}}", strconv.Itoa(articleInfo.FollowNum), -1) configValue = strings.Replace(configValue, "{{Introduction}}", articleInfo.Introduction, -1) htm2ImgReq := make(map[string]interface{}) htm2ImgReq["html_content"] = configValue htm2ImgReq["width"] = 2250 htm2ImgReq["height"] = 3813 res, err := postHtml2Img(htm2ImgReq) if err != nil || res == nil { err = errors.New("html转图片失败: " + res.Msg) return } if res.Code != 200 { err = errors.New("html转图片失败: " + res.Msg) return } imgUrl = res.Data fmt.Println(imgUrl) return } // 生成研选专栏分享到朋友圈的图片 func GetYanxuanSpecialAuthoListMomentsImg() (imgUrl string) { var err error //time.Sleep(3*time.Second) // 有时候同时添加多个活动,延迟三秒 defer func() { if err != nil { fmt.Println("err:", err) go utils.SendAlarmMsg("生成研选专栏分享到朋友圈的图片,MakeArticleMomentsImg Err:"+err.Error(), 3) } }() var condition string var pars []interface{} condition += ` AND a.nick_name <> '' ` condition += ` ORDER BY latest_publish_time DESC` list, e := models.GetYanxuanSpecialAuthorList(condition, pars, 0, 1) if e != nil { err = errors.New("GetYanxuanSpecialAuthor, Err: " + e.Error()) return } var SpecialName, HeadImg, NickName, Introduction, PublishTime, Title string var userIds []int for _, v := range list { SpecialName = v.SpecialName HeadImg = v.HeadImg NickName = v.NickName Introduction = v.Introduction userIds = append(userIds, v.UserId) } bestNew := GetBestNewYanxuanSpecialByUserId(userIds) for _, v := range list { if bestNew[v.UserId] != nil { PublishTime = bestNew[v.UserId].PublishTime Title = bestNew[v.UserId].Title } } configCode := "special_author_list_moments_img_html" detailConfig, e := models.GetConfigByCode(configCode) if e != nil { err = errors.New("GetCygxConfigDetailByCode 获取生成研选专栏分享到朋友圈的图片格式信息失败, Err: " + e.Error()) return } configValue := detailConfig.ConfigValue configValue = strings.Replace(configValue, "{{SpecialName}}", SpecialName, -1) configValue = strings.Replace(configValue, "{{HeadImg}}", HeadImg, -1) configValue = strings.Replace(configValue, "{{NickName}}", NickName, -1) configValue = strings.Replace(configValue, "{{Introduction}}", Introduction, -1) configValue = strings.Replace(configValue, "{{PublishTime}}", PublishTime, -1) configValue = strings.Replace(configValue, "{{Title}}", Title, -1) htm2ImgReq := make(map[string]interface{}) htm2ImgReq["html_content"] = configValue htm2ImgReq["width"] = 2250 htm2ImgReq["height"] = 3813 res, err := postHtml2Img(htm2ImgReq) if err != nil || res == nil { err = errors.New("html转图片失败: " + res.Msg) return } if res.Code != 200 { err = errors.New("html转图片失败: " + res.Msg) return } imgUrl = res.Data return }