MessageCenter/rpc/internal/common/MaterialService.go

package common

import (
	"app.yhyue.com/moapp/MessageCenter/entity"
	"app.yhyue.com/moapp/MessageCenter/rpc/internal/config"
	"app.yhyue.com/moapp/jybase/common"
	"app.yhyue.com/moapp/jybase/date"
	"app.yhyue.com/moapp/jybase/encrypt"
	"crypto/rand"
	"fmt"
	"github.com/gogf/gf/v2/util/gconv"
	"log"
	"math/big"
	"net/url"
	"strings"
	"time"
)

func MaterialDetail(msgId, positionId int64) (dataMap map[string]interface{}) {
	//查询到内容
	dataMap = map[string]interface{}{}
	data := entity.Mysql.SelectBySql("SELECT a.id,a.title,a.content,a.createtime,a.group_id,a.link,a.msg_type,b.file_url,b.material_content,b.qrcode_url,b.task_description,b.task_name,b.img_webpage,b.receive_user_name,b.receive_position_id,b.id as mid FROM jianyu.message_send_log a LEFT JOIN bi_service.operating_materials b ON a.id = b.msg_id WHERE a.id = ?", msgId)
	if data != nil && len(*data) > 0 {
		dataMap = (*data)[0]
		//查询图片信息
		attArr := []string{}
		res := entity.Mysql.SelectBySql("SELECT file_url FROM bi_service.operating_materials_attachment WHERE msg_id = ? AND position_id = ? ORDER BY id ASC", msgId, positionId)
		if res != nil && len(*res) > 0 {
			for _, v := range *res {
				attArr = append(attArr, gconv.String(v["file_url"]))
			}
			dataMap["attUrl"] = strings.Join(attArr, ",")
			code := ""
			ch := entity.JyBi.FindOne("dwd_d_userbase_belongto_rulecode", map[string]interface{}{"position_id": positionId}, "code", "")
			if ch != nil && len(*ch) > 0 {
				code = common.InterfaceToStr((*ch)["code"])
			}
			dataMap["qrcode_url"] = fmt.Sprintf("%s/material/%s/%s", config.ConfigJson.JyWebDomain, encrypt.SE.EncodeString(common.InterfaceToStr(dataMap["mid"])), code)
			dataMap["img_webpage"] = fmt.Sprintf("%s/material/%s/%s?types=1", config.ConfigJson.JyWebDomain, encrypt.SE.EncodeString(common.InterfaceToStr(dataMap["mid"])), code)
		} else {
			dataMap["attUrl"] = ""
		}
		return dataMap
	}
	return nil

}

func PersonImageSave(imgUrl string, msgId, positionId, materialsId int64) bool {
	saveMap := map[string]interface{}{
		"file_url":     imgUrl,
		"msg_id":       msgId,
		"position_id":  positionId,
		"materials_id": materialsId,
		"createtime":   time.Now().Format(date.Date_Full_Layout),
	}
	return entity.BiService.Insert("operating_materials_attachment", saveMap) > 0
}

func PersonImgSaveComposite(fileUrl, qrcodeUrl, name string, positionId int64) ([]byte, error) {
	//根据职位id获取到对应的渠道码
	code := ""
	data := entity.JyBi.FindOne("dwd_d_userbase_belongto_rulecode", map[string]interface{}{"position_id": positionId}, "code", "")
	if data != nil && len(*data) > 0 {
		code = gconv.String((*data)["code"])
	} else {
		//没有code，生成一个
		code = generateRandomString(6)
		entity.JyBi.Insert("dwd_d_userbase_belongto_rulecode", map[string]interface{}{
			"code":        code,
			"pcode":       "04",
			"level":       "2",
			"name":        "",
			"remark":      "人员渠道码",
			"SZ_LEVEL":    1,
			"SZ_LEAF":     1,
			"SZ_PID0":     "04",
			"SZ_PID1":     code,
			"position_id": positionId,
		})
	}
	//合成图片
	ok, err := hasAnyParameters(qrcodeUrl)
	if err != nil {
		return nil, err
	}
	if ok {
		qrcodeUrl = fmt.Sprintf("%s&personnelChannel=%s", qrcodeUrl, code)
	} else {
		qrcodeUrl = fmt.Sprintf("%s?personnelChannel=%s", qrcodeUrl, code)
	}
	log.Println("*********二维码图片链接：", qrcodeUrl)
	err, imgByte := compositeImage(fileUrl, qrcodeUrl)
	return imgByte, err
}

func generateRandomString(length int) string {
	const charset = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789"
	b := make([]byte, length)
	for i := range b {
		// 安全地生成随机索引 (0 到 len(charset)-1)
		idx, _ := rand.Int(rand.Reader, big.NewInt(int64(len(charset))))
		b[i] = charset[idx.Int64()]
	}
	return string(b)
}

func hasAnyParameters(rawURL string) (bool, error) {
	parsedURL, err := url.Parse(rawURL)
	if err != nil {
		return false, err
	}

	// 检查查询参数或片段标识符中的参数
	return parsedURL.RawQuery != "" || (parsedURL.Fragment != "" && strings.Contains(parsedURL.Fragment, "=")), nil
}