Procházet zdrojové kódy

fix: 代码生成器

jay před 1 rokem
rodič
revize
7f00e72495

+ 0 - 3
iot-module/iot-generator/src/main/resources/vm/java/bo.java.vm

@@ -2,9 +2,6 @@ package ${packageName}.dto.bo;
 
 import ${packageName}.model.${ClassName};
 import cc.iotkit.common.api.BaseDto;
-import cc.iotkit.common.api.PageRequest;
-import cc.iotkit.common.api.Paging;
-import cc.iotkit.common.api.Request;
 import cc.iotkit.common.validate.AddGroup;
 import cc.iotkit.common.validate.EditGroup;
 import io.github.linpeilie.annotations.AutoMapper;

+ 1 - 2
iot-module/iot-generator/src/main/resources/vm/java/controller.java.vm

@@ -6,7 +6,6 @@ import java.util.List;
 import io.swagger.annotations.ApiOperation;
 import lombok.RequiredArgsConstructor;
 import javax.servlet.http.HttpServletResponse;
-import jakarta.validation.constraints.*;
 import cn.dev33.satoken.annotation.SaCheckPermission;
 import org.springframework.web.bind.annotation.*;
 import org.springframework.validation.annotation.Validated;
@@ -49,7 +48,7 @@ public class ${ClassName}Controller extends BaseController {
         return ${className}Service.queryPageList(pageQuery);
     }
 #elseif($table.tree)
-    public List<${ClassName}Vo> list(Request<${ClassName}Bo> query) {
+    public List<${ClassName}Vo> list(@RequestBody Request<${ClassName}Bo> query) {
         List<${ClassName}Vo> list = ${className}Service.queryList(query.getData());
         return list;
     }

+ 27 - 24
iot-module/iot-generator/src/main/resources/vm/java/idataimpl.java.vm

@@ -5,13 +5,10 @@ import ${packageName}.data.I${ClassName}Data;
 import ${packageName}.data.model.Tb${ClassName};
 import ${packageName}.model.${ClassName};
 import java.util.List;
-import com.querydsl.core.QueryResults;
 import com.querydsl.core.types.Predicate;
-import com.querydsl.core.types.Projections;
 import com.querydsl.jpa.impl.JPAQueryFactory;
 import lombok.RequiredArgsConstructor;
 import org.springframework.context.annotation.Primary;
-import org.springframework.data.jpa.repository.JpaRepository;
 import org.springframework.stereotype.Service;
 import cc.iotkit.data.util.PredicateBuilder;
 
@@ -19,6 +16,11 @@ import cc.iotkit.common.api.PageRequest;
 import cc.iotkit.common.api.Paging;
 import cc.iotkit.common.utils.StringUtils;
 import java.util.Collection;
+import java.util.Objects;
+import java.util.Optional;
+import com.google.common.collect.Lists;
+
+
 import cc.iotkit.common.utils.MapstructUtils;
 import cc.iotkit.data.util.PageBuilder;
 
@@ -46,33 +48,35 @@ public class ${ClassName}DataImpl implements I${ClassName}Data {
 
     private Predicate buildQueryCondition(${ClassName} bo) {
         PredicateBuilder builder = PredicateBuilder.instance();
-
-        #foreach($column in $columns)
-            #if($column.query)
-                #set($queryType=$column.queryType)
-                #set($javaField=$column.javaField)
-                #set($javaType=$column.javaType)
-                #set($columnName=$column.columnName)
-                #set($AttrName=$column.javaField.substring(0,1).toUpperCase() + ${column.javaField.substring(1)})
-                #if($queryType != 'BETWEEN')
-                    #if($javaType == 'String')
-                        #set($condition='StringUtils.isNotBlank(bo.get'+$AttrName+'())')
+        if(Objects.nonNull(bo)) {
+
+            #foreach($column in $columns)
+                #if($column.query)
+                    #set($queryType=$column.queryType)
+                    #set($javaField=$column.javaField)
+                    #set($javaType=$column.javaType)
+                    #set($columnName=$column.columnName)
+                    #set($AttrName=$column.javaField.substring(0,1).toUpperCase() + ${column.javaField.substring(1)})
+                    #if($queryType != 'BETWEEN')
+                        #if($javaType == 'String')
+                            #set($condition='StringUtils.isNotBlank(bo.get'+$AttrName+'())')
+                        #else
+                            #set($condition='bo.get'+$AttrName+'() != null')
+                        #end
+                        builder.and($condition, () -> tb${ClassName}.${javaField}.eq(bo.get${AttrName}()));
                     #else
-                        #set($condition='bo.get'+$AttrName+'() != null')
+                        builder.and(params.get("begin$AttrName") != null && params.get("end$AttrName") != null,
+                                () -> tb${ClassName}.${javaField}.bettwen(params.get("begin$AttrName"), params.get("end$AttrName")));
                     #end
-                    builder.and($condition, ()->tb${ClassName}.${javaField}.eq(bo.get${AttrName}()));
-                #else
-                    builder.and(params.get("begin$AttrName") != null && params.get("end$AttrName") != null,
-                            ()->tb${ClassName}.${javaField}.bettwen(params.get("begin$AttrName"), params.get("end$AttrName")));
                 #end
             #end
-        #end
+        }
         return builder.build();
     }
 
     @Override
-    public List<${ClassName}> findByIds(Collection<Long> id) {
-        List allById = baseRepository.findAllById(id);
+    public List<${ClassName}> findByIds(Collection<Long> ids) {
+        List<Tb${ClassName}> allById = baseRepository.findAllById(ids);
         return MapstructUtils.convert(allById, ${ClassName}.class);
     }
 
@@ -100,8 +104,7 @@ public class ${ClassName}DataImpl implements I${ClassName}Data {
     @Override
     public ${ClassName} findById(Long id) {
         Tb${ClassName} ret = jpaQueryFactory.select(tb${ClassName}).from(tb${ClassName}).where(tb${ClassName}.id.eq(id)).fetchOne();
-        ${ClassName} convert = MapstructUtils.convert(ret, ${ClassName}.class);
-        return convert;
+        return MapstructUtils.convert(ret, ${ClassName}.class);
     }
 
     @Override

+ 0 - 2
iot-module/iot-generator/src/main/resources/vm/java/serviceImpl.java.vm

@@ -6,7 +6,6 @@ import cc.iotkit.common.utils.StringUtils;
 import cc.iotkit.common.api.PageRequest;
 import cc.iotkit.common.api.Paging;
 #end
-import cc.iotkit.common.api.Request;
 import lombok.RequiredArgsConstructor;
 import org.springframework.stereotype.Service;
 import ${packageName}.dto.bo.${ClassName}Bo;
@@ -16,7 +15,6 @@ import ${packageName}.service.I${ClassName}Service;
 import ${packageName}.data.I${ClassName}Data;
 
 import java.util.List;
-import java.util.Map;
 import java.util.Collection;
 import cc.iotkit.common.exception.BizException;