From c8bc8356448efdee5dd9c9e819a7aecac06b436c Mon Sep 17 00:00:00 2001
From: ChenYan <3139166962@qq.com>
Date: Mon, 13 Jan 2025 15:12:50 +0800
Subject: [PATCH] feat:

---
 .../resource/MallProductController.java       | 20 ++++++
 .../resource/WorkFlowController.java          |  3 +
 .../resource/domain/request/RequestModel.java | 31 +++++++++
 .../resource/mapper/MallProductMapper.java    |  1 +
 .../resource/mapper/ModelVersionMapper.java   |  8 +++
 .../mcwl/resource/service/ModelService.java   |  7 ++
 .../service/impl/MallProductServiceImpl.java  | 47 +++++++++++++
 .../mapper/resource/MallProductMapper.xml     | 67 ++++++++++++++++++-
 .../mapper/resource/ModelVersionMapper.xml    | 30 +++++++++
 9 files changed, 213 insertions(+), 1 deletion(-)
 create mode 100644 mcwl-resource/src/main/java/com/mcwl/resource/domain/request/RequestModel.java
 create mode 100644 mcwl-resource/src/main/resources/mapper/resource/ModelVersionMapper.xml

diff --git a/mcwl-admin/src/main/java/com/mcwl/web/controller/resource/MallProductController.java b/mcwl-admin/src/main/java/com/mcwl/web/controller/resource/MallProductController.java
index 25621ee..3c863b2 100644
--- a/mcwl-admin/src/main/java/com/mcwl/web/controller/resource/MallProductController.java
+++ b/mcwl-admin/src/main/java/com/mcwl/web/controller/resource/MallProductController.java
@@ -7,6 +7,8 @@ import com.mcwl.common.core.page.TableDataInfo;
 import com.mcwl.common.utils.SecurityUtils;
 import com.mcwl.resource.domain.ModelProduct;
 import com.mcwl.resource.domain.dto.ModelImagePageRes;
+import com.mcwl.resource.domain.request.RequestModel;
+import com.mcwl.resource.domain.request.RequestWorkFlow;
 import com.mcwl.resource.domain.vo.MallProductVo;
 import com.mcwl.resource.service.ModelService;
 import com.mcwl.system.service.ISysUserService;
@@ -123,6 +125,24 @@ public class MallProductController extends BaseController {
         return AjaxResult.success();
     }
 
+
+
+
+    @PostMapping("/insert")
+    public AjaxResult addupdateModel(@RequestBody RequestModel requestModel){
+
+        return modelService.addModel(requestModel);
+    }
+
+    @PostMapping("/update")
+    public AjaxResult updateModel(@RequestBody RequestModel requestModel){
+
+        modelService.updaModel(requestModel);
+
+        return AjaxResult.success("修改成功");
+    }
+
+
     @PostMapping("delete")
     public AjaxResult delete(@RequestBody ModelProduct modelVersion)
     {
diff --git a/mcwl-admin/src/main/java/com/mcwl/web/controller/resource/WorkFlowController.java b/mcwl-admin/src/main/java/com/mcwl/web/controller/resource/WorkFlowController.java
index b2bfc54..516e7d8 100644
--- a/mcwl-admin/src/main/java/com/mcwl/web/controller/resource/WorkFlowController.java
+++ b/mcwl-admin/src/main/java/com/mcwl/web/controller/resource/WorkFlowController.java
@@ -79,4 +79,7 @@ public class WorkFlowController extends BaseController {
 
         return AjaxResult.success("修改成功");
     }
+
+
+
 }
