11

server/websocket/internal/logic/datatransferlogic.go

@@ -80,7 +80,7 @@ var (
 	//渲染任务缓冲队列长度
 	renderChanLen = 500
 	//是否开启debug
-	openDebug = false
+	openDebug = true
 	//允许跨域的origin
 	mapAllowOrigin = map[string]struct{}{
 		"https://www.fusen.3718.cn": struct{}{},
@@ -118,11 +118,13 @@ type wsConnectItem struct {
 func (l *DataTransferLogic) DataTransfer(w http.ResponseWriter, r *http.Request) {
 	origin := r.Header.Get("Origin")
 	//判断是不是允许的跨域
-	upgrader.CheckOrigin = func(r *http.Request) bool {
-		if _, ok := mapAllowOrigin[origin]; !ok {
-			return false
+	if !openDebug {
+		upgrader.CheckOrigin = func(r *http.Request) bool {
+			if _, ok := mapAllowOrigin[origin]; !ok {
+				return false
+			}
+			return true
 		}
-		return true
 	}
 	//把子协议携带的token设置到标准token头信息中
 	token := r.Header.Get("Sec-Websocket-Protocol")

server/websocket/internal/logic/ws_render_image.go

@@ -11,6 +11,7 @@ import (
 	"fusenapi/service/repositories"
 	"fusenapi/utils/curl"
 	"fusenapi/utils/hash"
+	"fusenapi/utils/template_switch_info"
 	"fusenapi/utils/websocket_data"
 	"github.com/zeromicro/go-zero/core/logx"
 	"gorm.io/gorm"
@@ -135,6 +136,20 @@ func (w *wsConnectItem) renderImage(data []byte) {
 		logx.Error("failed to get element ,", err)
 		return
 	}
+	//获取模板开关信息并且对于没有默认值的给赋值默认值
+	templateSwitchInfo := template_switch_info.GetTemplateSwitchInfo(productTemplate.Id, *productTemplate.MaterialImg)
+	if renderImageData.RenderData.Website == "" {
+		renderImageData.RenderData.Website = templateSwitchInfo.MaterialData.Website.DefaultValue
+	}
+	if renderImageData.RenderData.Phone == "" {
+		renderImageData.RenderData.Phone = templateSwitchInfo.MaterialData.Phone.DefaultValue
+	}
+	if renderImageData.RenderData.Address == "" {
+		renderImageData.RenderData.Address = templateSwitchInfo.MaterialData.Address.DefaultValue
+	}
+	if renderImageData.RenderData.Qrcode == "" {
+		renderImageData.RenderData.Qrcode = templateSwitchInfo.MaterialData.QRcode.DefaultValue
+	}
 	//获取刀版图
 	combineReq := repositories.LogoCombineReq{
 		UserId:      renderImageData.RenderData.UserId,