feature:画图方法添加画意值消耗逻辑、添加获取画意值方法

bnyer-common/bnyer-common-core/src/main/java/com/bnyer/common/core/constant/TiktokConstant.java

@@ -90,4 +90,14 @@ public class TiktokConstant {
      *  邮箱已被注册使用
      */
     public static final Integer EMAIL_EXIST_ERROR = 105015;
+
+    /**
+     * 画意值不足
+     */
+    public static final Integer GOLD_NUM_IS_NOT_ENOUGH = 105016;
+
+    /**
+     * 用户类型错误
+     */
+    public static final Integer USER_TYPE_ERROR = 105017;
 }

bnyer-common/bnyer-common-core/src/main/java/com/bnyer/common/core/dto/TextToImgDto.java

@@ -64,4 +64,7 @@ public class TextToImgDto implements Serializable {
     @ApiModelProperty(value="平台")
     private String platform;
 
+    @ApiModelProperty(value="消耗画意值")
+    private int goldNum;
+
 }

bnyer-common/bnyer-common-core/src/main/java/com/bnyer/common/core/dto/TextToImgLoveDto.java

@@ -35,4 +35,7 @@ public class TextToImgLoveDto implements Serializable {
 
     @ApiModelProperty(value="平台")
     private String platform;
+
+    @ApiModelProperty(value="消耗画意值")
+    private int goldNum;
 }

bnyer-services/bnyer-img/src/main/java/com/bnyer/img/controller/FhMiniController.java

@@ -452,4 +452,10 @@ public class FhMiniController extends BaseController {
         List<DiamondLogVo> diamondLogs = diamondLogService.queryPage(dto.getUserId(), "2");
         return getDataTable(diamondLogs);
     }
+
+    @ApiOperation(value="获取用户画意值")
+    @GetMapping(value = "/queryGoldNum/{id}")
+    public AjaxResult queryGoldNum(@PathVariable @ApiParam("主键id") Long id){
+        return AjaxResult.success(fhUserService.getUserGoldNum(id));
+    }
 }

bnyer-services/bnyer-img/src/main/java/com/bnyer/img/controller/TiktokMiniController.java

@@ -488,4 +488,10 @@ public class TiktokMiniController extends BaseController {
         List<DiamondLogVo> diamondLogs = diamondLogService.queryPage(dto.getUserId(), "1");
         return getDataTable(diamondLogs);
     }
+
+    @ApiOperation(value="获取用户画意值")
+    @GetMapping(value = "/queryGoldNum/{id}")
+    public AjaxResult queryGoldNum(@PathVariable @ApiParam("主键id") Long id){
+        return AjaxResult.success(tiktokUserService.getUserGoldNum(id));
+    }
 }

bnyer-services/bnyer-img/src/main/java/com/bnyer/img/controller/WxMiniController.java

@@ -478,4 +478,10 @@ public class WxMiniController extends BaseController {
         List<DiamondLogVo> diamondLogs = diamondLogService.queryPage(dto.getUserId(), "3");
         return getDataTable(diamondLogs);
     }
+
+    @ApiOperation(value="获取用户画意值")
+    @GetMapping(value = "/queryGoldNum/{id}")
+    public AjaxResult queryGoldNum(@PathVariable @ApiParam("主键id") Long id){
+        return AjaxResult.success(wxUserService.getUserGoldNum(id));
+    }
 }

bnyer-services/bnyer-img/src/main/java/com/bnyer/img/mapper/FhUserMapper.java

@@ -3,6 +3,7 @@ package com.bnyer.img.mapper;
 import com.baomidou.mybatisplus.core.mapper.BaseMapper;
 import com.bnyer.common.core.domain.FhUser;
 import org.apache.ibatis.annotations.Mapper;