diff --git a/mcwl-resource/src/main/java/com/mcwl/resource/domain/request/RequestModel.java b/mcwl-resource/src/main/java/com/mcwl/resource/domain/request/RequestModel.java
new file mode 100644
index 0000000..288f3e8
--- /dev/null
+++ b/mcwl-resource/src/main/java/com/mcwl/resource/domain/request/RequestModel.java
@@ -0,0 +1,31 @@
+package com.mcwl.resource.domain.request;
+
+import com.mcwl.resource.domain.ModelProduct;
+import com.mcwl.resource.domain.ModelVersion;
+import com.mcwl.resource.domain.WorkFlow;
+import com.mcwl.resource.domain.WorkFlowVersion;
+import lombok.AllArgsConstructor;
+import lombok.Builder;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+import java.util.List;
+
+/**
+ * 模型入参 模型+版本
+ * @author DaiZibo
+ * @date 2025/1/9
+ * @apiNote
+ */
+
+@Builder
+@AllArgsConstructor
+@NoArgsConstructor
+@Data
+public class RequestModel {
+
+ private ModelProduct modelProduct;
+
+ private List<ModelVersion> modelVersionList;
+
+}
diff --git a/mcwl-resource/src/main/java/com/mcwl/resource/mapper/MallProductMapper.java b/mcwl-resource/src/main/java/com/mcwl/resource/mapper/MallProductMapper.java
index c0c28b1..394a47d 100644
--- a/mcwl-resource/src/main/java/com/mcwl/resource/mapper/MallProductMapper.java
+++ b/mcwl-resource/src/main/java/com/mcwl/resource/mapper/MallProductMapper.java
@@ -27,5 +27,6 @@ public interface MallProductMapper extends BaseMapper<ModelProduct> {
     Long sumNumber(@Param("userId") Long userId);
 
 
+    void updateModel(ModelProduct modelProduct);
 
 }
diff --git a/mcwl-resource/src/main/java/com/mcwl/resource/mapper/ModelVersionMapper.java b/mcwl-resource/src/main/java/com/mcwl/resource/mapper/ModelVersionMapper.java
index f2e243f..e495c36 100644
--- a/mcwl-resource/src/main/java/com/mcwl/resource/mapper/ModelVersionMapper.java
+++ b/mcwl-resource/src/main/java/com/mcwl/resource/mapper/ModelVersionMapper.java
@@ -1,8 +1,12 @@
 package com.mcwl.resource.mapper;
 
 import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+import com.mcwl.resource.domain.ModelProduct;
 import com.mcwl.resource.domain.ModelVersion;
 import org.apache.ibatis.annotations.Mapper;
+import org.apache.ibatis.annotations.Param;
+
+import java.util.List;
 
 /**
  * @Author：ChenYan
@@ -14,4 +18,8 @@ import org.apache.ibatis.annotations.Mapper;
  */
 @Mapper
 public interface ModelVersionMapper extends BaseMapper<ModelVersion> {
+    void addModelVersion(@Param("modelProduct") ModelProduct modelProduct, @Param("modelVersionList") List<ModelVersion> modelVersionList);
+
+
+
 }
diff --git a/mcwl-resource/src/main/java/com/mcwl/resource/service/ModelService.java b/mcwl-resource/src/main/java/com/mcwl/resource/service/ModelService.java
index 6c7d52d..2b39c16 100644
--- a/mcwl-resource/src/main/java/com/mcwl/resource/service/ModelService.java
+++ b/mcwl-resource/src/main/java/com/mcwl/resource/service/ModelService.java
@@ -2,11 +2,13 @@ package com.mcwl.resource.service;
 
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.IService;
+import com.mcwl.common.core.domain.AjaxResult;
 import com.mcwl.common.core.page.TableDataInfo;
 import com.mcwl.common.domain.IdsParam;
 import com.mcwl.resource.domain.ModelProduct;
 import com.mcwl.resource.domain.ModelVersion;
 import com.mcwl.resource.domain.dto.ModelImagePageRes;
+import com.mcwl.resource.domain.request.RequestModel;
 import com.mcwl.resource.domain.vo.MallProductVo;
 
 import java.util.List;
@@ -27,4 +29,9 @@ public interface ModelService extends IService<ModelProduct> {
 
     TableDataInfo listByPage(ModelImagePageRes imagePageRes);
 
+    AjaxResult addModel(RequestModel requestModel);
+
+
+    void updaModel(RequestModel requestModel);
+
 }
