From 881f273bca0f94345efb2c0cac52279af4a7f4e4 Mon Sep 17 00:00:00 2001
From: admin <weikou2014>
Date: 星期五, 24 九月 2021 15:16:08 +0800
Subject: [PATCH] 后台管理自动化代码生成

---
 src/main/java/org/yeshi/utils/generater/entity/MongoDBDaoData.java |   22 ++++++++++++++++++++++
 1 files changed, 22 insertions(+), 0 deletions(-)

diff --git a/src/main/java/org/yeshi/utils/generater/entity/MongoDBDaoData.java b/src/main/java/org/yeshi/utils/generater/entity/MongoDBDaoData.java
index 10f5e97..21637a9 100644
--- a/src/main/java/org/yeshi/utils/generater/entity/MongoDBDaoData.java
+++ b/src/main/java/org/yeshi/utils/generater/entity/MongoDBDaoData.java
@@ -2,6 +2,7 @@
 
 import org.springframework.data.annotation.Id;
 import org.springframework.data.annotation.Transient;
+import org.yeshi.utils.generater.annotation.admin.DaoQueryCondition;
 
 import java.lang.annotation.Annotation;
 import java.lang.reflect.Field;
@@ -19,6 +20,8 @@
 
     private ColumnData identityColumn;
     private List<ColumnData> columnDataList;
+
+    private List<DaoQueryColumnData> daoQueryColumnDataList;
 
     public static class Builder {
 
@@ -62,6 +65,7 @@
             String identityColumnName = null;
             String identityColumnType = null;
             List<ColumnData> columlist = new ArrayList<>();
+            List<DaoQueryColumnData> daoQueryColumnList = new ArrayList<>();
             for (Field fd : fields) {
                 //private/public 闈瀎inal锛宻tatic灞炴�ф墠浼氬弬涓庤В鏋�
                 if (fd.getModifiers() != 1 && fd.getModifiers() != 2) {
@@ -89,6 +93,14 @@
                     //涓嶅弬涓庤В鏋�
                     if (a instanceof Transient || a instanceof java.beans.Transient) {
                         columName = null;
+                        break;
+                    }
+                }
+
+                //鑾峰彇鏌ヨ鏉′欢
+                for (Annotation a : as) {
+                    if (a instanceof DaoQueryCondition) {
+                        daoQueryColumnList.add(new DaoQueryColumnData(columName, columnDataType, ((DaoQueryCondition) a).queryType()));
                         break;
                     }
                 }
@@ -121,6 +133,7 @@
             MongoDBDaoData data = new MongoDBDaoData();
             data.setBaseDaoClassName(baseDaoClass.getSimpleName());
             data.setColumnDataList(columnDataList);
+            data.setDaoQueryColumnDataList(daoQueryColumnList);
             data.setDaoName(daoName);
             data.setEntityClass(entityClass.getName());
             data.setEntityClassName(entityClass.getSimpleName());
@@ -185,4 +198,13 @@
     public void setImportClasses(Set<String> importClasses) {
         this.importClasses = importClasses;
     }
+
+
+    public List<DaoQueryColumnData> getDaoQueryColumnDataList() {
+        return daoQueryColumnDataList;
+    }
+
+    public void setDaoQueryColumnDataList(List<DaoQueryColumnData> daoQueryColumnDataList) {
+        this.daoQueryColumnDataList = daoQueryColumnDataList;
+    }
 }

--
Gitblit v1.8.0