fix

netflix-dao/src/main/java/com/cyksj/model/views/MidjourneyUserView.java

@@ -0,0 +1,40 @@
+package com.cyksj.model.views;
+
+import com.ejlchina.searcher.bean.DbField;
+import com.ejlchina.searcher.bean.SearchBean;
+import lombok.Data;
+
+import java.util.Date;
+
+/**
+ * @author zwhui
+ * @date 2024/4/28 17:43
+ */
+@Data
+@SearchBean(tables = "midjourney_user mu left join groups_relation gr on mu.relation_id = gr.id ")
+public class MidjourneyUserView {
+
+
+    @DbField("mu.id")
+    private Long id;
+
+    @DbField("mu.relation_id")
+    private String relationId;
+
+    @DbField("gr.user_id")
+    private Long userId;
+
+    @DbField("mu.mj_fast_num")
+    private Integer mjFastNum;
+
+    @DbField("mu.mj_relax_num")
+    private Integer mjRelaxNum;
+
+    @DbField("mu.expire_time")
+    private Date expireTime;
+
+
+
+
+
+}

netflix-web/src/main/java/com/cyksj/web/controller/manage/CmsOrderController.java

@@ -917,4 +917,14 @@ public class CmsOrderController {
 		cmsOrderDonService.unifiedUpgradeRefund(refundReq);
 		return GatewayResponse.SUCCESS.newBuilder().toResult("退款成功");
 	}
+
+	/**
+	 * midjourney查询订单用户账号剩余次数
+	 */
+	@GetMapping("/get/midjourney/user")
+	public Result<SearchResult<MidjourneyUserView>> getMidjourneyUser(Long userId,Long relationId) {
+		SearchResult<MidjourneyUserView> search = beanSearcher.search(MidjourneyUserView.class, MapUtils.flatBuilder(request.getParameterMap())
+				.field(MidjourneyUserView::getUserId,userId).field(MidjourneyUserView::getRelationId,relationId).build());
+		return GatewayResponse.SUCCESS.newBuilder().toResult(search);
+	}
 }

netflix-web/src/main/java/com/cyksj/web/controller/mirror/MidjourneyController.java

@@ -201,4 +201,15 @@ public class MidjourneyController {
         log.info("取消任务 id:{}",id);
         return GatewayResponse.SUCCESS.newBuilder().toResult(midjourneyService.cancelConversation(user,id));
     }
+
+
+    /**
+     * 修改midjourney账号模式
+     */
+    @PutMapping("/put/midjourney/user/{mode}")
+    public Result<String> putMidjourneyUserMode(@PathVariable Integer mode, Long userId) {
+        log.info("修改midjourney账号模式 userId:{},mode:{}", userId, mode);
+        midjourneyUserMapper.update(null,Wrappers.lambdaUpdate(MidjourneyUser.class).set(MidjourneyUser::getMode, mode).eq(MidjourneyUser::getId, userId));
+        return GatewayResponse.SUCCESS.newBuilder().toResult();
+    }
 }

netflix-web/src/main/java/com/cyksj/web/controller/payment/OrderController.java

@@ -34,6 +34,7 @@ import com.cyksj.model.request.AlipayOrderDonReq;
 import com.cyksj.model.request.OrderPayRequest;
 import com.cyksj.model.response.AliPayTradeStatus;
 import com.cyksj.model.views.GoodsDonSkuView;
+import com.cyksj.model.views.MidjourneyUserView;
 import com.cyksj.model.views.OrderCommentView;
 import com.cyksj.service.order.OrderDonService;
 import com.cyksj.service.paypal.PayPalService;
@@ -740,4 +741,14 @@ public class OrderController {
         orderDonService.closeUpgradeOrder(orderId, userId);
         return GatewayResponse.SUCCESS.newBuilder().toResult();
     }
+
+    /**
+     * midjourney查询订单用户账号剩余次数
+     */
+    @GetMapping("/get/midjourney/user")
+    public Result<SearchResult<MidjourneyUserView>> getMidjourneyUser(Long userId, Long relationId) {
+        SearchResult<MidjourneyUserView> search = beanSearcher.search(MidjourneyUserView.class, MapUtils.flatBuilder(request.getParameterMap())
+                .field(MidjourneyUserView::getUserId, userId).field(MidjourneyUserView::getRelationId, relationId).build());
+        return GatewayResponse.SUCCESS.newBuilder().toResult(search);
+    }
 }