diff --git a/mcwl-resource/src/main/java/com/mcwl/resource/service/impl/MallProductServiceImpl.java b/mcwl-resource/src/main/java/com/mcwl/resource/service/impl/MallProductServiceImpl.java
index 17ba144..3579566 100644
--- a/mcwl-resource/src/main/java/com/mcwl/resource/service/impl/MallProductServiceImpl.java
+++ b/mcwl-resource/src/main/java/com/mcwl/resource/service/impl/MallProductServiceImpl.java
@@ -4,9 +4,11 @@ import cn.hutool.core.bean.BeanUtil;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.OrderItem;
+import com.baomidou.mybatisplus.core.toolkit.ObjectUtils;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.mcwl.common.constant.HttpStatus;
+import com.mcwl.common.core.domain.AjaxResult;
 import com.mcwl.common.core.domain.entity.SysUser;
 import com.mcwl.common.core.page.TableDataInfo;
 import com.mcwl.common.core.redis.RedisCache;
@@ -15,13 +17,20 @@ import com.mcwl.common.utils.SecurityUtils;
 import com.mcwl.common.utils.StringUtils;
 import com.mcwl.resource.domain.ModelImage;
 import com.mcwl.resource.domain.ModelProduct;
+import com.mcwl.resource.domain.ModelVersion;
+import com.mcwl.resource.domain.WorkFlowVersion;
 import com.mcwl.resource.domain.dto.ModelImagePageRes;
+import com.mcwl.resource.domain.request.RequestModel;
+import com.mcwl.resource.domain.request.RequestWorkFlow;
 import com.mcwl.resource.domain.vo.MallProductVo;
 import com.mcwl.resource.domain.vo.ModelImageVo;
 import com.mcwl.resource.mapper.MallProductMapper;
 
+import com.mcwl.resource.mapper.ModelVersionMapper;
 import com.mcwl.resource.service.ModelService;
 import com.mcwl.system.service.ISysUserService;
+import lombok.extern.log4j.Log4j;
+import lombok.extern.log4j.Log4j2;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
@@ -38,11 +47,14 @@ import java.util.Objects;
  * @Description TODO
  * @Date：2024/12/30 18:22
  */
+@Log4j2
 @Service
 public class MallProductServiceImpl extends ServiceImpl<MallProductMapper,ModelProduct> implements ModelService {
     @Autowired
     private RedisCache redisCache;
 
+    @Autowired
+    private ModelVersionMapper modelVersionMapper;
     @Autowired
     private MallProductMapper postMapper;
     @Autowired
@@ -137,6 +149,41 @@ public class MallProductServiceImpl extends ServiceImpl<MallProductMapper,ModelP
         return rspData;
     }
 
+    @Override
+    public AjaxResult addModel(RequestModel requestModel) {
+        //添加模型表数据
+        postMapper.insert(requestModel.getModelProduct());
 
+        log.info("获取到的入参:{}",requestModel.getModelProduct());
+
+        //批量添加版本
+        modelVersionMapper.addModelVersion(requestModel.getModelProduct(),requestModel.getModelVersionList());
+
+        return AjaxResult.success("添加成功，等待审核");
+    }
+
+    @Override
+    public void updaModel(RequestModel requestModel) {
+        //修改模版的信息
+        ModelProduct modelProduct = requestModel.getModelProduct();
+        if (ObjectUtils.isEmpty(modelProduct)){
+
+        }
+        if (StringUtils.isNotNull(modelProduct.getId())){
+            modelProduct.setAuditSatus(3);
+            postMapper.updateModel(requestModel.getModelProduct());
+        }
+
+        //修改工作流版本的信息
+        if (requestModel.getModelVersionList().size() != 0){
+
+            //批量修改
+            for (ModelVersion modelVersion : requestModel.getModelVersionList()) {
+                modelVersion.setAuditSatus(3);
+                modelVersionMapper.updateById(modelVersion);
+            }
+        }
+
+    }
 
 }
diff --git a/mcwl-resource/src/main/resources/mapper/resource/MallProductMapper.xml b/mcwl-resource/src/main/resources/mapper/resource/MallProductMapper.xml
index 97efce7..4a24f81 100644
--- a/mcwl-resource/src/main/resources/mapper/resource/MallProductMapper.xml
+++ b/mcwl-resource/src/main/resources/mapper/resource/MallProductMapper.xml
@@ -3,7 +3,72 @@
         PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
         "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
 <mapper namespace="com.mcwl.resource.mapper.MallProductMapper">
