From 2896cd7dc7c3f72b1fc2fd4dd330259400a856f9 Mon Sep 17 00:00:00 2001
From: Penny <2500338766@qq.com>
Date: Tue, 23 May 2023 00:22:19 +0800
Subject: [PATCH] feature:fixbug

---
 .../java/com/bnyer/img/controller/FhMiniController.java     | 6 +++---
 .../java/com/bnyer/img/controller/TiktokMiniController.java | 6 +++---
 .../java/com/bnyer/img/controller/WxMiniController.java     | 6 +++---
 .../src/main/java/com/bnyer/img/service/PromptService.java  | 2 +-
 .../java/com/bnyer/img/service/impl/PromptServiceimpl.java  | 6 +++---
 5 files changed, 13 insertions(+), 13 deletions(-)

diff --git a/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/controller/FhMiniController.java b/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/controller/FhMiniController.java
index d27439f..e18e4c4 100644
--- a/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/controller/FhMiniController.java
+++ b/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/controller/FhMiniController.java
@@ -404,9 +404,9 @@ public class FhMiniController extends BaseController {
     }
 
     @ApiOperation(value="获取提示词列表")
-    @GetMapping(value = "/getPrompt")
-    public AjaxResult getPrompt(){
-        return AjaxResult.success(promptService.queryPromptList());
+    @GetMapping(value = "/getPrompt/{type}")
+    public AjaxResult getPrompt(@PathVariable @ApiParam("提示词类型") String type){
+        return AjaxResult.success(promptService.queryPromptList(type));
     }
 
     @ApiOperation(value="文生图")
diff --git a/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/controller/TiktokMiniController.java b/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/controller/TiktokMiniController.java
index 7d1a0fc..5e4f423 100644
--- a/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/controller/TiktokMiniController.java
+++ b/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/controller/TiktokMiniController.java
@@ -426,9 +426,9 @@ public class TiktokMiniController extends BaseController {
     }
 
     @ApiOperation(value="获取提示词列表")
-    @GetMapping(value = "/getPrompt")
-    public AjaxResult getPrompt(){
-        return AjaxResult.success(promptService.queryPromptList());
+    @GetMapping(value = "/getPrompt/{type}")
+    public AjaxResult getPrompt(@PathVariable @ApiParam("提示词类型") String type){
+        return AjaxResult.success(promptService.queryPromptList(type));
     }
 
     @ApiOperation(value="文生图")
diff --git a/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/controller/WxMiniController.java b/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/controller/WxMiniController.java
index 30674f9..198c5c4 100644
--- a/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/controller/WxMiniController.java
+++ b/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/controller/WxMiniController.java
@@ -431,9 +431,9 @@ public class WxMiniController extends BaseController {
     }
 
     @ApiOperation(value="获取提示词列表")
-    @GetMapping(value = "/getPrompt")
-    public AjaxResult getPrompt(){
-        return AjaxResult.success(promptService.queryPromptList());
+    @GetMapping(value = "/getPrompt/{type}")
+    public AjaxResult getPrompt(@PathVariable @ApiParam("提示词类型") String type){
+        return AjaxResult.success(promptService.queryPromptList(type));
     }
 
     @ApiOperation(value="文生图")
diff --git a/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/PromptService.java b/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/PromptService.java
index 22086d8..368f3c0 100644
--- a/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/PromptService.java
+++ b/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/PromptService.java
@@ -10,5 +10,5 @@ public interface PromptService {
      * 随机获取8条提示词列表
      * @return -
      */
-    List<PromptVo> queryPromptList();
+    List<PromptVo> queryPromptList(String type);
 }
diff --git a/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/impl/PromptServiceimpl.java b/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/impl/PromptServiceimpl.java
index 814a07f..117ec28 100644
--- a/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/impl/PromptServiceimpl.java
+++ b/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/impl/PromptServiceimpl.java
@@ -18,9 +18,9 @@ public class PromptServiceimpl implements PromptService {
     private PromptMapper promptMapper;
 
     @Override
-    public List<PromptVo> queryPromptList() {
-        //type:0->绘画;1->gpt
-        List<PromptVo> promptVos = promptMapper.queryList("0");
+    public List<PromptVo> queryPromptList(String type) {
+        //type:0->绘画;1->gpt;2->情侣头像
+        List<PromptVo> promptVos = promptMapper.queryList(type);
         List<PromptVo> news = new ArrayList<>();
         Map<Integer,String> map = new HashMap<>();
         //随机抽10条数据展示在页面上