|
@@ -0,0 +1,184 @@
|
|
|
|
+package services
|
|
|
|
+
|
|
|
|
+import (
|
|
|
|
+ "encoding/json"
|
|
|
|
+ "errors"
|
|
|
|
+ "fmt"
|
|
|
|
+ "hongze/hongze_yb/global"
|
|
|
|
+ "hongze/hongze_yb/utils"
|
|
|
|
+ "io/ioutil"
|
|
|
|
+ "net/http"
|
|
|
|
+ "os"
|
|
|
|
+ "strings"
|
|
|
|
+)
|
|
|
|
+
|
|
|
|
+// SharePosterReq 分享海报请求体
|
|
|
|
+type SharePosterReq struct {
|
|
|
|
+ Source string `json:"source" description:"来源"`
|
|
|
|
+ Pars string `json:"pars" description:"海报动态信息"`
|
|
|
|
+ Scene string `json:"scene" description:"太阳码scene"`
|
|
|
|
+ Path string `json:"path" description:"太阳码path"`
|
|
|
|
+}
|
|
|
|
+
|
|
|
|
+// CreatePosterFromSource 生成分享海报
|
|
|
|
+func CreatePosterFromSource(source string, pars string) (imgUrl string, err error) {
|
|
|
|
+ var errMsg string
|
|
|
|
+ defer func() {
|
|
|
|
+ if err != nil {
|
|
|
|
+ if global.CONFIG.Serve.RunMode == "release" {
|
|
|
|
+ go SendEmail(utils.APPNAME+"【"+global.CONFIG.Serve.RunMode+"】"+"失败提醒", "生成分享海报失败, "+errMsg+", CreatePosterFromSource: "+err.Error(), utils.EmailSendToUsers)
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+ }()
|
|
|
|
+
|
|
|
|
+ // 根据请求路径及版本号获取本地图片, 无对应图片则生成图片
|
|
|
|
+
|
|
|
|
+ // 根据来源生成相应的html文件, 确定生成图片的长宽
|
|
|
|
+ resourceUrl := ""
|
|
|
|
+ filePath := ""
|
|
|
|
+ switch source {
|
|
|
|
+ case "chart_detail":
|
|
|
|
+ filePath, err = CreateChartDetailHtml(pars)
|
|
|
|
+ if err != nil {
|
|
|
|
+ return
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+ if filePath == "" {
|
|
|
|
+ errMsg = "html资源生成失败"
|
|
|
|
+ err = errors.New(errMsg)
|
|
|
|
+ return
|
|
|
|
+ }
|
|
|
|
+ resourceUrl = "http://127.0.0.1:8612/" + filePath
|
|
|
|
+ fmt.Println(resourceUrl)
|
|
|
|
+
|
|
|
|
+ // 请求python服务htm2img
|
|
|
|
+ htm2ImgReq := make(map[string]interface{})
|
|
|
|
+ htm2ImgReq["html_url"] = resourceUrl
|
|
|
|
+ res, err := postHtml2Img(htm2ImgReq)
|
|
|
|
+ if err != nil {
|
|
|
|
+ return
|
|
|
|
+ }
|
|
|
|
+ if res == nil {
|
|
|
|
+ err = errors.New("html转图片请求失败, 未知错误")
|
|
|
|
+ return
|
|
|
|
+ }
|
|
|
|
+ if res.Code != 200 {
|
|
|
|
+ err = errors.New("html转图片失败: " + res.Msg)
|
|
|
|
+ return
|
|
|
|
+ }
|
|
|
|
+ imgUrl = res.Data
|
|
|
|
+ fmt.Println(imgUrl)
|
|
|
|
+ // 清除本地html文件
|
|
|
|
+
|
|
|
|
+ // 记录请求路径对应的图片信息
|
|
|
|
+
|
|
|
|
+ return
|
|
|
|
+}
|
|
|
|
+
|
|
|
|
+type ChartDetailPosterReq struct {
|
|
|
|
+ ChartInfoId int `json:"chart_info_id"`
|
|
|
|
+ ChartName string `json:"chart_name"`
|
|
|
|
+ ChartImage string `json:"chart_image"`
|
|
|
|
+}
|
|
|
|
+
|
|
|
|
+// 生成html-图表详情页
|
|
|
|
+func CreateChartDetailHtml(pars string) (filePath string, err error) {
|
|
|
|
+ // 解析html填充的参数
|
|
|
|
+ params := new(ChartDetailPosterReq)
|
|
|
|
+ if err = json.Unmarshal([]byte(pars), ¶ms); err != nil {
|
|
|
|
+ return
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ // 获取模板内容
|
|
|
|
+ demoPath := "static/htm2img/chart_detail.html"
|
|
|
|
+ contentByte, err := ioutil.ReadFile(demoPath)
|
|
|
|
+ if err != nil {
|
|
|
|
+ return
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ // 获取动态太阳码
|
|
|
|
+
|
|
|
|
+ // 替换模板内容
|
|
|
|
+ contentStr := string(contentByte)
|
|
|
|
+ contentStr = strings.ReplaceAll(contentStr, "{{CHART_NAME}}", params.ChartName)
|
|
|
|
+ contentStr = strings.ReplaceAll(contentStr, "{{CHART_IMAGE}}", params.ChartImage)
|
|
|
|
+ fmt.Println("ChartName:", params.ChartName)
|
|
|
|
+ fmt.Println("ChartImage:", params.ChartImage)
|
|
|
|
+
|
|
|
|
+ // 生成静态html文件
|
|
|
|
+ filePath = "static/htm2img/" + utils.GetRandStringNoSpecialChar(28) + ".html"
|
|
|
|
+ f, err := os.Create(filePath)
|
|
|
|
+ defer f.Close()
|
|
|
|
+ if err != nil {
|
|
|
|
+ return
|
|
|
|
+ }
|
|
|
|
+ if _, err = f.Write([]byte(contentStr)); err != nil {
|
|
|
|
+ return
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ return
|
|
|
|
+}
|
|
|
|
+
|
|
|
|
+// 生成静态html文件
|
|
|
|
+func PutHtmlIntoFile(content []byte) (filePath string, err error) {
|
|
|
|
+ filePath = "static/htm2img/" + utils.GetRandStringNoSpecialChar(28) + ".html"
|
|
|
|
+ f, err := os.Create(filePath)
|
|
|
|
+ defer f.Close()
|
|
|
|
+ if err != nil {
|
|
|
|
+ return
|
|
|
|
+ }
|
|
|
|
+ _, err = f.Write(content)
|
|
|
|
+ return
|
|
|
|
+}
|
|
|
|
+
|
|
|
|
+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) {
|
|
|
|
+ pythonUrl := ""
|
|
|
|
+ if global.CONFIG.Serve.RunMode == "release" {
|
|
|
|
+ //edbLibUrl = "http://172.19.173.232:8300/"
|
|
|
|
+ } else {
|
|
|
|
+ pythonUrl = "http://127.0.0.1:8888/"
|
|
|
|
+ }
|
|
|
|
+ postUrl := pythonUrl + "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请求edb库
|
|
|
|
+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
|
|
|
|
+}
|