resource quota

jeecg-module-system/jeecg-system-biz/src/main/java/org/jeecg/modules/adweb/quota/controller/ResourceQuotaController.java

@@ -3,7 +3,9 @@ package org.jeecg.modules.adweb.quota.controller;
 import lombok.extern.slf4j.Slf4j;
 
 import org.jeecg.common.api.vo.Result;
-import org.jeecg.modules.adweb.quota.vo.ResourceQuota;
+import org.jeecg.modules.adweb.quota.service.IResourceQuotaService;
+import org.jeecg.modules.adweb.quota.vo.ResourceQuotaVO;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.*;
 
 /**
@@ -16,12 +18,11 @@ import org.springframework.web.bind.annotation.*;
 @Slf4j
 public class ResourceQuotaController {
 
+    @Autowired private IResourceQuotaService resourceQuotaService;
+
     @GetMapping("/getByUid")
     @ResponseBody
-    public Result<ResourceQuota> getByUid(@RequestParam String uid) {
-        ResourceQuota resourceQuota = new ResourceQuota();
-        resourceQuota.setUid(uid);
-
-        return Result.OK(resourceQuota);
+    public Result<ResourceQuotaVO> getByUid(@RequestParam String uid) {
+        return Result.OK(resourceQuotaService.getResourceQuotaByUid(uid));
     }
 }

jeecg-module-system/jeecg-system-biz/src/main/java/org/jeecg/modules/adweb/quota/entity/ResourceQuota.java

@@ -17,16 +17,16 @@ import org.springframework.format.annotation.DateTimeFormat;
 import java.io.Serializable;
 
 /**
- * @Description: adweb_resource_quote
+ * @Description: adweb_resource_quota
  * @Author: jeecg-boot
  * @Date: 2025-01-07
  * @Version: V1.0
  */
 @Data
-@TableName("adweb_resource_quote")
+@TableName("adweb_resource_quota")
 @Accessors(chain = true)
 @EqualsAndHashCode(callSuper = false)