+import org.apache.ibatis.annotations.Param;
 
 /**
  * @author  chengkun
@@ -11,4 +12,10 @@ import org.apache.ibatis.annotations.Mapper;
 @Mapper
 public interface FhUserMapper extends BaseMapper<FhUser> {
 
+    /**
+     * 根据id获取画意值
+     * @param id 主键id
+     * @return -
+     */
+    int queryGoldNum(@Param("id") Long id);
 }

bnyer-services/bnyer-img/src/main/java/com/bnyer/img/mapper/TiktokUserMapper.java

@@ -3,8 +3,15 @@ package com.bnyer.img.mapper;
 import com.baomidou.mybatisplus.core.mapper.BaseMapper;
 import com.bnyer.common.core.domain.TiktokUser;
 import org.apache.ibatis.annotations.Mapper;
+import org.apache.ibatis.annotations.Param;
 
 @Mapper
 public interface TiktokUserMapper extends BaseMapper<TiktokUser> {
 
+    /**
+     * 根据id获取画意值
+     * @param id 主键id
+     * @return -
+     */
+    int queryGoldNum(@Param("id") Long id);
 }

bnyer-services/bnyer-img/src/main/java/com/bnyer/img/mapper/UserVipRecordMapper.java

@@ -8,5 +8,6 @@ import org.apache.ibatis.annotations.Mapper;
 
 @Mapper
 public interface UserVipRecordMapper extends BaseMapper<UserVipRecord> {
+
     UserVipRecordVo queryUserVipRecord(UserVipRecordQuery query);
 }

bnyer-services/bnyer-img/src/main/java/com/bnyer/img/mapper/WxUserMapper.java

@@ -3,6 +3,7 @@ package com.bnyer.img.mapper;
 import com.baomidou.mybatisplus.core.mapper.BaseMapper;
 import com.bnyer.common.core.domain.WxUser;
 import org.apache.ibatis.annotations.Mapper;
+import org.apache.ibatis.annotations.Param;
 
 /**
  * @author  chengkun
@@ -10,4 +11,10 @@ import org.apache.ibatis.annotations.Mapper;
  */
 @Mapper
 public interface WxUserMapper extends BaseMapper<WxUser> {
+    /**
+     * 根据id获取画意值
+     * @param id 主键id
+     * @return -
+     */
+    int queryGoldNum(@Param("id") Long id);
 }

bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/FhUserService.java

@@ -45,4 +45,11 @@ public interface FhUserService {
      * @param tiktokUser 快手用户
      */
     void updateFhUser(FhUser tiktokUser);
+
+    /**
+     * 根据id获取用户画意值
+     * @param id 主键id
+     * @return -
+     */
+    int getUserGoldNum(Long id);
 }

bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/StableDiffusionService.java

@@ -11,15 +11,15 @@ public interface StableDiffusionService {
 
     /**
      * 文生图
-     * @param param
+     * @param param 参数
-     * @return
+     * @return -
      */
     TextToImgVo textToImg(TextToImgDto param);
 
     /**
      * 情侣头像
-     * @param param
+     * @param param 参数
-     * @return
+     * @return -
      */
     List<String> textToImgLove(TextToImgLoveDto param);
 

bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/TiktokUserService.java

@@ -44,4 +44,11 @@ public interface TiktokUserService {
      * @param tiktokUser 抖音用户
      */
     void updateTiktokUser(TiktokUser tiktokUser);
+
+    /**
+     * 根据id获取用户画意值
+     * @param id 主键id
+     * @return -
+     */
+    int getUserGoldNum(Long id);
 }

bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/WxUserService.java

@@ -44,4 +44,11 @@ public interface WxUserService {
      * @param tiktokUser 微信用户
      */
     void updateWxUser(WxUser tiktokUser);
+
+    /**
+     * 根据id获取用户画意值
+     * @param id 主键id
+     * @return -
+     */
+    int getUserGoldNum(Long id);
 }

bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/impl/FhUserServiceImpl.java

@@ -190,4 +190,9 @@ public class FhUserServiceImpl implements FhUserService {
         fhUser.setUpdateTime(new Date());
         fhUserMapper.updateById(fhUser);
     }
+
+    @Override
+    public int getUserGoldNum(Long id) {
+        return fhUserMapper.queryGoldNum(id);
+    }
 }

bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/impl/StableDiffusionServiceImpl.java

@@ -4,18 +4,22 @@ import cn.hutool.core.util.IdUtil;
 import com.alibaba.fastjson.JSON;
 import com.alibaba.fastjson.JSONObject;
 import com.bnyer.common.core.constant.RedisKeyConstant;
+import com.bnyer.common.core.constant.TiktokConstant;
 import com.bnyer.common.core.domain.AiPaint;
 import com.bnyer.common.core.domain.FhUser;
 import com.bnyer.common.core.domain.TiktokUser;
 import com.bnyer.common.core.domain.WxUser;
 import com.bnyer.common.core.dto.TextToImgDto;
 import com.bnyer.common.core.dto.TextToImgLoveDto;
+import com.bnyer.common.core.enums.EnumUserClientType;
 import com.bnyer.common.core.exception.ServiceException;
 import com.bnyer.common.core.utils.StringUtils;
 import com.bnyer.common.core.utils.TranslateUtils;
 import com.bnyer.common.core.utils.file.Base64ToMultipartFileUtils;
 import com.bnyer.common.core.vo.TextToImgVo;
+import com.bnyer.common.core.vo.UserInfoVo;
 import com.bnyer.common.redis.service.RedisService;
+import com.bnyer.common.security.utils.SecurityUtils;
 import com.bnyer.file.api.RemoteFileService;
 import com.bnyer.img.config.StableDiffusionConfig;
 import com.bnyer.img.config.TencentTranslateConfig;
