fix

virgo.api/src/main/java/com/bosshand/virgo/api/workark/controller/DifyController.java

@@ -3,6 +3,7 @@ package com.bosshand.virgo.api.workark.controller;
 import com.bosshand.virgo.api.workark.model.DifyCompletion;
 import com.bosshand.virgo.api.workark.model.DifyType;
 import com.bosshand.virgo.api.workark.model.DifyWorkFlow;
+import com.bosshand.virgo.api.workark.model.HtmlCode;
 import com.bosshand.virgo.api.workark.service.DifyService;
 import com.bosshand.virgo.core.response.Response;
 import io.swagger.annotations.Api;
@@ -40,6 +41,32 @@ public class DifyController {
         return Response.ok(difyService.getListType());
     }
 
+    @ApiOperation("保存HtmlCode")
+    @RequestMapping(value = "/htmlCode", method = RequestMethod.POST)
+    public Response insertHtmlCode(@RequestBody HtmlCode htmlCode) {
+        difyService.saveHtmlCode(htmlCode);
+        return Response.ok();
+    }
+
+    @ApiOperation("更新HtmlCode")
+    @RequestMapping(value = "/htmlCode/update", method = RequestMethod.PUT)
+    public Response updateHtmlCode(@RequestBody HtmlCode htmlCode) {
+        difyService.updateHtmlCode(htmlCode);
+        return Response.ok();
+    }
+
+    @ApiOperation("查询HtmlCode")
+    @RequestMapping(value = "/htmlCode/query", method = RequestMethod.POST)
+    public Response getListHtmlCode(@RequestBody HtmlCode htmlCode) {
+        return Response.ok(difyService.getListHtmlCode(htmlCode));
+    }
+
+    @ApiOperation("HtmlCode详情")
+    @RequestMapping(value = "/htmlCode/{id}", method = RequestMethod.GET)
+    public Response getHtmlCode(@PathVariable long id) {
+        return Response.ok(difyService.getHtmlCode(id));
+    }
+
     @ApiOperation("执行workflow")
     @RequestMapping(value = "/workflow/run/{difyTypeId}", method = RequestMethod.POST)
     public Response workFlowRun(@PathVariable long difyTypeId, @RequestBody Map<String, Object> inputs) {

virgo.api/src/main/java/com/bosshand/virgo/api/workark/dao/HtmlCodeDao.java

@@ -0,0 +1,19 @@
+package com.bosshand.virgo.api.workark.dao;
+
+import com.bosshand.virgo.api.workark.model.HtmlCode;
+import org.apache.ibatis.annotations.Mapper;
+
+import java.util.List;
+
+@Mapper
+public interface HtmlCodeDao {
+
+    void save(HtmlCode htmlCode);
+
+    void update(HtmlCode htmlCode);
+
+    List<HtmlCode> getList(HtmlCode htmlCode);
+
+    HtmlCode get(long id);
+
+}

virgo.api/src/main/java/com/bosshand/virgo/api/workark/model/HtmlCode.java

@@ -0,0 +1,63 @@
+package com.bosshand.virgo.api.workark.model;
+
+import com.fasterxml.jackson.annotation.JsonFormat;
+
+import java.util.Date;
+
+/**
+ * 网站页面代码
+ */
+public class HtmlCode {
+
+    private Long id;
+
+    private String data;
+
+    @JsonFormat(shape = JsonFormat.Shape.STRING, pattern = "yyyy-MM-dd HH:mm:ss", timezone = "GMT+8")
+    private Date date;
+
+    @JsonFormat(shape = JsonFormat.Shape.STRING, pattern = "yyyy-MM-dd HH:mm:ss", timezone = "GMT+8")
+    private Date updateDate;
+
+    private long userId;
+
+    public Long getId() {
+        return id;
+    }
+
+    public void setId(Long id) {
+        this.id = id;
+    }
+
+    public String getData() {
+        return data;
+    }
+
+    public void setData(String data) {
+        this.data = data;
+    }
+
+    public Date getDate() {
+        return date;
+    }
+
+    public void setDate(Date date) {
+        this.date = date;
+    }
+
+    public Date getUpdateDate() {
+        return updateDate;
+    }
+
+    public void setUpdateDate(Date updateDate) {
+        this.updateDate = updateDate;
+    }
+
+    public long getUserId() {
+        return userId;
+    }
+
+    public void setUserId(long userId) {
+        this.userId = userId;
+    }
+}

virgo.api/src/main/java/com/bosshand/virgo/api/workark/service/DifyService.java

@@ -4,9 +4,11 @@ import com.alibaba.fastjson.JSONObject;
 import com.bosshand.virgo.api.workark.dao.DifyCompletionDao;
 import com.bosshand.virgo.api.workark.dao.DifyTypeDao;
 import com.bosshand.virgo.api.workark.dao.DifyWorkFlowDao;
+import com.bosshand.virgo.api.workark.dao.HtmlCodeDao;
 import com.bosshand.virgo.api.workark.model.DifyCompletion;
 import com.bosshand.virgo.api.workark.model.DifyType;
 import com.bosshand.virgo.api.workark.model.DifyWorkFlow;
+import com.bosshand.virgo.api.workark.model.HtmlCode;
 import com.bosshand.virgo.core.model.UserContext;
 import com.bosshand.virgo.core.utils.ContextUtils;
 import io.github.imfangs.dify.client.DifyClientFactory;
@@ -40,6 +42,9 @@ public class DifyService {
     @Autowired
     DifyCompletionDao difyCompletionDao;
 
+    @Autowired
+    HtmlCodeDao htmlCodeDao;
+
     public void saveType(DifyType difyType) {
         difyTypeDao.save(difyType);
     }
@@ -56,6 +61,25 @@ public class DifyService {
         return difyWorkFlowDao.getList(difyWorkFlow);
     }
 
+    public void saveHtmlCode(HtmlCode htmlCode) {
+        long userId = ContextUtils.getUserContext().getUserId();
+        htmlCode.setUserId(userId);
+        htmlCodeDao.save(htmlCode);
+    }
+
+    public void updateHtmlCode(HtmlCode htmlCode) {
+        htmlCodeDao.update(htmlCode);
+    }
+
+    public HtmlCode getHtmlCode(long id) {
+        return htmlCodeDao.get(id);
+    }
+
+    public List<HtmlCode> getListHtmlCode(HtmlCode htmlCode) {
+        return htmlCodeDao.getList(htmlCode);
+    }
+
+
     /**
      * 工作流应用
      */

virgo.api/src/main/resources/mapper/HtmlCodeMapper.xml

@@ -0,0 +1,39 @@
+<?xml version="1.0" encoding="UTF-8" ?>
+<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd" >
+<mapper namespace="com.bosshand.virgo.api.workark.dao.HtmlCodeDao">
+
+    <resultMap type="com.bosshand.virgo.api.workark.model.HtmlCode" id="result">
+        <id column="id" property="id"/>
+        <result column="data" property="data"/>
+        <result column="date" property="date"/>
+        <result column="updateDate" property="updateDate"/>
+        <result column="userId" property="userId"/>
+    </resultMap>
+
+    <insert id="save" parameterType="com.bosshand.virgo.api.workark.model.HtmlCode" useGeneratedKeys="true" keyProperty="id">
+        INSERT INTO dify_html_code(`data`, `date`, `userId`) VALUES (#{data}, now(), #{userId})
+    </insert>
+
+    <update id="update" parameterType="com.bosshand.virgo.api.workark.model.HtmlCode">
+        UPDATE dify_html_code
+        <trim prefix="set" suffixOverrides=",">
+            <if test="data != null">data = #{data},</if>
+            updateDate = now(),
+        </trim>
+        WHERE id=#{id}
+    </update>
+
+    <select id="getList" resultMap="result">
+        select * from dify_html_code
+        <where>
+            <if test="userId != 0">
+                and userId = #{userId}
+            </if>
+        </where>
+    </select>
+
+    <select id="get" resultMap="result">
+        select * from dify_html_code where id =#{id}
+    </select>
+
+</mapper>