添加新的代码规则模板

Author: davidfantasy

src-tauri/src/storage/init.sql

@@ -26,7 +26,7 @@ CREATE TABLE IF NOT EXISTS sys_config (
 
 -- 初始化规则示例
 INSERT INTO code_sample (id,name,content) VALUES
-	 ('dfebebb3-d75e-4ef7-8522-4db1cd2ba63c','mybatis-plus DO对象','### 数据库DO对象
+	 ('dfebebb3-d75e-4ef7-8522-4db1cd2ba63c','数据库DO对象','### 数据库DO对象
 1.对象名使用表名的驼峰格式，以DO结尾；
 2.采用mybatis-plus的entity对象规范；
 3.主键采用IdType.ASSIGN_ID的方式自动生成；
@@ -71,4 +71,146 @@ public class UserDTO {
     @Schema(description = "创建时间")
     private LocalDate createTime;
 }  
+~~~'),
+	 ('338ec139-df26-4d40-98f8-51a9a0a27ee8','Converter对象','### 接口DTO和数据库DO对象的互相转换的Converter对象
+1.  使用MapStruct框架，用于将数据库的DO对象转换为DTO对象，或者将DTO对象转换为数据库的DO对象；
+2.  对象的名称为XXXConverter（例如DeviceConverter ）
+3.  字段不完全匹配时，仅处理匹配的字段，忽略错误
+ 举例：
+~~~java
+import org.mapstruct.factory.Mappers;
+
+@Mapper(unmappedTargetPolicy = ReportingPolicy.IGNORE)
+public interface DeviceConverter {
+
+    DeviceConverter INSTANCE = Mappers.getMapper(DeviceConverter.class);
+
+    DeviceDO toDataObj(DeviceDTO dto);
+
+    DeviceDTO toDTO(DeviceDO deviceDO);
+
+}
+~~~
+
+'),
+	 ('5c5ff59e-9d6f-4ac9-81a1-304f95698ed8','数据库mapper对象','### 数据库mapper对象
+1. mybatis-plus的mapper接口，用于封装对某个DO对象的数据库操作；
+
+~~~java
+import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+
+public interface BaseDeviceCheckPointMapper extends BaseMapper<BaseDeviceCheckPointDO> {
+
+}
+~~~'),
+	 ('ec69f9a9-4607-4f97-bac8-ac0908f9fd94','Service对象','### Service
+
+1. 对象名使用表名的驼峰格式，以Service结尾，用于封装对数据库的相关业务操作，底层使用Myabtis-Plus的Mapper对象操作数据库；
+2. 覆盖数据库的增，删，改，按ID查询详情，分页查询列表的相关操作；
+3. create，edit接收DTO对象，并利用Converter对象转换成对应的DO对象，并调用Mapper对象进行数据库操作；
+4. 如果当前上下文中存在PageQuery对象的相关信息，则生成分页查询方法。该方法接收一个分页查询的PageQuery对象作为参数，进行分页查询，并返回分页后的查询结果（使用mybatis-plus的分页插件）；
+5. 所有涉及数据库的操作都需要添加事务注解；
+
+
+举例：
+~~~java
+
+@Service
+@RequiredArgsConstructor
+public class DeviceService {
+
+    private final DeviceMapper deviceMapper;
+
+    @Transactional
+    @Override
+    public String create(DeviceDTO command) {
+        DeviceDO deviceDO = DeviceConverter.INSTANCE.toDataObj(command);
+        deviceMapper.insert(deviceDO);
+        return deviceDO.getId();
+    }
+
+    @Transactional
+    @Override
+    public void delete(String id) {
+        deviceMapper.deleteById(id);
+    }
+
+    @Transactional
+    @Override
+    public void edit(DeviceDTO command) {
+        DeviceDO deviceDO = DeviceConverter.INSTANCE.toDataObj(command);
+        deviceMapper.updateById(orderDO);
+    }
+
+    @Override
+    public DeviceDTO detail(String id) {
+        DeviceDO deviceDO = deviceMapper.selectById(id);
+        if(deviceDO==null){
+            return null;
+        }
+        return DeviceConverter.INSTANCE.toDTO(deviceDO);
+    }
+
+    @Override
+    public IPage<DeviceDTO> page(DevicePageQuery query) {
+        var queryWrapper = Wrappers.<DeviceDO>lambdaQuery()
+                .like(StrUtil.isNotBlank(query.getOrderNo()), DeviceMaintenanceOrderDO::getOrderNo, query.getOrderNo())
+                .like(StrUtil.isNotBlank(query.getCustomerName()), DeviceMaintenanceOrderDO::getCustomerName, query.getCustomerName())
+                .like(StrUtil.isNotBlank(query.getDeviceName()), DeviceMaintenanceOrderDO::getDeviceName, query.getDeviceName());
+        return deviceMapper.selectPage(new Page<>(query.getCurrent(), query.getSize()),
+            queryWrapper).convert(DeviceMaintenanceOrderConverter.INSTANCE::toDTO);
+    }
+
+}
+~~~'),
+	 ('cb51e146-f673-470c-bb2b-694023cbd50a','分页接口查询对象','### 用于API 分页接口的查询对象
+1. 封装了需要'),
+	 ('43c93804-cd08-4fc1-b604-a9568491c32e','Controller对象','### 封装API接口的Controller对象
+1.  对象名使用表名的驼峰格式，以Controller结尾；
+2.  采用SpringBoot的相关技术框架，并满足RESTFUL规范的要求；
+3.  如果上下文中存在Service对象的定义，则根据Service对象的相关方法生成API接口, 否则不生成任何接口；
+4.  接口参数（DTO对象）应该采用Spring validation的规范进行校验
+5.   接口的API地址以“/api/${实体对象名}”开头，例如“/api/device”
+示例
+~~~java
+@RestController
+@RequestMapping("/api/device")
+@RequiredArgsConstructor
+public class DeviceController {
+
+    private final DeviceService deviceService;
+
+    /**
+     * 创建
+     */
+    @PostMapping
+    public void create(@Validated @RequestBody DeviceDTO deviceDTO) {
+        deviceService.create(deviceDTO);
+    }
+
+    /**
+     * 更新
+     */
+    @PutMapping
+    public void update(@Validated @RequestBody DeviceDTO deviceDTO) {
+        deviceService.update(deviceDTO);
+    }
+
+    /**
+     * 删除
+     */
+    @DeleteMapping("/{id}")
+    public void delete(@PathVariable String id) {
+        deviceService.delete(id);
+    }
+
+    /**
+     * 分页查询
+     */
+    @PostMapping("/page")
+    public IPage<DeviceDTO>  page(@RequestBody DevicePageQuery pageQuery) {
+        return   deviceService.page(pageQuery)
+    }
+
+}
 ~~~');

src-tauri/src/task/code_gen_task.rs

@@ -142,7 +142,6 @@ impl CodeGenTask {
             .await?;
         let context = LLM_CONTEXT_BUILDER.build(&self.req).await?;
         self.send_log(sender, "上下文已构建完成").await?;
-        println!("完整的上下文为：{}", context);
         Ok(context)
     }
 

src/services/RuleService.ts

@@ -13,8 +13,8 @@ export const ruleService = {
             const codeStandards = await invoke<Rule[]>('get_all_samples');
             return codeStandards;
         } catch (e) {
-            ElMessage.error('查询代码规范失败:' + e);
-            throw new Error('查询代码规范失败');
+            ElMessage.error('查询规则失败:' + JSON.stringify(e));
+            throw e;
         }
     },
 
@@ -28,9 +28,8 @@ export const ruleService = {
             });
             return { ...data, id: newId };
         } catch (e) {
-            console.log(e);
-            ElMessage.error('创建代码规范发生错误:' + e);
-            throw new Error('创建代码规范失败');
+            ElMessage.error('创建规则发生错误:' + JSON.stringify(e));
+            throw e;
         }
     },
 
@@ -39,7 +38,7 @@ export const ruleService = {
             const item = await invoke<Rule>('get_sample_by_id', { id });
             return item;
         } catch (e) {
-            ElMessage.error('查询代码规范发生错误:' + e);
+            ElMessage.error('查询规则发生错误:' + JSON.stringify(e));
             return null;
         }
     },
@@ -50,7 +49,7 @@ export const ruleService = {
                 cs: data
             });
         } catch (e) {
-            ElMessage.error('更新代码规范失败:' + e);
+            ElMessage.error('更新规则失败:' + JSON.stringify(e));
             throw e;
         }
     },
@@ -59,7 +58,7 @@ export const ruleService = {
         try {
             return await invoke<boolean>('delete_sample', { id });
         } catch (e) {
-            ElMessage.error('删除代码规范发生错误:' + e);
+            ElMessage.error('删除规则发生错误:' + JSON.stringify(e));
             return false;
         }
     }