-@Schema(description = "adweb_resource_quote")
+@Schema(description = "adweb_resource_quota")
 public class ResourceQuota implements Serializable {
     private static final long serialVersionUID = 1L;
 
@@ -43,7 +43,7 @@ public class ResourceQuota implements Serializable {
     /** 临时站点额度 */
     @Excel(name = "临时站点额度", width = 15)
     @Schema(description = "临时站点额度")
-    private Integer websitesQuota;
+    private Integer websiteQuota;
 
     /** AI算力额度 */
     @Excel(name = "AI算力额度", width = 15)
@@ -58,7 +58,7 @@ public class ResourceQuota implements Serializable {
     /** 外贸大数据用量 */
     @Excel(name = "外贸大数据用量", width = 15)
     @Schema(description = "外贸大数据用量")
-    private Integer customsDataUsed;
+    private Integer customsDataUsage;
 
     /** 创建时间 */
     @Excel(name = "创建时间", width = 20, format = "yyyy-MM-dd HH:mm:ss")

jeecg-module-system/jeecg-system-biz/src/main/java/org/jeecg/modules/adweb/quota/service/IResourceQuotaService.java

@@ -2,14 +2,16 @@ package org.jeecg.modules.adweb.quota.service;
 
 import com.baomidou.mybatisplus.extension.service.IService;
 
+import org.apache.commons.lang3.tuple.Pair;
 import org.jeecg.modules.adweb.quota.entity.ResourceQuota;
+import org.jeecg.modules.adweb.quota.vo.ResourceQuotaVO;
 
 /**
- * @Description: adweb_resource_quota
- * @Author: jeecg-boot
- * @Date: 2025-12-07
- * @Version: V1.0
+ * @Description: adweb_resource_quota @Author: jeecg-boot @Date: 2025-12-07 @Version: V1.0
  */
 public interface IResourceQuotaService extends IService<ResourceQuota> {
 
+    ResourceQuotaVO getResourceQuotaByUid(String uid);
+
+    Pair<Integer, Integer> getWebsiteQuota(String uid);
 }

jeecg-module-system/jeecg-system-biz/src/main/java/org/jeecg/modules/adweb/quota/service/impl/ResourceQuotaServiceImpl.java

@@ -1,20 +1,84 @@
 package org.jeecg.modules.adweb.quota.service.impl;
 
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 
+import lombok.extern.slf4j.Slf4j;
+
+import org.apache.commons.lang3.tuple.Pair;
+import org.apache.shiro.SecurityUtils;
+import org.jeecg.common.system.vo.LoginUser;
+import org.jeecg.common.util.FastJsonUtil;
 import org.jeecg.modules.adweb.quota.entity.ResourceQuota;
 import org.jeecg.modules.adweb.quota.mapper.ResourceQuotaMapper;
 import org.jeecg.modules.adweb.quota.service.IResourceQuotaService;
+import org.jeecg.modules.adweb.quota.vo.ResourceQuotaVO;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
 
+import java.util.Objects;
+
 /**
- * @Description: adweb_resource_quota
- * @Author: jeecg-boot
- * @Date: 2025-12-07
- * @Version: V1.0
+ * @Description: adweb_resource_quota @Author: jeecg-boot @Date: 2025-12-07 @Version: V1.0
  */
+@Slf4j
 @Service
 public class ResourceQuotaServiceImpl extends ServiceImpl<ResourceQuotaMapper, ResourceQuota>
         implements IResourceQuotaService {
 
+    @Value("${resource-quota.default-website-quota}")
+    private int defaultWebsiteQuota;
+
+    @Value("${resource-quota.default-ai-power-quota}")
+    private int defaultAiPowerQuota;
+
+    @Value("${resource-quota.default-customs-data-quota}")
+    private int defaultCustomsDataQuota;
+
+    public ResourceQuotaVO getResourceQuotaByUid(String uid) {
+        ResourceQuota resourceQuota =
+                this.getOne(new LambdaQueryWrapper<ResourceQuota>().eq(ResourceQuota::getUid, uid));
+        if (Objects.isNull(resourceQuota)) {
+            resourceQuota = this.createResourceQuota(uid);
+        }
+
+        ResourceQuotaVO resourceQuotaVO = new ResourceQuotaVO();
+        resourceQuotaVO.setUid(uid);
+        resourceQuotaVO.setWebsiteQuota(resourceQuota.getWebsiteQuota());
+        resourceQuotaVO.setWebsiteUsage(0);
+        resourceQuotaVO.setAiPowerQuota(resourceQuota.getAiPowerQuota());
+        resourceQuotaVO.setAiPowerUsage(0);
+        resourceQuotaVO.setCustomsDataQuota(resourceQuota.getCustomsDataQuota());
+        resourceQuotaVO.setCustomsDataUsage(resourceQuota.getCustomsDataUsage());
+
+        return resourceQuotaVO;
+    }
+
+    public Pair<Integer, Integer> getWebsiteQuota(String uid) {
+        return null;
+    }
+
+    /**
+     * 新增{@link ResourceQuota}记录
+     *
+     * @param uid
+     * @return
+     */
+    private ResourceQuota createResourceQuota(String uid) {
+        ResourceQuota resourceQuota = new ResourceQuota();
+        resourceQuota.setUid(uid);
+        resourceQuota.setWebsiteQuota(defaultWebsiteQuota);
+        resourceQuota.setAiPowerQuota(defaultAiPowerQuota);
+        resourceQuota.setCustomsDataQuota(defaultCustomsDataQuota);
+        resourceQuota.setCustomsDataUsage(0);
+
+        this.save(resourceQuota);
+        log.info(
+                "管理员 {} 为用户 {} 配置资源额度 {}",
+                ((LoginUser) SecurityUtils.getSubject().getPrincipal()).getUsername(),
+                uid,
+                FastJsonUtil.toJSONString(resourceQuota));
+
+        return resourceQuota;
+    }
 }

jeecg-module-system/jeecg-system-biz/src/main/java/org/jeecg/modules/adweb/quota/vo/ResourceQuota.java → jeecg-module-system/jeecg-system-biz/src/main/java/org/jeecg/modules/adweb/quota/vo/ResourceQuotaVO.java

@@ -6,15 +6,15 @@ import lombok.Data;
  * @author wfansh
  */
 @Data
-public class ResourceQuota {
+public class ResourceQuotaVO {
     private String uid;
 
-    private long websitesQuota;
-    private long websitesUsed;
+    private long websiteQuota;
+    private long websiteUsage;
 
     private long aiPowerQuota;
-    private long aiPowerUsed;
+    private long aiPowerUsage;
 
     private long customsDataQuota;
-    private long customsDataUsed;
+    private long customsDataUsage;
 }

jeecg-module-system/jeecg-system-start/src/main/resources/application-dev.yml

@@ -449,7 +449,13 @@ AdwebSiteConnect:
   username: ubuntu
   password: adweb123@2024
   tempDomain: v3-site.adwebcloud.com
-  tempCname: devci2.adwebcloud.com
 
+### WP临时服务器
 serverIp:
   test: 35.87.155.71
+
+### 资源额度
+resource-quota:
+  default-website-quota: 1
+  default-ai-power-quota: 10
+  default-customs-data-quota: 10

jeecg-module-system/jeecg-system-start/src/main/resources/application-prod.yml

@@ -453,7 +453,13 @@ AdwebSiteConnect:
   username: ubuntu
   password: adweb123@2024
   tempDomain: v3-site.adwebcloud.com
-  tempCname: devci2.adwebcloud.com
 
+### WP临时服务器
 serverIp:
   test: 35.87.155.71
+
+### 资源额度
+resource-quota:
+  default-website-quota: 3
+  default-ai-power-quota: 200
+  default-customs-data-quota: 100

jeecg-module-system/jeecg-system-start/src/main/resources/application-test.yml

@@ -450,7 +450,13 @@ AdwebSiteConnect:
   username: ubuntu
   password: adweb123@2024
   tempDomain: v3-site.adwebcloud.com
-  tempCname: devci2.adwebcloud.com
 
+### WP临时服务器
 serverIp:
   test: 35.87.155.71
+
+### 资源额度
+resource-quota:
+  default-website-quota: 1
+  default-ai-power-quota: 10
+  default-customs-data-quota: 10