小程序修改

ruoyi-system/src/main/java/com/ruoyi/system/service/impl/UsageRecordServiceImpl.java

@@ -5,18 +5,16 @@ import java.util.*;
 import java.util.stream.Collectors;
 
 import com.ruoyi.common.core.domain.AjaxResult;
+import com.ruoyi.common.core.domain.entity.PolicyMenu;
 import com.ruoyi.common.utils.DateUtils;
 import com.ruoyi.common.utils.StringUtils;
 import com.ruoyi.system.domain.CreditUser;
 import com.ruoyi.system.domain.InterestsNumberTable;
 import com.ruoyi.system.domain.InterestsTable;
-import com.ruoyi.system.mapper.CreditUserMapper;
-import com.ruoyi.system.mapper.InterestsNumberTableMapper;
-import com.ruoyi.system.mapper.InterestsTableMapper;
+import com.ruoyi.system.mapper.*;
 import lombok.SneakyThrows;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
-import com.ruoyi.system.mapper.UsageRecordMapper;
 import com.ruoyi.system.domain.UsageRecord;
 import com.ruoyi.system.service.IUsageRecordService;
 
@@ -39,6 +37,9 @@ public class UsageRecordServiceImpl implements IUsageRecordService
     private InterestsNumberTableMapper interestsNumberTableMapper;
     @Autowired
     private InterestsTableMapper interestsTableMapper;
+    @Autowired
+    private PolicyMenuMapper policyMenuMapper;
+
 
     /**
      * 查询使用记录
@@ -109,6 +110,16 @@ public class UsageRecordServiceImpl implements IUsageRecordService
         usageRecord.setCreditName(creditUser.getUserName());
         usageRecord.setCreateTime(DateUtils.getNowDate());
 
+        //根据分类id查询一级标题（记录类型：record_type）和二级标题（记录名称：record_name）
+        PolicyMenu policyMenu = policyMenuMapper.selectPolicyMenuByMenuId(usageRecord.getPolicyId());
+        if(Objects.nonNull(policyMenu)){
+            usageRecord.setRecordName(policyMenu.getMenuName());
+            if(policyMenu.getParentId()!=0L){
+                PolicyMenu policyMenup = policyMenuMapper.selectPolicyMenuByMenuId(policyMenu.getParentId());
+                usageRecord.setRecordType(policyMenup.getMenuName());
+            }
+        }
+
         int c = usageRecordMapper.insertUsageRecord(usageRecord);
 
         return c > 0 ? AjaxResult.success() : AjaxResult.error();