@@ -101,6 +105,44 @@ public class StableDiffusionServiceImpl implements StableDiffusionService {
 
     @Override
     public TextToImgVo textToImg(TextToImgDto param) {
+        //消耗画意值
+        switch (param.getPlatform()){
+            case "1":
+                //查询抖音用户信息
+                TiktokUser tiktokUser = tiktokUserService.getUserById(param.getPainterId());
+                //检查画意值是否足够，不足则直接提示不足请充值或完成相应任务
+                if (param.getGoldNum() > tiktokUser.getGoldNum()) {
+                    throw new ServiceException("画意值不足!请充值或完成相应任务获取", TiktokConstant.GOLD_NUM_IS_NOT_ENOUGH);
+                }
+                tiktokUser.setGoldNum(tiktokUser.getGoldNum() - param.getGoldNum());
+                //更新用户信息
+                tiktokUserService.updateTiktokUser(tiktokUser);
+                break;
+            case "2":
+                //查询快手用户信息
+                FhUser fhUser = fhUserService.getUserById(param.getPainterId());
+                //检查画意值是否足够，不足则直接提示不足请充值或完成相应任务
+                if (param.getGoldNum() > fhUser.getGoldNum()) {
+                    throw new ServiceException("画意值不足!请充值或完成相应任务获取", TiktokConstant.GOLD_NUM_IS_NOT_ENOUGH);
+                }
+                fhUser.setGoldNum(fhUser.getGoldNum() - param.getGoldNum());
+                //更新用户信息
+                fhUserService.updateFhUser(fhUser);
+                break;
+            case "3":
+                //查询微信用户信息
+                WxUser wxUser = wxUserService.getUserById(param.getPainterId());
+                //检查画意值是否足够，不足则直接提示不足请充值或完成相应任务
+                if (param.getGoldNum() > wxUser.getGoldNum()) {
+                    throw new ServiceException("画意值不足!请充值或完成相应任务获取", TiktokConstant.GOLD_NUM_IS_NOT_ENOUGH);
+                }
+                wxUser.setGoldNum(wxUser.getGoldNum() - param.getGoldNum());
+                //更新用户信息
+                wxUserService.updateWxUser(wxUser);
+                break;
+            default:
+                throw new ServiceException("用户类型错误!", TiktokConstant.USER_TYPE_ERROR);
+        }
         //配置控制采用sd还是tx文生图
         int button = redisService.getCacheObject(RedisKeyConstant.AI_PAINT_BUTTON);
         if(button == AiPaintButtonEnum.TX.getCode()){
@@ -165,7 +207,7 @@ public class StableDiffusionServiceImpl implements StableDiffusionService {
                     paint.setPaintId(paintId);
                     paint.setCreateTime(paintTime);
                     paint.setImgUrl(imgStr);
-                    paint.setPrompt(resultPrompt);
+                    paint.setPrompt(contentPrompt); //只存储用户输入的内容
                     paint.setNegativePrompt(negaPrompt);
                     paint.setModel(param.getModelName());
                     paint.setStyleName(param.getStyleName());
@@ -230,7 +272,7 @@ public class StableDiffusionServiceImpl implements StableDiffusionService {
                 Double cfgScale = param.getCfgScale() == null ? 7 : param.getCfgScale();
                 Integer steps = param.getSteps() == null ? 25 : param.getSteps();
                 String samplerIndex = StringUtils.isEmpty(param.getSamplerIndex()) ? "DPM++ 2S a Karras" : param.getSamplerIndex();
-                map.put("prompt", resultPrompt);
+                map.put("prompt", contentPrompt);  //只存储用户输入的内容
                 map.put("restore_faces",false); //面部修复,卡通模型不支持,只适合真人模型
                 map.put("tiling",false); //生成可平铺的周期性图片
                 map.put("eta",eta); //取值为0-1
@@ -297,6 +339,44 @@ public class StableDiffusionServiceImpl implements StableDiffusionService {
 
     @Override
     public List<String> textToImgLove(TextToImgLoveDto param) {
+        //消耗画意值
+        switch (param.getPlatform()){
+            case "1":
+                //查询抖音用户信息
+                TiktokUser tiktokUser = tiktokUserService.getUserById(param.getPainterId());
+                //检查画意值是否足够，不足则直接提示不足请充值或完成相应任务
+                if (param.getGoldNum() > tiktokUser.getGoldNum()) {
+                    throw new ServiceException("画意值不足!请充值或完成相应任务获取", TiktokConstant.GOLD_NUM_IS_NOT_ENOUGH);
+                }
+                tiktokUser.setGoldNum(tiktokUser.getGoldNum() - param.getGoldNum());
+                //更新用户信息
+                tiktokUserService.updateTiktokUser(tiktokUser);
+                break;
+            case "2":
+                //查询快手用户信息
+                FhUser fhUser = fhUserService.getUserById(param.getPainterId());
+                //检查画意值是否足够，不足则直接提示不足请充值或完成相应任务
+                if (param.getGoldNum() > fhUser.getGoldNum()) {
+                    throw new ServiceException("画意值不足!请充值或完成相应任务获取", TiktokConstant.GOLD_NUM_IS_NOT_ENOUGH);
+                }
+                fhUser.setGoldNum(fhUser.getGoldNum() - param.getGoldNum());
+                //更新用户信息
+                fhUserService.updateFhUser(fhUser);
+                break;
+            case "3":
+                //查询微信用户信息
+                WxUser wxUser = wxUserService.getUserById(param.getPainterId());
+                //检查画意值是否足够，不足则直接提示不足请充值或完成相应任务
+                if (param.getGoldNum() > wxUser.getGoldNum()) {
+                    throw new ServiceException("画意值不足!请充值或完成相应任务获取", TiktokConstant.GOLD_NUM_IS_NOT_ENOUGH);
+                }
+                wxUser.setGoldNum(wxUser.getGoldNum() - param.getGoldNum());
+                //更新用户信息
+                wxUserService.updateWxUser(wxUser);
+                break;
+            default:
+                throw new ServiceException("用户类型错误!", TiktokConstant.USER_TYPE_ERROR);
+        }
         //描述词处理
         //内容提示词
         String contentPrompt = param.getPrompt();
@@ -317,9 +397,9 @@ public class StableDiffusionServiceImpl implements StableDiffusionService {
             String boyHeadStr = "1个男孩" + activePrompt + resultPrompt;
             //女头提示词
             String girlHeadStr = "1个女孩" + activePrompt + resultPrompt;
-            String boyHead = getCoupleHead(boyHeadStr,negaPrompt,param.getPlatform(), param.getPainterId(), param.getPainterName(), param.getStyleName(),button);
+            String boyHead = getCoupleHead(boyHeadStr,negaPrompt,param.getPlatform(), param.getPainterId(), param.getPainterName(), param.getStyleName(),button,contentPrompt);
             log.info("腾讯男头像提示词为：【{}】",girlHeadStr);
-            String girlHead = getCoupleHead(girlHeadStr,negaPrompt,param.getPlatform(), param.getPainterId(), param.getPainterName(), param.getStyleName(),button);
+            String girlHead = getCoupleHead(girlHeadStr,negaPrompt,param.getPlatform(), param.getPainterId(), param.getPainterName(), param.getStyleName(),button,contentPrompt);
             log.info("腾讯女头像提示词为：【{}】",girlHeadStr);
             List<String> imgList = new ArrayList<>();
             imgList.add(boyHead);
@@ -343,9 +423,9 @@ public class StableDiffusionServiceImpl implements StableDiffusionService {
                 //女头提示词
                 String girlHeadStr = "(1girl)" + activePrompt + prompt;
 
-                String boyHead = getCoupleHead(boyHeadStr,negaPrompt,param.getPlatform(), param.getPainterId(), param.getPainterName(), param.getStyleName(),button);
+                String boyHead = getCoupleHead(boyHeadStr,negaPrompt,param.getPlatform(), param.getPainterId(), param.getPainterName(), param.getStyleName(),button,contentPrompt);
                 log.info("sd男头像提示词为：【{}】",boyHeadStr);
-                String girlHead = getCoupleHead(girlHeadStr,negaPrompt,param.getPlatform(), param.getPainterId(), param.getPainterName(), param.getStyleName(),button);
+                String girlHead = getCoupleHead(girlHeadStr,negaPrompt,param.getPlatform(), param.getPainterId(), param.getPainterName(), param.getStyleName(),button,contentPrompt);
                 log.info("sd女头像提示词为：【{}】",girlHeadStr);
                 List<String> imgList = new ArrayList<>();
                 imgList.add(boyHead);
@@ -367,9 +447,10 @@ public class StableDiffusionServiceImpl implements StableDiffusionService {
      * @param painterId 绘图者id
      * @param painterName 绘图者名称
      * @param styleName 风格名称
+     * @param protoHead 原用户输入的描述词
      * @return -
      */
-    private String getCoupleHead(String headStr,String negaPrompt,String platform,Long painterId,String painterName,String styleName,int button){
+    private String getCoupleHead(String headStr,String negaPrompt,String platform,Long painterId,String painterName,String styleName,int button,String protoHead){
 
         if(button == AiPaintButtonEnum.TX.getCode()){
             //采用腾讯文生图
@@ -419,7 +500,7 @@ public class StableDiffusionServiceImpl implements StableDiffusionService {
                     paint.setPaintId(paintId);
                     paint.setCreateTime(paintTime);
                     paint.setImgUrl(imgStr);
-                    paint.setPrompt(headStr);
+                    paint.setPrompt(protoHead);
                     paint.setNegativePrompt(negaPrompt);
                     paint.setModel("腾讯情侣头像");
                     paint.setStyleName(styleName);
@@ -430,6 +511,7 @@ public class StableDiffusionServiceImpl implements StableDiffusionService {
                     paint.setPainterId(painterId);
                     paint.setPainterName(painterName);
                     aiPaintService.insert(paint);
+
                     return image;
                 }
             } catch (TencentCloudSDKException e) {
@@ -479,7 +561,7 @@ public class StableDiffusionServiceImpl implements StableDiffusionService {
                         paint.setPaintId(paintId);
                         paint.setCreateTime(paintTime);
                         paint.setImgUrl(imgStr);
-                        paint.setPrompt(headStr);
+                        paint.setPrompt(protoHead);
                         paint.setNegativePrompt(negaPrompt);
                         paint.setEta(eta);
                         paint.setBatchSize(batchSize);

bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/impl/TiktokUserServiceImpl.java

@@ -196,4 +196,9 @@ public class TiktokUserServiceImpl implements TiktokUserService {
         tiktokUser.setUpdateTime(new Date());
         tiktokUserMapper.updateById(tiktokUser);
     }
+
+    @Override
+    public int getUserGoldNum(Long id) {
+        return tiktokUserMapper.queryGoldNum(id);
+    }
 }

bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/impl/WxUserServiceImpl.java

@@ -192,4 +192,9 @@ public class WxUserServiceImpl implements WxUserService {
         wxUser.setUpdateTime(new Date());
         wxUserMapper.updateById(wxUser);
     }
+
+    @Override
+    public int getUserGoldNum(Long id) {
+        return wxUserMapper.queryGoldNum(id);
+    }
 }

bnyer-services/bnyer-img/src/main/resources/com/bnyer/img/mapper/FhUserMapper.xml

@@ -22,4 +22,10 @@
     id, username, fh_code,phone, img,is_vip,gold_num,diamond_num, is_show, create_time, update_time, sort
   </sql>
 
+  <select id="queryGoldNum" resultType="int">
+    select
+        gold_num
+    from img_fh_user
+    where id = #{id} and is_show = "1"
+  </select>
 </mapper>

bnyer-services/bnyer-img/src/main/resources/com/bnyer/img/mapper/TiktokUserMapper.xml

@@ -23,4 +23,11 @@
     <!--@mbg.generated-->
     id, username, tiktok_code,phone, img,is_vip,gold_num,diamond_num,is_show, create_time, update_time, sort
   </sql>
+
+  <select id="queryGoldNum" resultType="int">
+    select
+        gold_num
+    from img_tiktok_user
+    where id = #{id} and is_show = "1"
+    </select>
 </mapper>

bnyer-services/bnyer-img/src/main/resources/com/bnyer/img/mapper/UserVipRecordMapper.xml

@@ -35,20 +35,21 @@
     a.update_time,
     a.sort
   </sql>
-  <select id="queryUserVipRecord" resultMap="BaseResultMap">
+
+  <select id="queryUserVipRecord" resultType="com.bnyer.img.vo.UserVipRecordVo">
     select
-        <include refid="Base_Column_List"/>
+    <include refid="Base_Column_List"/>
     FROM img_user_vip_record a
     WHERE
-        a.status = 1
+    a.status = 1
-        <if test="id != null">
+    <if test="id != null">
-          and a.id = #{id}
+      and a.id = #{id}
-        </if>
+    </if>
-        <if test="userId != null">
+    <if test="userId != null">
-          and a.user_id = #{userId}
+      and a.user_id = #{userId}
-        </if>
+    </if>
-        <if test="userClientType != null">
+    <if test="userClientType != null">
-          and a.user_client_type = #{userClientType}
+      and a.user_client_type = #{userClientType}
-        </if>
+    </if>
   </select>
 </mapper>

bnyer-services/bnyer-img/src/main/resources/com/bnyer/img/mapper/WxUserMapper.xml

@@ -21,4 +21,11 @@
     <!--@mbg.generated-->
     id, username, wx_code,phone, img,is_vip,gold_num,diamond_num, is_show, create_time, update_time, sort
   </sql>
+
+  <select id="queryGoldNum" resultType="int">
+    select
+        gold_num
+    from img_wx_user
+    where id = #{id} and is_show = "1"
+  </select>
 </mapper>