-
+    <update id="updateModel">
+        update model
+        <set>
+            <if test="modelName != null and modelName != ''">
+                model_name = #{modelName},
+            </if>
+            <if test="versionDescription != null and versionDescription != ''">
+                version_description = #{versionDescription},
+            </if>
+            <if test="updateBy != null and updateBy != ''">
+                update_by = #{updateBy},
+            </if>
+            <if test="updateTime != null">
+                update_time = #{updateTime},
+            </if>
+            <if test="modelTypeId != null">
+                model_type_id = #{modelTypeId},
+            </if>
+            <if test="category != null and category != ''">
+                category = #{category},
+            </if>
+            <if test="functions != null and functions != ''">
+                functions = #{functions},
+            </if>
+            <if test="tags != null and tags != ''">
+                tags = #{tags},
+            </if>
+            <if test="activityId != null and activityId != ''">
+                activity_id = #{activityId},
+            </if>
+            <if test="isOriginal != null">
+                is_original = #{isOriginal},
+            </if>
+            <if test="originalAuthorName != null and originalAuthorName != ''">
+                original_author_name = #{originalAuthorName},
+            </if>
+            <if test="reals != null">
+                reals = #{reals},
+            </if>
+            <if test="numbers != null">
+                numbers = #{numbers},
+            </if>
+            <if test="auditSatus != null">
+                audit_satus = #{auditSatus},
+            </if>
+            <if test="isRecommend != null">
+                is_recommend = #{isRecommend},
+            </if>
+            <if test="delFlag != null and delFlag != ''">
+                del_flag = #{delFlag},
+            </if>
+            <if test="remark != null and remark != ''">
+                remark = #{remark},
+            </if>
+            <if test="auditText != null and auditText != ''">
+                audit_text = #{auditText},
+            </if>
+            <if test="userId != null">
+                user_id = #{userId},
+            </if>
+            <if test="versionId != null">
+                version_id = #{versionId},
+            </if>
+        </set>
+        where id = #{id}
+    </update>
 
 
     <select id="sumNumber" resultType="java.lang.Long">
diff --git a/mcwl-resource/src/main/resources/mapper/resource/ModelVersionMapper.xml b/mcwl-resource/src/main/resources/mapper/resource/ModelVersionMapper.xml
new file mode 100644
index 0000000..7ac880f
--- /dev/null
+++ b/mcwl-resource/src/main/resources/mapper/resource/ModelVersionMapper.xml
@@ -0,0 +1,30 @@
+<?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.mcwl.resource.mapper.ModelVersionMapper">
+
+
+    <insert id="addModelVersion">
+            INSERT INTO model_version (
+            version_name, model_type, file_path, trigger_words, sampling, high, vae, cfg,
+            is_free, is_public, is_encrypt, is_online_use, allow_download_image,
+            allow_software_use, allow_fusion, allow_commercial_use, allow_usage,
+            is_exclusive_model, sample_image_paths, hide_image_gen_info, audit_status,
+            audit_text, del_flag
+            )
+            VALUES
+            <foreach collection="list" item="item" separator=",">
+                (
+                #{item.versionName}, #{item.modelType}, #{item.filePath}, #{item.triggerWords},
+                #{item.sampling}, #{item.high}, #{item.vae}, #{item.cfg},
+                #{item.isFree}, #{item.isPublic}, #{item.isEncrypt}, #{item.isOnlineUse},
+                #{item.allowDownloadImage}, #{item.allowSoftwareUse}, #{item.allowFusion},
+                #{item.allowCommercialUse}, #{item.allowUsage}, #{item.isExclusiveModel},
+                #{item.sampleImagePaths}, #{item.hideImageGenInfo}, #{item.auditStatus},
+                #{item.auditText}, #{item.delFlag}
+                )
+            </foreach>
+    </insert>
+
+</mapper>