toOptional();
-
- /**
- * Factory method for failure.
- *
- * @param e throwable to create the failed Try with
- * @param Type
- * @return a new Failure
- */
-
- public static Try failure(Throwable e) {
- return new Failure<>(e);
- }
-
- /**
- * Factory method for success.
- *
- * @param x value to create the successful Try with
- * @param Type
- * @return a new Success
- */
- public static Try successful(U x) {
- return new Success<>(x);
- }
+
+ protected Try() {
+ }
+
+ public static Try ofFailable(TrySupplier f) {
+ Objects.requireNonNull(f);
+
+ try {
+ return Try.successful(f.get());
+ } catch (Throwable t) {
+ return Try.failure(t);
+ }
+ }
+
+ /**
+ * Transform success or pass on failure.
+ * Takes an optional type parameter of the new type.
+ * You need to be specific about the new type if changing type
+ *
+ * Try.ofFailable(() -> "1").<Integer>map((x) -> Integer.valueOf(x))
+ *
+ * @param f function to apply to successful value.
+ * @param new type (optional)
+ * @return Success<U> or Failure<U>
+ */
+
+ public abstract Try map(TryMapFunction f);
+
+ /**
+ * Transform success or pass on failure, taking a Try<U> as the result.
+ * Takes an optional type parameter of the new type.
+ * You need to be specific about the new type if changing type.
+ *
+ * Try.ofFailable(() -> "1").<Integer>flatMap((x) -> Try.ofFailable(() -> Integer.valueOf(x)))
+ * returns Integer(1)
+ *
+ * @param f function to apply to successful value.
+ * @param new type (optional)
+ * @return new composed Try
+ */
+ public abstract Try flatMap(TryMapFunction> f);
+
+ /**
+ * Specifies a result to use in case of failure.
+ * Gives access to the exception which can be pattern matched on.
+ *
+ * Try.ofFailable(() -> "not a number")
+ * .<Integer>flatMap((x) -> Try.ofFailable(() ->Integer.valueOf(x)))
+ * .recover((t) -> 1)
+ * returns Integer(1)
+ *
+ * @param f function to execute on successful result.
+ * @return new composed Try
+ */
+
+ public abstract T recover(Function f);
+
+ /**
+ * Try applying f(t) on the case of failure.
+ *
+ * @param f function that takes throwable and returns result
+ * @return a new Try in the case of failure, or the current Success.
+ */
+ public abstract Try recoverWith(TryMapFunction> f);
+
+ /**
+ * Return a value in the case of a failure.
+ * This is similar to recover but does not expose the exception type.
+ *
+ * @param value return the try's value or else the value specified.
+ * @return new composed Try
+ */
+ public abstract T orElse(T value);
+
+ /**
+ * Return another try in the case of failure.
+ * Like recoverWith but without exposing the exception.
+ *
+ * @param f return the value or the value from the new try.
+ * @return new composed Try
+ */
+ public abstract Try orElseTry(TrySupplier f);
+
+ /**
+ * Gets the value T on Success or throws the cause of the failure.
+ *
+ * @return T
+ * @throws Throwable produced by the supplier function argument
+ */
+
+ public abstract T orElseThrow(Supplier exceptionSupplier) throws X;
+
+ /**
+ * Gets the value T on Success or throws the cause of the failure.
+ *
+ * @return T
+ * @throws Throwable
+ */
+ public abstract T get() throws Throwable;
+
+ /**
+ * Gets the value T on Success or throws the cause of the failure wrapped into a RuntimeException
+ *
+ * @return T
+ * @throws RuntimeException
+ */
+ public abstract T getUnchecked();
+
+ public abstract boolean isSuccess();
+
+ /**
+ * Performs the provided action, when successful
+ *
+ * @param action action to run
+ * @return new composed Try
+ * @throws E if the action throws an exception
+ */
+ public abstract Try onSuccess(TryConsumer action) throws E;
+
+ /**
+ * Performs the provided action, when failed
+ *
+ * @param action action to run
+ * @return new composed Try
+ * @throws E if the action throws an exception
+ */
+ public abstract Try onFailure(TryConsumer action) throws E;
+
+ /**
+ * If a Try is a Success and the predicate holds true, the Success is passed further.
+ * Otherwise (Failure or predicate doesn't hold), pass Failure.
+ *
+ * @param pred predicate applied to the value held by Try
+ * @return For Success, the same success if predicate holds true, otherwise Failure
+ */
+ public abstract Try filter(Predicate pred);
+
+ /**
+ * Try contents wrapped in Optional.
+ *
+ * @return Optional of T, if Success, Empty if Failure or null value
+ */
+ public abstract Optional toOptional();
+
+ /**
+ * Factory method for failure.
+ *
+ * @param e throwable to create the failed Try with
+ * @param Type
+ * @return a new Failure
+ */
+
+ public static Try failure(Throwable e) {
+ return new Failure<>(e);
+ }
+
+ /**
+ * Factory method for success.
+ *
+ * @param x value to create the successful Try with
+ * @param Type
+ * @return a new Success
+ */
+ public static Try successful(U x) {
+ return new Success<>(x);
+ }
}
class Success extends Try {
- private final T value;
-
- public Success(T value) {
- this.value = value;
- }
-
- @Override
- public Try flatMap(TryMapFunction> f) {
- Objects.requireNonNull(f);
- try {
- return f.apply(value);
- } catch (Throwable t) {
- return Try.failure(t);
- }
- }
-
- @Override
- public T recover(Function f) {
- Objects.requireNonNull(f);
- return value;
- }
-
- @Override
- public Try recoverWith(TryMapFunction> f) {
- Objects.requireNonNull(f);
- return this;
- }
-
- @Override
- public T orElse(T value) {
- return this.value;
- }
-
- @Override
- public Try orElseTry(TrySupplier f) {
- Objects.requireNonNull(f);
- return this;
- }
-
- @Override
- public T orElseThrow(Supplier exceptionSupplier) throws X {
- return value;
- }
-
- @Override
- public T get() throws Throwable {
- return value;
- }
-
- @Override
- public T getUnchecked() {
- return value;
- }
-
- @Override
- public Try map(TryMapFunction f) {
- Objects.requireNonNull(f);
- try {
- return new Success<>(f.apply(value));
- } catch (Throwable t) {
- return Try.failure(t);
- }
- }
-
- @Override
- public boolean isSuccess() {
- return true;
- }
-
- @Override
- public Try onSuccess(TryConsumer action) throws E {
- action.accept(value);
- return this;
- }
-
- @Override
- public Try filter(Predicate p) {
- Objects.requireNonNull(p);
-
- if (p.test(value)) {
- return this;
- } else {
- return Try.failure(new NoSuchElementException("Predicate does not match for " + value));
- }
- }
-
- @Override
- public Optional toOptional() {
- return Optional.ofNullable(value);
- }
-
- @Override
- public Try onFailure(TryConsumer action) {
- return this;
- }
+ private final T value;
+
+ public Success(T value) {
+ this.value = value;
+ }
+
+ @Override
+ public Try flatMap(TryMapFunction> f) {
+ Objects.requireNonNull(f);
+ try {
+ return f.apply(value);
+ } catch (Throwable t) {
+ return Try.failure(t);
+ }
+ }
+
+ @Override
+ public T recover(Function f) {
+ Objects.requireNonNull(f);
+ return value;
+ }
+
+ @Override
+ public Try recoverWith(TryMapFunction> f) {
+ Objects.requireNonNull(f);
+ return this;
+ }
+
+ @Override
+ public T orElse(T value) {
+ return this.value;
+ }
+
+ @Override
+ public Try orElseTry(TrySupplier f) {
+ Objects.requireNonNull(f);
+ return this;
+ }
+
+ @Override
+ public T orElseThrow(Supplier exceptionSupplier) throws X {
+ return value;
+ }
+
+ @Override
+ public T get() throws Throwable {
+ return value;
+ }
+
+ @Override
+ public T getUnchecked() {
+ return value;
+ }
+
+ @Override
+ public Try map(TryMapFunction f) {
+ Objects.requireNonNull(f);
+ try {
+ return new Success<>(f.apply(value));
+ } catch (Throwable t) {
+ return Try.failure(t);
+ }
+ }
+
+ @Override
+ public boolean isSuccess() {
+ return true;
+ }
+
+ @Override
+ public Try onSuccess(TryConsumer action) throws E {
+ action.accept(value);
+ return this;
+ }
+
+ @Override
+ public Try filter(Predicate p) {
+ Objects.requireNonNull(p);
+
+ if (p.test(value)) {
+ return this;
+ } else {
+ return Try.failure(new NoSuchElementException("Predicate does not match for " + value));
+ }
+ }
+
+ @Override
+ public Optional toOptional() {
+ return Optional.ofNullable(value);
+ }
+
+ @Override
+ public Try onFailure(TryConsumer action) {
+ return this;
+ }
}
class Failure extends Try {
- private final Throwable e;
-
- Failure(Throwable e) {
- this.e = e;
- }
-
- @Override
- public Try map(TryMapFunction f) {
- Objects.requireNonNull(f);
- return Try.failure(e);
- }
-
- @Override
- public Try flatMap(TryMapFunction> f) {
- Objects.requireNonNull(f);
- return Try.failure(e);
- }
-
- @Override
- public T recover(Function f) {
- Objects.requireNonNull(f);
- return f.apply(e);
- }
-
- @Override
- public Try recoverWith(TryMapFunction> f) {
- Objects.requireNonNull(f);
- try{
- return f.apply(e);
- }catch(Throwable t){
- return Try.failure(t);
- }
- }
-
- @Override
- public T orElse(T value) {
- return value;
- }
-
- @Override
- public Try orElseTry(TrySupplier f) {
- Objects.requireNonNull(f);
- return Try.ofFailable(f);
- }
-
- @Override
- public T orElseThrow(Supplier exceptionSupplier) throws X {
- throw exceptionSupplier.get();
- }
-
- @Override
- public T get() throws Throwable {
- throw e;
- }
-
- @Override
- public T getUnchecked() {
- throw new RuntimeException(e);
- }
-
- @Override
- public boolean isSuccess() {
- return false;
- }
-
- @Override
- public Try onSuccess(TryConsumer action) {
- return this;
- }
-
- @Override
- public Try filter(Predicate pred) {
- return this;
- }
-
- @Override
- public Optional toOptional() {
- return Optional.empty();
- }
-
- @Override
- public Try onFailure(TryConsumer action) throws E {
- action.accept(e);
- return this;
- }
+ private final Throwable e;
+
+ Failure(Throwable e) {
+ this.e = e;
+ }
+
+ @Override
+ public Try map(TryMapFunction f) {
+ Objects.requireNonNull(f);
+ return Try.failure(e);
+ }
+
+ @Override
+ public Try flatMap(TryMapFunction> f) {
+ Objects.requireNonNull(f);
+ return Try.failure(e);
+ }
+
+ @Override
+ public T recover(Function f) {
+ Objects.requireNonNull(f);
+ return f.apply(e);
+ }
+
+ @Override
+ public Try recoverWith(TryMapFunction> f) {
+ Objects.requireNonNull(f);
+ try {
+ return f.apply(e);
+ } catch (Throwable t) {
+ return Try.failure(t);
+ }
+ }
+
+ @Override
+ public T orElse(T value) {
+ return value;
+ }
+
+ @Override
+ public Try orElseTry(TrySupplier f) {
+ Objects.requireNonNull(f);
+ return Try.ofFailable(f);
+ }
+
+ @Override
+ public T orElseThrow(Supplier exceptionSupplier) throws X {
+ throw exceptionSupplier.get();
+ }
+
+ @Override
+ public T get() throws Throwable {
+ throw e;
+ }
+
+ @Override
+ public T getUnchecked() {
+ throw new RuntimeException(e);
+ }
+
+ @Override
+ public boolean isSuccess() {
+ return false;
+ }
+
+ @Override
+ public Try onSuccess(TryConsumer action) {
+ return this;
+ }
+
+ @Override
+ public Try filter(Predicate pred) {
+ return this;
+ }
+
+ @Override
+ public Optional toOptional() {
+ return Optional.empty();
+ }
+
+ @Override
+ public Try onFailure(TryConsumer action) throws E {
+ action.accept(e);
+ return this;
+ }
}
\ No newline at end of file
diff --git a/src/main/java/aiyh/utils/httpUtil/HttpManager.java b/src/main/java/aiyh/utils/httpUtil/HttpManager.java
index d01fffc..41a7222 100644
--- a/src/main/java/aiyh/utils/httpUtil/HttpManager.java
+++ b/src/main/java/aiyh/utils/httpUtil/HttpManager.java
@@ -75,10 +75,11 @@ public class HttpManager {
return true;
}).build();
HostnameVerifier hostnameVerifier = NoopHostnameVerifier.INSTANCE;
- sslsf = new SSLConnectionSocketFactory(sslContext,
- new String[]{"TLSv1"},
- null,
- hostnameVerifier);
+ sslsf = new SSLConnectionSocketFactory(sslContext, hostnameVerifier);
+ // sslsf = new SSLConnectionSocketFactory(sslContext,
+ // new String[]{"TLSv1"},
+ // null,
+ // hostnameVerifier);
} catch (NoSuchAlgorithmException | KeyStoreException | KeyManagementException e) {
throw new RuntimeException(e);
}
diff --git a/src/main/java/aiyh/utils/mapper/UtilMapper.java b/src/main/java/aiyh/utils/mapper/UtilMapper.java
index e9f3960..07170c3 100644
--- a/src/main/java/aiyh/utils/mapper/UtilMapper.java
+++ b/src/main/java/aiyh/utils/mapper/UtilMapper.java
@@ -2,6 +2,7 @@ package aiyh.utils.mapper;
import aiyh.utils.annotation.recordset.*;
import aiyh.utils.entity.DocImageInfo;
+import aiyh.utils.entity.FieldViewInfo;
import aiyh.utils.entity.SelectValueEntity;
import aiyh.utils.entity.WorkflowNodeConfig;
@@ -17,139 +18,158 @@ import java.util.Map;
@SqlMapper
public interface UtilMapper {
-
- /**
- * 查询日志级别是否开启Debug模式
- *
- * @return 是否开启Debug模式
- */
- @Select("select param_value from $t{configTableName} where only_mark = 'enableDebugLog'")
- public Boolean selectLogLevel(@ParamMapper("configTableName") String configTableName);
-
-
- /**
- * 根据唯一标识查询参数值
- *
- * @param onlyMark 唯一标识
- * @return 参数值
- */
- @Select("select param_value from $t{configTableName} where only_mark = #{onlyMark} and enable_param = 1")
- public String selectCusConfigParam(@ParamMapper("onlyMark") String onlyMark,
- @ParamMapper("configTableName") String cusConfigTableName);
-
-
- /**
- * 查询文件名
- *
- * @param imageFileId 查询文件名
- * @return 文件名
- */
- @Select("select imagefilename from imagefile where imagefileid = #{imageFileId}")
- String selectFileNameByImageFileId(@ParamMapper("imageFileId") int imageFileId);
-
- /**
- * 查询流程主表
- *
- * @param workflowId 流程id
- * @return 流程表名
- */
- @Select("select bill.tablename from workflow_bill bill join workflow_base base on base.formid = bill.id where base.id = #{workflowId}")
- String selectWorkfowMainTable(@ParamMapper("workflowId") String workflowId);
-
-
- @Select("select id,docid doc_id,imagefileid image_file_id,imagefilename image_file_name from docimagefile where docid = #{docId}")
- DocImageInfo selectDocImageInfo(@ParamMapper("docId") String docId);
-
- @Select("select id,docid doc_id,imagefileid image_file_id,imagefilename image_file_name from docimagefile where docid in ($t{docIds})")
- List selectDocImageInfos(@ParamMapper("docIds") String docIds);
-
- @Select("select id,docid doc_id,imagefileid image_file_id,imagefilename image_file_name from docimagefile where docid in (${docIds})")
- List selectDocImageInfos(@ParamMapper("docIds") String[] docIds);
-
-
- @Select("select id,workflow_type,mark_only,workflow_nodes,enable_nodes from uf_node_config where enable_nodes = 1 and workflow_type in ($t{allVersion}) and mark_only = #{markOnly}")
- WorkflowNodeConfig selectNodeConfig(@ParamMapper("allVersion") String allVersion, @ParamMapper("markOnly") String markOnly);
-
- /**
- * 查询下拉框值
- *
- * @param tableName 表明
- * @param fileName 字段名
- * @return 下拉框
- */
- @Select("select wbf.id,wbf.fieldname,wbf.fieldlabel,wb.tablename, ws.selectname,ws.selectvalue " +
- "from workflow_billfield wbf left join workflow_bill wb on wbf.billid = wb.id " +
- "left join workflow_selectitem ws on ws.fieldid = wbf.id where wb.tablename = #{tableName} and fieldname = #{fileName}")
- List selectSelectFieldValue(@ParamMapper("tableName") String tableName, @ParamMapper("fileName") String fileName);
-
- @Select("select ws.selectname " +
- "from workflow_billfield wbf left join workflow_bill wb on wbf.billid = wb.id " +
- "left join workflow_selectitem ws on ws.fieldid = wbf.id where wb.tablename = #{tableName} and fieldname = #{fileName} and selectvalue = #{value}")
- String selectSelectFieldValueByValue(@ParamMapper("tableName") String tableName, @ParamMapper("fileName") String fileName,
- @ParamMapper("value") String value);
-
- /**
- * 查询文件信息
- *
- * @param imageFileId 查询文件名
- * @return 文件名
- */
- @Select("select * from imagefile where imagefileid = #{imageFileId}")
- Map selectFileInfoByImageFileId(@ParamMapper("imageFileId") int imageFileId);
-
- /**
- * 删除文件信息
- *
- * @param imageFileId 文件ID
- * @return 是否删除成功
- */
- @Delete("delete from imagefile where imagefileid = #{imageFileId}")
- boolean deleteImageFileInfo(@ParamMapper("imageFileId") Integer imageFileId);
-
- /**
- * 插入自定义配置数据
- *
- * @param onlyMark 唯一标识
- * @param value 参数值
- * @param desc 描述
- * @return 是否插入成功
- */
- @Update("update $t{configTableName} set only_mark = #{onlyMark},param_value = #{paramValue}, \n" +
- "param_desc = #{paramDesc} where id = #{id}")
- boolean updateConfigValueById(@ParamMapper("onlyMark") String onlyMark,
- @ParamMapper("paramValue") String value,
- @ParamMapper("paramDesc") String desc,
- @ParamMapper("id") String id,
- @ParamMapper("configTableName") String configTableName);
-
- /**
- * 修改自定义配置数据
- *
- * @param onlyMark 唯一标识
- * @param value 值
- * @param desc 描述
- * @return 是否更新成功
- */
- @Update("update $t{configTableName} set param_value = #{paramValue}, \n" +
- "param_desc = #{paramDesc} where only_mark = #{onlyMark}")
- boolean updateConfigValueByOnlyMark(@ParamMapper("onlyMark") String onlyMark,
- @ParamMapper("paramValue") String value,
- @ParamMapper("paramDesc") String desc,
- @ParamMapper("configTableName") String configTableName);
-
- /**
- * selectBillTableByFromId 根据fromId查询billTable表明
- * 2023/2/6 13:40
- * ************************************************************
- *
- * @param fromId fromId
- * @return String 表名
- * @author youHong.ai
- * ******************************************
- */
- @Select("select * from workflow_bill where id = #{fromId}")
- String selectBillTableByFromId(@ParamMapper("fromId") String fromId);
-
- @Delete("delete from $t{tableName} where id = #{dataId}")
- void deleteModeId(@ParamMapper("tableName") String tableName, @ParamMapper("dataId") Integer dataId);
+
+ /**
+ * 查询日志级别是否开启Debug模式
+ *
+ * @return 是否开启Debug模式
+ */
+ @Select("select param_value from $t{configTableName} where only_mark = 'enableDebugLog'")
+ Boolean selectLogLevel(@ParamMapper("configTableName") String configTableName);
+
+
+ /**
+ * 根据唯一标识查询参数值
+ *
+ * @param onlyMark 唯一标识
+ * @return 参数值
+ */
+ @Select("select param_value from $t{configTableName} where only_mark = #{onlyMark} and enable_param = 1")
+ String selectCusConfigParam(@ParamMapper("onlyMark") String onlyMark,
+ @ParamMapper("configTableName") String cusConfigTableName);
+
+
+ /**
+ * 查询文件名
+ *
+ * @param imageFileId 查询文件名
+ * @return 文件名
+ */
+ @Select("select imagefilename from imagefile where imagefileid = #{imageFileId}")
+ String selectFileNameByImageFileId(@ParamMapper("imageFileId") int imageFileId);
+
+ /**
+ * 查询流程主表
+ *
+ * @param workflowId 流程id
+ * @return 流程表名
+ */
+ @Select("select bill.tablename from workflow_bill bill join workflow_base base on base.formid = bill.id where base.id = #{workflowId}")
+ String selectWorkfowMainTable(@ParamMapper("workflowId") String workflowId);
+
+
+ @Select("select id,docid doc_id,imagefileid image_file_id,imagefilename image_file_name from docimagefile where docid = #{docId}")
+ DocImageInfo selectDocImageInfo(@ParamMapper("docId") String docId);
+
+ @Select("select id,docid doc_id,imagefileid image_file_id,imagefilename image_file_name from docimagefile where docid in ($t{docIds})")
+ List selectDocImageInfos(@ParamMapper("docIds") String docIds);
+
+ @Select("select id,docid doc_id,imagefileid image_file_id,imagefilename image_file_name from docimagefile where docid in (${docIds})")
+ List selectDocImageInfos(@ParamMapper("docIds") String[] docIds);
+
+
+ @Select("select id,workflow_type,mark_only,workflow_nodes,enable_nodes from uf_node_config where enable_nodes = 1 and workflow_type in ($t{allVersion}) and mark_only = #{markOnly}")
+ WorkflowNodeConfig selectNodeConfig(@ParamMapper("allVersion") String allVersion, @ParamMapper("markOnly") String markOnly);
+
+ /**
+ * 查询下拉框值
+ *
+ * @param tableName 表明
+ * @param fileName 字段名
+ * @return 下拉框
+ */
+ @Select("select wbf.id,wbf.fieldname,wbf.fieldlabel,wb.tablename, ws.selectname,ws.selectvalue " +
+ "from workflow_billfield wbf left join workflow_bill wb on wbf.billid = wb.id " +
+ "left join workflow_selectitem ws on ws.fieldid = wbf.id where wb.tablename = #{tableName} and fieldname = #{fileName}")
+ List selectSelectFieldValue(@ParamMapper("tableName") String tableName, @ParamMapper("fileName") String fileName);
+
+ @Select("select ws.selectname " +
+ "from workflow_billfield wbf left join workflow_bill wb on wbf.billid = wb.id " +
+ "left join workflow_selectitem ws on ws.fieldid = wbf.id where wb.tablename = #{tableName} and fieldname = #{fileName} and selectvalue = #{value}")
+ String selectSelectFieldValueByValue(@ParamMapper("tableName") String tableName, @ParamMapper("fileName") String fileName,
+ @ParamMapper("value") String value);
+
+ /**
+ * 查询文件信息
+ *
+ * @param imageFileId 查询文件名
+ * @return 文件名
+ */
+ @Select("select * from imagefile where imagefileid = #{imageFileId}")
+ Map selectFileInfoByImageFileId(@ParamMapper("imageFileId") int imageFileId);
+
+ /**
+ * 删除文件信息
+ *
+ * @param imageFileId 文件ID
+ * @return 是否删除成功
+ */
+ @Delete("delete from imagefile where imagefileid = #{imageFileId}")
+ boolean deleteImageFileInfo(@ParamMapper("imageFileId") Integer imageFileId);
+
+ /**
+ * 插入自定义配置数据
+ *
+ * @param onlyMark 唯一标识
+ * @param value 参数值
+ * @param desc 描述
+ * @return 是否插入成功
+ */
+ @Update("update $t{configTableName} set only_mark = #{onlyMark},param_value = #{paramValue}, \n" +
+ "param_desc = #{paramDesc} where id = #{id}")
+ boolean updateConfigValueById(@ParamMapper("onlyMark") String onlyMark,
+ @ParamMapper("paramValue") String value,
+ @ParamMapper("paramDesc") String desc,
+ @ParamMapper("id") String id,
+ @ParamMapper("configTableName") String configTableName);
+
+ /**
+ * 修改自定义配置数据
+ *
+ * @param onlyMark 唯一标识
+ * @param value 值
+ * @param desc 描述
+ * @return 是否更新成功
+ */
+ @Update("update $t{configTableName} set param_value = #{paramValue}, \n" +
+ "param_desc = #{paramDesc} where only_mark = #{onlyMark}")
+ boolean updateConfigValueByOnlyMark(@ParamMapper("onlyMark") String onlyMark,
+ @ParamMapper("paramValue") String value,
+ @ParamMapper("paramDesc") String desc,
+ @ParamMapper("configTableName") String configTableName);
+
+ /**
+ * selectBillTableByFromId 根据fromId查询billTable表明
+ * 2023/2/6 13:40
+ * ************************************************************
+ *
+ * @param fromId fromId
+ * @return String 表名
+ * @author youHong.ai
+ * ******************************************
+ */
+ @Select("select * from workflow_bill where id = #{fromId}")
+ String selectBillTableByFromId(@ParamMapper("fromId") String fromId);
+
+ /**
+ * 删除指定表数据
+ *
+ * @param tableName 表名
+ * @param dataId 数据id
+ */
+ @Delete("delete from $t{tableName} where id = #{dataId}")
+ void deleteModeId(@ParamMapper("tableName") String tableName, @ParamMapper("dataId") Integer dataId);
+
+
+ /**
+ * 根据字段id查询字段信息
+ *
+ * @param id 字段id
+ * @return 字段信息
+ */
+ @Select("select id,fieldname field_name,tablename table_name,\n" +
+ " billid bill_id,fieldtype field_type,\n" +
+ " fieldhtmltype field_html_type\n" +
+ "from workflow_field_table_view where id = #{id}")
+ FieldViewInfo selectFieldInfo(Integer id);
}
diff --git a/src/main/java/aiyh/utils/recordset/RecordsetUtil.java b/src/main/java/aiyh/utils/recordset/RecordsetUtil.java
index 29145e3..55220f9 100644
--- a/src/main/java/aiyh/utils/recordset/RecordsetUtil.java
+++ b/src/main/java/aiyh/utils/recordset/RecordsetUtil.java
@@ -21,463 +21,478 @@ import java.util.List;
public class RecordsetUtil implements InvocationHandler {
-
-
- public static final String SQL_LOG = "sql_log";
- private final RsThreadLocalManager rsManager = new RsThreadLocalManager();
-
- private boolean autoCommit = true;
-
- public T getMapper(Class tClass) {
- return getMapper(tClass, true);
- }
-
- public T getMapper(Class tClass, boolean autoCommit) {
- if (tClass == null) {
- throw new BindingException("class is null!");
- }
- if (tClass.getAnnotation(SqlMapper.class) == null) {
- throw new BindingException("can not find SqlMapper annotation!");
- }
- this.autoCommit = autoCommit;
- return (T) Proxy.newProxyInstance(tClass.getClassLoader(), new Class[]{tClass}, this);
- }
-
-
- @Override
- public Object invoke(Object proxy, Method method, Object[] args) {
- if (autoCommit) {
- return invokeRs(proxy, method, args);
- }
- return invokeRsTrans(proxy, method, args);
- }
-
-
- private Object invokeRs(Object proxy, Method method, Object[] args) {
- RecordSet rs = rsManager.getRs(method.getDeclaringClass().getName());
- if (rs == null) {
- rsManager.setRecordSet(method.getDeclaringClass().getName());
- rs = rsManager.getRs(method.getDeclaringClass().getName());
- }
- SqlHandler sqlHandler = new SqlHandler();
- ResultMapper resultMapper = new ResultMapper();
- Select select = method.getAnnotation(Select.class);
- if (select != null) {
- // 查询
- String sql = select.value();
- boolean custom = select.custom();
- PrepSqlResultImpl handler = sqlHandler.handler(sql, custom, method, args);
- if (!handler.getSqlStr().trim().toLowerCase().startsWith("select ")) {
- throw new CustomerException("The sql statement does not match, the @Select annotation can only getDataId the select statement, please check whether the sql statement matches!");
- }
- Util.getLogger(SQL_LOG).info("解析sql===>" + handler);
- if (handler.getArgs().isEmpty()) {
- rs.executeQuery(handler.getSqlStr());
- } else {
- rs.executeQuery(handler.getSqlStr(), handler.getArgs());
- }
- return resultMapper.mapperResult(rs, method, method.getReturnType());
- }
- Update update = method.getAnnotation(Update.class);
- if (update != null) {
- // 查询
- String sql = update.value();
- boolean custom = update.custom();
- PrepSqlResultImpl handler = sqlHandler.handler(sql, custom, method, args);
- if (!handler.getSqlStr().trim().toLowerCase().startsWith("update ")) {
- throw new CustomerException("The sql statement does not match, the @Update annotation can only getDataId the update statement, please check whether the sql statement matches!");
- }
- Util.getLogger(SQL_LOG).info(handler.toString());
- Class returnType = method.getReturnType();
- boolean b;
- if (handler.getArgs().isEmpty()) {
- b = rs.executeUpdate(handler.getSqlStr());
- } else {
- b = rs.executeUpdate(handler.getSqlStr(), handler.getArgs());
- }
- if (returnType.equals(void.class)) {
- return null;
- }
- if (returnType.equals(int.class) || returnType.equals(Integer.class)) {
- if (b) {
- return 1;
- } else {
- return 0;
- }
- }
- if (returnType.equals(boolean.class) || returnType.equals(Boolean.class)) {
- return b;
- }
- }
- Insert insert = method.getAnnotation(Insert.class);
- if (insert != null) {
- // 查询
- String sql = insert.value();
- boolean custom = insert.custom();
- PrepSqlResultImpl handler = sqlHandler.handler(sql, custom, method, args);
- if (!handler.getSqlStr().trim().toLowerCase().startsWith("insert ")) {
- throw new CustomerException("The sql statement does not match, the @Insert annotation can only getDataId the insert statement, please check whether the sql statement matches!");
- }
- Util.getLogger(SQL_LOG).info(handler.toString());
- Class returnType = method.getReturnType();
- boolean b;
- if (handler.getArgs().isEmpty()) {
- b = rs.executeUpdate(handler.getSqlStr());
- } else {
- b = rs.executeUpdate(handler.getSqlStr(), handler.getArgs());
- }
- if (returnType.equals(void.class)) {
- return null;
- }
- if (returnType.equals(boolean.class) || returnType.equals(Boolean.class)) {
- return b;
- }
- }
- Delete delete = method.getAnnotation(Delete.class);
- if (delete != null) {
- // 查询
- String sql = delete.value();
- boolean custom = delete.custom();
- PrepSqlResultImpl handler = sqlHandler.handler(sql, custom, method, args);
- if (!handler.getSqlStr().trim().toLowerCase().startsWith("delete ")) {
- throw new CustomerException("The sql statement does not match, the @Delete annotation can only getDataId the delete statement, please check whether the sql statement matches!");
- }
- Util.getLogger(SQL_LOG).info(handler.toString());
- Class returnType = method.getReturnType();
- boolean b;
- if (handler.getArgs().isEmpty()) {
- b = rs.executeUpdate(handler.getSqlStr());
- } else {
- b = rs.executeUpdate(handler.getSqlStr(), handler.getArgs());
- }
- if (returnType.equals(void.class)) {
- return null;
- }
- if (returnType.equals(boolean.class) || returnType.equals(Boolean.class)) {
- return b;
- }
- }
- boolean hasBatchInsert = method.isAnnotationPresent(BatchInsert.class);
- if (hasBatchInsert) {
- BatchInsert batchInsert = method.getAnnotation(BatchInsert.class);
- String sql = batchInsert.value();
- Class returnType = method.getReturnType();
- boolean custom = batchInsert.custom();
- BatchSqlResultImpl batchSqlResult = sqlHandler.handlerBatch(sql, custom, method, args);
- Util.getLogger(SQL_LOG).info(batchSqlResult.toString());
- if (batchSqlResult.getBatchList().isEmpty()) {
- throw new CustomerException("getDataId batch sql error , batch sql args is empty!");
- }
- if (!batchSqlResult.getSqlStr().trim().toLowerCase().startsWith("insert ")) {
- throw new CustomerException("The sql statement does not match, the @Insert annotation can only getDataId the insert statement, please check whether the sql statement matches!");
- }
- boolean b = rs.executeBatchSql(batchSqlResult.getSqlStr(), batchSqlResult.getBatchList());
- if (returnType.equals(void.class)) {
- return null;
- }
- if (returnType.equals(boolean.class) || returnType.equals(Boolean.class)) {
- return b;
- }
-
- }
- boolean hasBatchUpdate = method.isAnnotationPresent(BatchUpdate.class);
- if (hasBatchUpdate) {
- BatchUpdate batchUpdate = method.getAnnotation(BatchUpdate.class);
- String sql = batchUpdate.value();
- Class returnType = method.getReturnType();
- boolean custom = batchUpdate.custom();
- BatchSqlResultImpl batchSqlResult = sqlHandler.handlerBatch(sql, custom, method, args);
- Util.getLogger(SQL_LOG).info(batchSqlResult.toString());
- if (batchSqlResult.getBatchList().isEmpty()) {
- throw new CustomerException("getDataId batch sql error , batch sql args is empty!");
- }
- if (!batchSqlResult.getSqlStr().trim().toLowerCase().startsWith("update ")) {
- throw new CustomerException("The sql statement does not match, the @Update annotation can only getDataId the update statement, please check whether the sql statement matches!");
- }
- boolean b = rs.executeBatchSql(batchSqlResult.getSqlStr(), batchSqlResult.getBatchList());
- if (returnType.equals(void.class)) {
- return null;
- }
- if (returnType.equals(boolean.class) || returnType.equals(Boolean.class)) {
- return b;
- }
-
- }
- boolean hasBatchDelete = method.isAnnotationPresent(BatchDelete.class);
- if (hasBatchDelete) {
- BatchDelete batchDelete = method.getAnnotation(BatchDelete.class);
- String sql = batchDelete.value();
- Class returnType = method.getReturnType();
- boolean custom = batchDelete.custom();
- BatchSqlResultImpl batchSqlResult = sqlHandler.handlerBatch(sql, custom, method, args);
- Util.getLogger(SQL_LOG).info(batchSqlResult.toString());
- if (batchSqlResult.getBatchList().isEmpty()) {
- throw new CustomerException("getDataId batch sql error , batch sql args is empty!");
- }
- if (!batchSqlResult.getSqlStr().trim().toLowerCase().startsWith("delete ")) {
- throw new CustomerException("The sql statement does not match, the @Delete annotation can only getDataId the delete statement, please check whether the sql statement matches!");
- }
- boolean b = rs.executeBatchSql(batchSqlResult.getSqlStr(), batchSqlResult.getBatchList());
- if (returnType.equals(void.class)) {
- return null;
- }
- if (returnType.equals(boolean.class) || returnType.equals(Boolean.class)) {
- return b;
- }
-
- }
- throw new CustomerException("该方法没有添加注解！请检查是否正确添加注解！@Select、@Update、@Insert、@Delete、@BatchUpdate、@BatchInsert、@BatchDelete");
- }
-
- private Object invokeRsTrans(Object proxy, Method method, Object[] args) {
- RecordSetTrans rs = rsManager.getTrans(method.getDeclaringClass().getName());
- if (rs == null) {
- rsManager.setRecordSetTrans(method.getDeclaringClass().getName());
- rs = rsManager.getTrans(method.getDeclaringClass().getName());
- }
- SqlHandler sqlHandler = new SqlHandler();
- ResultMapper resultMapper = new ResultMapper();
- Select select = method.getAnnotation(Select.class);
- if (select != null) {
- // 查询
- String sql = select.value();
- boolean custom = select.custom();
- PrepSqlResultImpl handler = sqlHandler.handler(sql, custom, method, args);
- if (!handler.getSqlStr().trim().toLowerCase().startsWith("select ")) {
- throw new CustomerException("The sql statement does not match, the @Select annotation can only getDataId the select statement, please check whether the sql statement matches!");
- }
- Util.getLogger(SQL_LOG).info("解析sql===>" + handler);
- try {
- if (handler.getArgs().isEmpty()) {
- rs.executeQuery(handler.getSqlStr());
- } else {
- rs.executeQuery(handler.getSqlStr(), handler.getArgs());
- }
- } catch (Exception e) {
- Util.getLogger(SQL_LOG).error("execute sql error! " + Util.getErrString(e));
- throw new CustomerException("execute sql error!" + e.getMessage());
- }
- return resultMapper.mapperResult(rs, method, method.getReturnType());
- }
-
- Update update = method.getAnnotation(Update.class);
- if (update != null) {
- // 查询
- String sql = update.value();
- boolean custom = update.custom();
- PrepSqlResultImpl handler = sqlHandler.handler(sql, custom, method, args);
- if (!handler.getSqlStr().trim().toLowerCase().startsWith("update ")) {
- throw new CustomerException("The sql statement does not match, the @Update annotation can only getDataId the update statement, please check whether the sql statement matches!");
- }
- Util.getLogger(SQL_LOG).info(handler.toString());
- Class returnType = method.getReturnType();
- boolean b;
- try {
- if (handler.getArgs().isEmpty()) {
- b = rs.executeUpdate(handler.getSqlStr());
- } else {
- b = rs.executeUpdate(handler.getSqlStr(), handler.getArgs());
- }
- } catch (Exception e) {
- Util.getLogger(SQL_LOG).error("execute sql error! " + Util.getErrString(e));
- throw new CustomerException("execute sql error!" + e.getMessage());
- }
- if (returnType.equals(void.class)) {
- return null;
- }
- if (returnType.equals(int.class) || returnType.equals(Integer.class)) {
- if (b) {
- return 1;
- } else {
- return 0;
- }
- }
- if (returnType.equals(boolean.class) || returnType.equals(Boolean.class)) {
- return b;
- }
- }
- Insert insert = method.getAnnotation(Insert.class);
- if (insert != null) {
- // 查询
- String sql = insert.value();
- boolean custom = insert.custom();
- PrepSqlResultImpl handler = sqlHandler.handler(sql, custom, method, args);
- if (!handler.getSqlStr().trim().toLowerCase().startsWith("insert ")) {
- throw new CustomerException("The sql statement does not match, the @Insert annotation can only getDataId the insert statement, please check whether the sql statement matches!");
- }
- Util.getLogger(SQL_LOG).info(handler.toString());
- Class returnType = method.getReturnType();
- boolean b;
- try {
- if (handler.getArgs().isEmpty()) {
- b = rs.executeUpdate(handler.getSqlStr());
- } else {
- b = rs.executeUpdate(handler.getSqlStr(), handler.getArgs());
- }
- } catch (Exception e) {
- Util.getLogger(SQL_LOG).error("execute sql error! " + Util.getErrString(e));
- throw new CustomerException("execute sql error!" + e.getMessage());
- }
- if (returnType.equals(void.class)) {
- return null;
- }
- if (returnType.equals(boolean.class) || returnType.equals(Boolean.class)) {
- return b;
- }
- }
- Delete delete = method.getAnnotation(Delete.class);
- if (delete != null) {
- // 查询
- String sql = delete.value();
- boolean custom = delete.custom();
- PrepSqlResultImpl handler = sqlHandler.handler(sql, custom, method, args);
- if (!handler.getSqlStr().trim().toLowerCase().startsWith("delete ")) {
- throw new CustomerException("The sql statement does not match, the @Delete annotation can only getDataId the delete statement, please check whether the sql statement matches!");
- }
- Util.getLogger(SQL_LOG).info(handler.toString());
- Class returnType = method.getReturnType();
- boolean b;
- try {
- if (handler.getArgs().isEmpty()) {
- b = rs.executeUpdate(handler.getSqlStr());
- } else {
- b = rs.executeUpdate(handler.getSqlStr(), handler.getArgs());
- }
- } catch (Exception e) {
- Util.getLogger(SQL_LOG).error("execute sql error! " + Util.getErrString(e));
- throw new CustomerException("execute sql error!" + e.getMessage());
- }
- if (returnType.equals(void.class)) {
- return null;
- }
- if (returnType.equals(boolean.class) || returnType.equals(Boolean.class)) {
- return b;
- }
- }
- boolean hasBatchInsert = method.isAnnotationPresent(BatchInsert.class);
- if (hasBatchInsert) {
- BatchInsert batchInsert = method.getAnnotation(BatchInsert.class);
- String sql = batchInsert.value();
- Class returnType = method.getReturnType();
- boolean custom = batchInsert.custom();
- BatchSqlResultImpl batchSqlResult = sqlHandler.handlerBatch(sql, custom, method, args);
- Util.getLogger(SQL_LOG).info(batchSqlResult.toString());
- List batchList = batchSqlResult.getBatchList();
- if (batchList.isEmpty()) {
- throw new CustomerException("getDataId batch sql error , batch sql args is empty!");
- }
- List> batchListTrans = new ArrayList<>();
- for (List list : batchList) {
- List listItem = new ArrayList<>();
- for (Object o : list) {
- listItem.add(o);
- }
- batchListTrans.add(listItem);
- }
-
- if (!batchSqlResult.getSqlStr().trim().toLowerCase().startsWith("insert ")) {
- throw new CustomerException("The sql statement does not match, the @Insert annotation can only getDataId the insert statement, please check whether the sql statement matches!");
- }
- boolean b = true;
- try {
- rs.executeBatchSql(batchSqlResult.getSqlStr(), batchListTrans);
- } catch (Exception e) {
- Util.getLogger(SQL_LOG).error("execute sql error! " + Util.getErrString(e));
- throw new CustomerException("execute sql error!" + e.getMessage());
- }
- if (returnType.equals(void.class)) {
- return null;
- }
- if (returnType.equals(boolean.class) || returnType.equals(Boolean.class)) {
- return b;
- }
-
- }
- boolean hasBatchUpdate = method.isAnnotationPresent(BatchUpdate.class);
- if (hasBatchUpdate) {
- BatchUpdate batchUpdate = method.getAnnotation(BatchUpdate.class);
- String sql = batchUpdate.value();
- Class returnType = method.getReturnType();
- boolean custom = batchUpdate.custom();
- BatchSqlResultImpl batchSqlResult = sqlHandler.handlerBatch(sql, custom, method, args);
- Util.getLogger(SQL_LOG).info(batchSqlResult.toString());
- if (batchSqlResult.getBatchList().isEmpty()) {
- throw new CustomerException("getDataId batch sql error , batch sql args is empty!");
- }
- if (!batchSqlResult.getSqlStr().trim().toLowerCase().startsWith("update ")) {
- throw new CustomerException("The sql statement does not match, the @Update annotation can only getDataId the update statement, please check whether the sql statement matches!");
- }
-
- List batchList = batchSqlResult.getBatchList();
- if (batchList.isEmpty()) {
- throw new CustomerException("getDataId batch sql error , batch sql args is empty!");
- }
- List> batchListTrans = new ArrayList<>();
- for (List list : batchList) {
- List listItem = new ArrayList<>();
- for (Object o : list) {
- listItem.add(o);
- }
- batchListTrans.add(listItem);
- }
- try {
- rs.executeBatchSql(batchSqlResult.getSqlStr(), batchListTrans);
- } catch (Exception e) {
- Util.getLogger(SQL_LOG).error("execute sql error! " + Util.getErrString(e));
- throw new CustomerException("execute sql error!" + e.getMessage());
- }
- if (returnType.equals(void.class)) {
- return null;
- }
- if (returnType.equals(boolean.class) || returnType.equals(Boolean.class)) {
- return true;
- }
-
- }
- boolean hasBatchDelete = method.isAnnotationPresent(BatchDelete.class);
- if (hasBatchDelete) {
- BatchDelete batchDelete = method.getAnnotation(BatchDelete.class);
- String sql = batchDelete.value();
- Class returnType = method.getReturnType();
- boolean custom = batchDelete.custom();
- BatchSqlResultImpl batchSqlResult = sqlHandler.handlerBatch(sql, custom, method, args);
- Util.getLogger(SQL_LOG).info(batchSqlResult.toString());
- if (batchSqlResult.getBatchList().isEmpty()) {
- throw new CustomerException("getDataId batch sql error , batch sql args is empty!");
- }
- if (!batchSqlResult.getSqlStr().trim().toLowerCase().startsWith("delete ")) {
- throw new CustomerException("The sql statement does not match, the @Delete annotation can only getDataId the delete statement, please check whether the sql statement matches!");
- }
- List batchList = batchSqlResult.getBatchList();
- if (batchList.isEmpty()) {
- throw new CustomerException("getDataId batch sql error , batch sql args is empty!");
- }
- List> batchListTrans = new ArrayList<>();
- for (List list : batchList) {
- List listItem = new ArrayList<>();
- for (Object o : list) {
- listItem.add(o);
- }
- batchListTrans.add(listItem);
- }
- try {
- rs.executeBatchSql(batchSqlResult.getSqlStr(), batchListTrans);
- } catch (Exception e) {
- Util.getLogger(SQL_LOG).error("execute sql error! " + Util.getErrString(e));
- throw new CustomerException("execute sql error!" + e.getMessage());
- }
- if (returnType.equals(void.class)) {
- return null;
- }
- if (returnType.equals(boolean.class) || returnType.equals(Boolean.class)) {
- return true;
- }
-
- }
- throw new CustomerException("该方法没有添加注解！请检查是否正确添加注解！@Select、@Update、@Insert、@Delete、@BatchUpdate、@BatchInsert、@BatchDelete");
- }
-
- public RsThreadLocalManager getRsManager() {
- return rsManager;
- }
+
+
+ public static final String SQL_LOG = "sql_log";
+ private final RsThreadLocalManager rsManager = new RsThreadLocalManager();
+
+ private boolean autoCommit = true;
+
+ public T getMapper(Class tClass) {
+ return getMapper(tClass, true);
+ }
+
+ public T getMapper(Class tClass, boolean autoCommit) {
+ if (tClass == null) {
+ throw new BindingException("class is null!");
+ }
+ if (tClass.getAnnotation(SqlMapper.class) == null) {
+ throw new BindingException("can not find SqlMapper annotation!");
+ }
+ this.autoCommit = autoCommit;
+ return (T) Proxy.newProxyInstance(tClass.getClassLoader(), new Class[]{tClass}, this);
+ }
+
+
+ @Override
+ public Object invoke(Object proxy, Method method, Object[] args) {
+ if (autoCommit) {
+ return invokeRs(proxy, method, args, "");
+ }
+ return invokeRsTrans(proxy, method, args, "");
+ }
+
+ public Object invoke(Object proxy, Method method, Object[] args, String name) {
+ if (autoCommit) {
+ return invokeRs(proxy, method, args, name);
+ }
+ return invokeRsTrans(proxy, method, args, name);
+ }
+
+
+ private Object invokeRs(Object proxy, Method method, Object[] args, String name) {
+ String mapperKey = method.getDeclaringClass().getName();
+ if (!"".equals(name) && null != name) {
+ mapperKey += "." + name;
+ }
+ RecordSet rs = rsManager.getRs(mapperKey);
+ if (rs == null) {
+ rsManager.setRecordSet(mapperKey);
+ rs = rsManager.getRs(mapperKey);
+ }
+ SqlHandler sqlHandler = new SqlHandler();
+ ResultMapper resultMapper = new ResultMapper();
+ Select select = method.getAnnotation(Select.class);
+ if (select != null) {
+ // 查询
+ String sql = select.value();
+ boolean custom = select.custom();
+ PrepSqlResultImpl handler = sqlHandler.handler(sql, custom, method, args);
+ if (!handler.getSqlStr().trim().toLowerCase().startsWith("select ")) {
+ throw new CustomerException("The sql statement does not match, the @Select annotation can only execute the select statement, please check whether the sql statement matches!");
+ }
+ Util.getLogger(SQL_LOG).info("解析sql===>" + handler);
+ if (handler.getArgs().isEmpty()) {
+ rs.executeQuery(handler.getSqlStr());
+ } else {
+ rs.executeQuery(handler.getSqlStr(), handler.getArgs());
+ }
+ return resultMapper.mapperResult(rs, method, method.getReturnType(), this);
+ }
+ Update update = method.getAnnotation(Update.class);
+ if (update != null) {
+ // 查询
+ String sql = update.value();
+ boolean custom = update.custom();
+ PrepSqlResultImpl handler = sqlHandler.handler(sql, custom, method, args);
+ if (!handler.getSqlStr().trim().toLowerCase().startsWith("update ")) {
+ throw new CustomerException("The sql statement does not match, the @Update annotation can only execute the update statement, please check whether the sql statement matches!");
+ }
+ Util.getLogger(SQL_LOG).info(handler.toString());
+ Class returnType = method.getReturnType();
+ boolean b;
+ if (handler.getArgs().isEmpty()) {
+ b = rs.executeUpdate(handler.getSqlStr());
+ } else {
+ b = rs.executeUpdate(handler.getSqlStr(), handler.getArgs());
+ }
+ if (returnType.equals(void.class)) {
+ return null;
+ }
+ if (returnType.equals(int.class) || returnType.equals(Integer.class)) {
+ if (b) {
+ return 1;
+ } else {
+ return 0;
+ }
+ }
+ if (returnType.equals(boolean.class) || returnType.equals(Boolean.class)) {
+ return b;
+ }
+ }
+ Insert insert = method.getAnnotation(Insert.class);
+ if (insert != null) {
+ // 查询
+ String sql = insert.value();
+ boolean custom = insert.custom();
+ PrepSqlResultImpl handler = sqlHandler.handler(sql, custom, method, args);
+ if (!handler.getSqlStr().trim().toLowerCase().startsWith("insert ")) {
+ throw new CustomerException("The sql statement does not match, the @Insert annotation can only execute the insert statement, please check whether the sql statement matches!");
+ }
+ Util.getLogger(SQL_LOG).info(handler.toString());
+ Class returnType = method.getReturnType();
+ boolean b;
+ if (handler.getArgs().isEmpty()) {
+ b = rs.executeUpdate(handler.getSqlStr());
+ } else {
+ b = rs.executeUpdate(handler.getSqlStr(), handler.getArgs());
+ }
+ if (returnType.equals(void.class)) {
+ return null;
+ }
+ if (returnType.equals(boolean.class) || returnType.equals(Boolean.class)) {
+ return b;
+ }
+ }
+ Delete delete = method.getAnnotation(Delete.class);
+ if (delete != null) {
+ // 查询
+ String sql = delete.value();
+ boolean custom = delete.custom();
+ PrepSqlResultImpl handler = sqlHandler.handler(sql, custom, method, args);
+ if (!handler.getSqlStr().trim().toLowerCase().startsWith("delete ")) {
+ throw new CustomerException("The sql statement does not match, the @Delete annotation can only execute the delete statement, please check whether the sql statement matches!");
+ }
+ Util.getLogger(SQL_LOG).info(handler.toString());
+ Class returnType = method.getReturnType();
+ boolean b;
+ if (handler.getArgs().isEmpty()) {
+ b = rs.executeUpdate(handler.getSqlStr());
+ } else {
+ b = rs.executeUpdate(handler.getSqlStr(), handler.getArgs());
+ }
+ if (returnType.equals(void.class)) {
+ return null;
+ }
+ if (returnType.equals(boolean.class) || returnType.equals(Boolean.class)) {
+ return b;
+ }
+ }
+ boolean hasBatchInsert = method.isAnnotationPresent(BatchInsert.class);
+ if (hasBatchInsert) {
+ BatchInsert batchInsert = method.getAnnotation(BatchInsert.class);
+ String sql = batchInsert.value();
+ Class returnType = method.getReturnType();
+ boolean custom = batchInsert.custom();
+ BatchSqlResultImpl batchSqlResult = sqlHandler.handlerBatch(sql, custom, method, args);
+ Util.getLogger(SQL_LOG).info(batchSqlResult.toString());
+ if (batchSqlResult.getBatchList().isEmpty()) {
+ throw new CustomerException("execute batch sql error , batch sql args is empty!");
+ }
+ if (!batchSqlResult.getSqlStr().trim().toLowerCase().startsWith("insert ")) {
+ throw new CustomerException("The sql statement does not match, the @Insert annotation can only execute the insert statement, please check whether the sql statement matches!");
+ }
+ boolean b = rs.executeBatchSql(batchSqlResult.getSqlStr(), batchSqlResult.getBatchList());
+ if (returnType.equals(void.class)) {
+ return null;
+ }
+ if (returnType.equals(boolean.class) || returnType.equals(Boolean.class)) {
+ return b;
+ }
+
+ }
+ boolean hasBatchUpdate = method.isAnnotationPresent(BatchUpdate.class);
+ if (hasBatchUpdate) {
+ BatchUpdate batchUpdate = method.getAnnotation(BatchUpdate.class);
+ String sql = batchUpdate.value();
+ Class returnType = method.getReturnType();
+ boolean custom = batchUpdate.custom();
+ BatchSqlResultImpl batchSqlResult = sqlHandler.handlerBatch(sql, custom, method, args);
+ Util.getLogger(SQL_LOG).info(batchSqlResult.toString());
+ if (batchSqlResult.getBatchList().isEmpty()) {
+ throw new CustomerException("execute batch sql error , batch sql args is empty!");
+ }
+ if (!batchSqlResult.getSqlStr().trim().toLowerCase().startsWith("update ")) {
+ throw new CustomerException("The sql statement does not match, the @Update annotation can only execute the update statement, please check whether the sql statement matches!");
+ }
+ boolean b = rs.executeBatchSql(batchSqlResult.getSqlStr(), batchSqlResult.getBatchList());
+ if (returnType.equals(void.class)) {
+ return null;
+ }
+ if (returnType.equals(boolean.class) || returnType.equals(Boolean.class)) {
+ return b;
+ }
+
+ }
+ boolean hasBatchDelete = method.isAnnotationPresent(BatchDelete.class);
+ if (hasBatchDelete) {
+ BatchDelete batchDelete = method.getAnnotation(BatchDelete.class);
+ String sql = batchDelete.value();
+ Class returnType = method.getReturnType();
+ boolean custom = batchDelete.custom();
+ BatchSqlResultImpl batchSqlResult = sqlHandler.handlerBatch(sql, custom, method, args);
+ Util.getLogger(SQL_LOG).info(batchSqlResult.toString());
+ if (batchSqlResult.getBatchList().isEmpty()) {
+ throw new CustomerException("execute batch sql error , batch sql args is empty!");
+ }
+ if (!batchSqlResult.getSqlStr().trim().toLowerCase().startsWith("delete ")) {
+ throw new CustomerException("The sql statement does not match, the @Delete annotation can only execute the delete statement, please check whether the sql statement matches!");
+ }
+ boolean b = rs.executeBatchSql(batchSqlResult.getSqlStr(), batchSqlResult.getBatchList());
+ if (returnType.equals(void.class)) {
+ return null;
+ }
+ if (returnType.equals(boolean.class) || returnType.equals(Boolean.class)) {
+ return b;
+ }
+
+ }
+ throw new CustomerException("该方法没有添加注解！请检查是否正确添加注解！@Select、@Update、@Insert、@Delete、@BatchUpdate、@BatchInsert、@BatchDelete");
+ }
+
+ private Object invokeRsTrans(Object proxy, Method method, Object[] args, String name) {
+ String mapperKey = method.getDeclaringClass().getName();
+ if (!"".equals(name) && null != name) {
+ mapperKey += "." + name;
+ }
+ RecordSetTrans rs = rsManager.getTrans(mapperKey);
+ if (rs == null) {
+ rsManager.setRecordSetTrans(mapperKey);
+ rs = rsManager.getTrans(mapperKey);
+ }
+ SqlHandler sqlHandler = new SqlHandler();
+ ResultMapper resultMapper = new ResultMapper();
+ Select select = method.getAnnotation(Select.class);
+ if (select != null) {
+ // 查询
+ String sql = select.value();
+ boolean custom = select.custom();
+ PrepSqlResultImpl handler = sqlHandler.handler(sql, custom, method, args);
+ if (!handler.getSqlStr().trim().toLowerCase().startsWith("select ")) {
+ throw new CustomerException("The sql statement does not match, the @Select annotation can only execute the select statement, please check whether the sql statement matches!");
+ }
+ Util.getLogger(SQL_LOG).info("解析sql===>" + handler);
+ try {
+ if (handler.getArgs().isEmpty()) {
+ rs.executeQuery(handler.getSqlStr());
+ } else {
+ rs.executeQuery(handler.getSqlStr(), handler.getArgs());
+ }
+ } catch (Exception e) {
+ Util.getLogger(SQL_LOG).error("execute sql error! " + Util.getErrString(e));
+ throw new CustomerException("execute sql error!" + e.getMessage());
+ }
+ return resultMapper.mapperResult(rs, method, method.getReturnType(), this);
+ }
+
+ Update update = method.getAnnotation(Update.class);
+ if (update != null) {
+ // 查询
+ String sql = update.value();
+ boolean custom = update.custom();
+ PrepSqlResultImpl handler = sqlHandler.handler(sql, custom, method, args);
+ if (!handler.getSqlStr().trim().toLowerCase().startsWith("update ")) {
+ throw new CustomerException("The sql statement does not match, the @Update annotation can only execute the update statement, please check whether the sql statement matches!");
+ }
+ Util.getLogger(SQL_LOG).info(handler.toString());
+ Class returnType = method.getReturnType();
+ boolean b;
+ try {
+ if (handler.getArgs().isEmpty()) {
+ b = rs.executeUpdate(handler.getSqlStr());
+ } else {
+ b = rs.executeUpdate(handler.getSqlStr(), handler.getArgs());
+ }
+ } catch (Exception e) {
+ Util.getLogger(SQL_LOG).error("execute sql error! " + Util.getErrString(e));
+ throw new CustomerException("execute sql error!" + e.getMessage());
+ }
+ if (returnType.equals(void.class)) {
+ return null;
+ }
+ if (returnType.equals(int.class) || returnType.equals(Integer.class)) {
+ if (b) {
+ return 1;
+ } else {
+ return 0;
+ }
+ }
+ if (returnType.equals(boolean.class) || returnType.equals(Boolean.class)) {
+ return b;
+ }
+ }
+ Insert insert = method.getAnnotation(Insert.class);
+ if (insert != null) {
+ // 查询
+ String sql = insert.value();
+ boolean custom = insert.custom();
+ PrepSqlResultImpl handler = sqlHandler.handler(sql, custom, method, args);
+ if (!handler.getSqlStr().trim().toLowerCase().startsWith("insert ")) {
+ throw new CustomerException("The sql statement does not match, the @Insert annotation can only execute the insert statement, please check whether the sql statement matches!");
+ }
+ Util.getLogger(SQL_LOG).info(handler.toString());
+ Class returnType = method.getReturnType();
+ boolean b;
+ try {
+ if (handler.getArgs().isEmpty()) {
+ b = rs.executeUpdate(handler.getSqlStr());
+ } else {
+ b = rs.executeUpdate(handler.getSqlStr(), handler.getArgs());
+ }
+ } catch (Exception e) {
+ Util.getLogger(SQL_LOG).error("execute sql error! " + Util.getErrString(e));
+ throw new CustomerException("execute sql error!" + e.getMessage());
+ }
+ if (returnType.equals(void.class)) {
+ return null;
+ }
+ if (returnType.equals(boolean.class) || returnType.equals(Boolean.class)) {
+ return b;
+ }
+ }
+ Delete delete = method.getAnnotation(Delete.class);
+ if (delete != null) {
+ // 查询
+ String sql = delete.value();
+ boolean custom = delete.custom();
+ PrepSqlResultImpl handler = sqlHandler.handler(sql, custom, method, args);
+ if (!handler.getSqlStr().trim().toLowerCase().startsWith("delete ")) {
+ throw new CustomerException("The sql statement does not match, the @Delete annotation can only execute the delete statement, please check whether the sql statement matches!");
+ }
+ Util.getLogger(SQL_LOG).info(handler.toString());
+ Class returnType = method.getReturnType();
+ boolean b;
+ try {
+ if (handler.getArgs().isEmpty()) {
+ b = rs.executeUpdate(handler.getSqlStr());
+ } else {
+ b = rs.executeUpdate(handler.getSqlStr(), handler.getArgs());
+ }
+ } catch (Exception e) {
+ Util.getLogger(SQL_LOG).error("execute sql error! " + Util.getErrString(e));
+ throw new CustomerException("execute sql error!" + e.getMessage());
+ }
+ if (returnType.equals(void.class)) {
+ return null;
+ }
+ if (returnType.equals(boolean.class) || returnType.equals(Boolean.class)) {
+ return b;
+ }
+ }
+ boolean hasBatchInsert = method.isAnnotationPresent(BatchInsert.class);
+ if (hasBatchInsert) {
+ BatchInsert batchInsert = method.getAnnotation(BatchInsert.class);
+ String sql = batchInsert.value();
+ Class returnType = method.getReturnType();
+ boolean custom = batchInsert.custom();
+ BatchSqlResultImpl batchSqlResult = sqlHandler.handlerBatch(sql, custom, method, args);
+ Util.getLogger(SQL_LOG).info(batchSqlResult.toString());
+ List batchList = batchSqlResult.getBatchList();
+ if (batchList.isEmpty()) {
+ throw new CustomerException("execute batch sql error , batch sql args is empty!");
+ }
+ List> batchListTrans = new ArrayList<>();
+ for (List list : batchList) {
+ List listItem = new ArrayList<>();
+ for (Object o : list) {
+ listItem.add(o);
+ }
+ batchListTrans.add(listItem);
+ }
+
+ if (!batchSqlResult.getSqlStr().trim().toLowerCase().startsWith("insert ")) {
+ throw new CustomerException("The sql statement does not match, the @Insert annotation can only execute the insert statement, please check whether the sql statement matches!");
+ }
+ boolean b = true;
+ try {
+ rs.executeBatchSql(batchSqlResult.getSqlStr(), batchListTrans);
+ } catch (Exception e) {
+ Util.getLogger(SQL_LOG).error("execute sql error! " + Util.getErrString(e));
+ throw new CustomerException("execute sql error!" + e.getMessage());
+ }
+ if (returnType.equals(void.class)) {
+ return null;
+ }
+ if (returnType.equals(boolean.class) || returnType.equals(Boolean.class)) {
+ return b;
+ }
+
+ }
+ boolean hasBatchUpdate = method.isAnnotationPresent(BatchUpdate.class);
+ if (hasBatchUpdate) {
+ BatchUpdate batchUpdate = method.getAnnotation(BatchUpdate.class);
+ String sql = batchUpdate.value();
+ Class returnType = method.getReturnType();
+ boolean custom = batchUpdate.custom();
+ BatchSqlResultImpl batchSqlResult = sqlHandler.handlerBatch(sql, custom, method, args);
+ Util.getLogger(SQL_LOG).info(batchSqlResult.toString());
+ if (batchSqlResult.getBatchList().isEmpty()) {
+ throw new CustomerException("execute batch sql error , batch sql args is empty!");
+ }
+ if (!batchSqlResult.getSqlStr().trim().toLowerCase().startsWith("update ")) {
+ throw new CustomerException("The sql statement does not match, the @Update annotation can only execute the update statement, please check whether the sql statement matches!");
+ }
+
+ List batchList = batchSqlResult.getBatchList();
+ if (batchList.isEmpty()) {
+ throw new CustomerException("execute batch sql error , batch sql args is empty!");
+ }
+ List> batchListTrans = new ArrayList<>();
+ for (List list : batchList) {
+ List listItem = new ArrayList<>();
+ for (Object o : list) {
+ listItem.add(o);
+ }
+ batchListTrans.add(listItem);
+ }
+ try {
+ rs.executeBatchSql(batchSqlResult.getSqlStr(), batchListTrans);
+ } catch (Exception e) {
+ Util.getLogger(SQL_LOG).error("execute sql error! " + Util.getErrString(e));
+ throw new CustomerException("execute sql error!" + e.getMessage());
+ }
+ if (returnType.equals(void.class)) {
+ return null;
+ }
+ if (returnType.equals(boolean.class) || returnType.equals(Boolean.class)) {
+ return true;
+ }
+
+ }
+ boolean hasBatchDelete = method.isAnnotationPresent(BatchDelete.class);
+ if (hasBatchDelete) {
+ BatchDelete batchDelete = method.getAnnotation(BatchDelete.class);
+ String sql = batchDelete.value();
+ Class returnType = method.getReturnType();
+ boolean custom = batchDelete.custom();
+ BatchSqlResultImpl batchSqlResult = sqlHandler.handlerBatch(sql, custom, method, args);
+ Util.getLogger(SQL_LOG).info(batchSqlResult.toString());
+ if (batchSqlResult.getBatchList().isEmpty()) {
+ throw new CustomerException("execute batch sql error , batch sql args is empty!");
+ }
+ if (!batchSqlResult.getSqlStr().trim().toLowerCase().startsWith("delete ")) {
+ throw new CustomerException("The sql statement does not match, the @Delete annotation can only execute the delete statement, please check whether the sql statement matches!");
+ }
+ List batchList = batchSqlResult.getBatchList();
+ if (batchList.isEmpty()) {
+ throw new CustomerException("execute batch sql error , batch sql args is empty!");
+ }
+ List> batchListTrans = new ArrayList<>();
+ for (List list : batchList) {
+ List listItem = new ArrayList<>();
+ for (Object o : list) {
+ listItem.add(o);
+ }
+ batchListTrans.add(listItem);
+ }
+ try {
+ rs.executeBatchSql(batchSqlResult.getSqlStr(), batchListTrans);
+ } catch (Exception e) {
+ Util.getLogger(SQL_LOG).error("execute sql error! " + Util.getErrString(e));
+ throw new CustomerException("execute sql error!" + e.getMessage());
+ }
+ if (returnType.equals(void.class)) {
+ return null;
+ }
+ if (returnType.equals(boolean.class) || returnType.equals(Boolean.class)) {
+ return true;
+ }
+
+ }
+ throw new CustomerException("该方法没有添加注解！请检查是否正确添加注解！@Select、@Update、@Insert、@Delete、@BatchUpdate、@BatchInsert、@BatchDelete");
+ }
+
+ public RsThreadLocalManager getRsManager() {
+ return rsManager;
+ }
}
diff --git a/src/main/java/aiyh/utils/recordset/ResultMapper.java b/src/main/java/aiyh/utils/recordset/ResultMapper.java
index 9621a4e..80467bf 100644
--- a/src/main/java/aiyh/utils/recordset/ResultMapper.java
+++ b/src/main/java/aiyh/utils/recordset/ResultMapper.java
@@ -1,7 +1,7 @@
package aiyh.utils.recordset;
import aiyh.utils.Util;
-import aiyh.utils.annotation.recordset.CaseConversion;
+import aiyh.utils.annotation.recordset.*;
import aiyh.utils.excention.CustomerException;
import aiyh.utils.excention.TypeNonsupportException;
import com.google.common.base.Strings;
@@ -13,486 +13,661 @@ import java.beans.Introspector;
import java.beans.PropertyDescriptor;
import java.lang.reflect.*;
import java.util.*;
+import java.util.function.Function;
/**
* @author EBU7-dev1-ayh create 2021/12/21 0021 11:03
*/
+@SuppressWarnings("all")
public class ResultMapper {
-
- private static Map, TypeHandler> typeHandler = new HashMap<>();
-
- static {
- IntegerTypeHandler integerTypeHandler = new IntegerTypeHandler();
- typeHandler.put(String.class, new StringTypeHandler());
- typeHandler.put(Integer.class, integerTypeHandler);
- typeHandler.put(int.class, integerTypeHandler);
- typeHandler.put(byte.class, integerTypeHandler);
- typeHandler.put(short.class, integerTypeHandler);
- typeHandler.put(long.class, integerTypeHandler);
- typeHandler.put(Long.class, integerTypeHandler);
- typeHandler.put(Boolean.class, new BooleanTypeHandler());
- typeHandler.put(boolean.class, new BooleanTypeHandler());
- typeHandler.put(Date.class, new DataTypeHandler());
- typeHandler.put(Float.class, new FloatTypeHandler());
- typeHandler.put(float.class, new FloatTypeHandler());
- }
-
- public T mapperResult(RecordSet rs, Method method, Class tClass) {
- if (tClass.equals(void.class)) {
- return null;
- }
- if (tClass.equals(RecordSet.class) || tClass.equals(RecordSetTrans.class)) {
- return (T) rs;
- }
- try {
- if (tClass.equals(List.class)) {
- tClass = (Class) ArrayList.class;
- }
- if (tClass.equals(Map.class)) {
- tClass = (Class) HashMap.class;
- }
- if (ResultMapper.typeHandler.containsKey(tClass)) {
- rs.next();
- return (T) ResultMapper.typeHandler.get(tClass).getValue(rs, 1, null);
- }
- T t = tClass.newInstance();
- if (t instanceof Collection) {
- // 集合求出泛型
- //获取返回值的类型
- Type genericReturnType = method.getGenericReturnType();
- Type actualTypeArgument = ((ParameterizedType) genericReturnType).getActualTypeArguments()[0];
- Class rawType = this.getRawType(actualTypeArgument);
- if (rawType.equals(Map.class)) {
- rawType = HashMap.class;
- }
- while (rs.next()) {
- Object o = null;
- try {
- Constructor constructor = rawType.getConstructor();
- o = constructor.newInstance();
- } catch (NoSuchMethodException | InvocationTargetException ignored) {
- if (Number.class.isAssignableFrom(rawType)) {
- Constructor constructor;
- try {
- constructor = rawType.getConstructor(String.class);
- o = constructor.newInstance("-1");
- } catch (NoSuchMethodException | InvocationTargetException e) {
- throw new CustomerException("can not Initialization " + t.getClass() + " [" + rawType + "]", e);
- }
- }
- }
- if (o == null) {
- throw new CustomerException("can not Initialization " + t.getClass() + " [" + rawType + "]");
- }
- Object object = getObject(rs, o, method);
- ((Collection) t).add(object);
- }
- return t;
- }
- if (t instanceof Map) {
- // map
- Type genericReturnType = method.getGenericReturnType();
- Type actualTypeArgument = ((ParameterizedType) genericReturnType).getActualTypeArguments()[0];
- Class rawType = this.getRawType(actualTypeArgument);
- if (rawType.equals(List.class)) {
- rawType = (Class) ArrayList.class;
- }
- if (rawType.equals(Map.class)) {
- rawType = HashMap.class;
- }
- // Object o = rawType.newInstance();
- Object o = null;
- try {
- Constructor constructor = rawType.getConstructor();
- o = constructor.newInstance();
- } catch (NoSuchMethodException | InvocationTargetException ignored) {
- if (Number.class.isAssignableFrom(rawType)) {
- Constructor constructor;
- try {
- constructor = rawType.getConstructor(String.class);
- o = constructor.newInstance("-1");
- } catch (NoSuchMethodException | InvocationTargetException e) {
- throw new CustomerException("can not Initialization " + t.getClass() + " [" + rawType + "]", e);
- }
- }
- }
- if (o == null) {
- throw new CustomerException("can not Initialization " + t.getClass() + " [" + rawType + "]");
- }
- if (o instanceof Map || o instanceof Collection) {
- throw new TypeNonsupportException("An unsupported return type!");
- }
- if (rs.next()) {
- return (T) getObject(rs, t, method);
- }
- return null;
- }
- if (t.getClass().isArray()) {
- throw new TypeNonsupportException("An unsupported return type!");
- }
- if (rs.next()) {
- return (T) getObject(rs, t, method);
- }
- return null;
- } catch (InstantiationException | IllegalAccessException e) {
- e.printStackTrace();
- }
- return null;
- }
-
- public T mapperResult(RecordSetTrans rs, Method method, Class tClass) {
- if (tClass.equals(void.class)) {
- return null;
- }
- if (tClass.equals(RecordSet.class) || tClass.equals(RecordSetTrans.class)) {
- return (T) rs;
- }
- try {
- if (tClass.equals(List.class)) {
- tClass = (Class) ArrayList.class;
- }
- if (tClass.equals(Map.class)) {
- tClass = (Class) HashMap.class;
- }
- if (ResultMapper.typeHandler.containsKey(tClass)) {
- rs.next();
- return (T) ResultMapper.typeHandler.get(tClass).getValue(rs, 1, null);
- }
- T t = tClass.newInstance();
- if (t instanceof Collection) {
- // 集合求出泛型
- //获取返回值的类型
- Type genericReturnType = method.getGenericReturnType();
- Type actualTypeArgument = ((ParameterizedType) genericReturnType).getActualTypeArguments()[0];
- Class rawType = this.getRawType(actualTypeArgument);
- if (rawType.equals(Map.class)) {
- rawType = HashMap.class;
- }
- while (rs.next()) {
- Object o = null;
- try {
- Constructor constructor = rawType.getConstructor();
- o = constructor.newInstance();
- } catch (NoSuchMethodException | InvocationTargetException ignored) {
- if (Number.class.isAssignableFrom(rawType)) {
- Constructor constructor;
- try {
- constructor = rawType.getConstructor(String.class);
- o = constructor.newInstance("-1");
- } catch (NoSuchMethodException | InvocationTargetException e) {
- throw new CustomerException("can not Initialization " + t.getClass() + " [" + rawType + "]", e);
- }
- }
- }
- if (o == null) {
- throw new CustomerException("can not Initialization " + t.getClass() + " [" + rawType + "]");
- }
- Object object = getObjectTrans(rs, o, method);
- ((Collection) t).add(object);
- }
- return t;
- }
- if (t instanceof Map) {
- // map
- Type genericReturnType = method.getGenericReturnType();
- Type actualTypeArgument = ((ParameterizedType) genericReturnType).getActualTypeArguments()[0];
- Class rawType = this.getRawType(actualTypeArgument);
- if (rawType.equals(List.class)) {
- rawType = (Class) ArrayList.class;
- }
- if (rawType.equals(Map.class)) {
- rawType = HashMap.class;
- }
- // Object o = rawType.newInstance();
- Object o = null;
- try {
- Constructor constructor = rawType.getConstructor();
- o = constructor.newInstance();
- } catch (NoSuchMethodException | InvocationTargetException ignored) {
- if (Number.class.isAssignableFrom(rawType)) {
- Constructor constructor;
- try {
- constructor = rawType.getConstructor(String.class);
- o = constructor.newInstance("-1");
- } catch (NoSuchMethodException | InvocationTargetException e) {
- throw new CustomerException("can not Initialization " + t.getClass() + " [" + rawType + "]", e);
- }
- }
- }
- if (o == null) {
- throw new CustomerException("can not Initialization " + t.getClass() + " [" + rawType + "]");
- }
- if (o instanceof Map || o instanceof Collection) {
- throw new TypeNonsupportException("An unsupported return type!");
- }
- if (rs.next()) {
- return (T) getObjectTrans(rs, t, method);
- }
- return null;
- }
- if (t.getClass().isArray()) {
- throw new TypeNonsupportException("An unsupported return type!");
- }
- if (rs.next()) {
- return (T) getObjectTrans(rs, t, method);
- }
- return null;
- } catch (InstantiationException | IllegalAccessException e) {
- e.printStackTrace();
- }
- return null;
- }
-
- public Object getObjectTrans(RecordSetTrans rs, Object o, Method method) {
- CaseConversion annotation = method.getAnnotation(CaseConversion.class);
- boolean enable = annotation == null ? true : annotation.value();
- String[] columnName = rs.getColumnName();
- String[] columnTypeName = rs.getColumnTypeName();
- int[] columnTypes = rs.getColumnType();
- if (columnTypeName == null) {
- columnTypeName = new String[columnTypes.length];
- for (int i = 0; i < columnTypes.length; i++) {
- int type = columnTypes[i];
- switch (type) {
- case -1:
- columnTypeName[i] = "TEXT";
- break;
- case 4:
- columnTypeName[i] = "INT";
- break;
- case 12:
- columnTypeName[i] = "VARCHAR";
- break;
- default:
- columnTypeName[i] = "VARCHAR";
- }
- }
- }
-
- try {
- if (o instanceof Map) {
- for (int i = 0; i < columnName.length; i++) {
- String columnType = columnTypeName[i];
- if ("int".equalsIgnoreCase(columnType) || "long".equalsIgnoreCase(columnType) || "number".equalsIgnoreCase(columnType) || "MEDIUMINT".equalsIgnoreCase(columnType) || "TINYINT".equalsIgnoreCase(columnType) || "SMALLINT".equalsIgnoreCase(columnType) || "BIGINT".equalsIgnoreCase(columnType) || "INTEGER".equalsIgnoreCase(columnType)) {
- if (enable) {
- ((Map) o).put(Util.toCamelCase(columnName[i]), rs.getInt(i + 1));
- continue;
- }
- ((Map) o).put(columnName[i], rs.getInt(i + 1));
- continue;
- }
- if ("FLOAT".equalsIgnoreCase(columnType) || "DOUBLE".equalsIgnoreCase(columnType) || "DECIMAL".equalsIgnoreCase(columnType)) {
- if (enable) {
- ((Map) o).put(Util.toCamelCase(columnName[i]), rs.getFloat(i + 1));
- continue;
- }
- ((Map) o).put(columnName[i], rs.getFloat(i + 1));
- continue;
- }
- if (enable) {
- ((Map) o).put(Util.toCamelCase(columnName[i]), rs.getString(i + 1));
- continue;
- }
- ((Map) o).put(columnName[i], rs.getString(i + 1));
- continue;
- }
- return o;
- }
- if (o instanceof Collection) {
- throw new TypeNonsupportException("An unsupported return type!");
- }
- if (o instanceof Number) {
- for (int i = 0; i < columnName.length; i++) {
- String columnType = columnTypeName[i];
- if ("int".equalsIgnoreCase(columnType) || "long".equalsIgnoreCase(columnType) || "number".equalsIgnoreCase(columnType) || "MEDIUMINT".equalsIgnoreCase(columnType) || "TINYINT".equalsIgnoreCase(columnType) || "SMALLINT".equalsIgnoreCase(columnType) || "BIGINT".equalsIgnoreCase(columnType) || "INTEGER".equalsIgnoreCase(columnType)) {
- return rs.getInt(i + 1);
- }
- if ("FLOAT".equalsIgnoreCase(columnType) || "DOUBLE".equalsIgnoreCase(columnType) || "DECIMAL".equalsIgnoreCase(columnType)) {
- return rs.getFloat(i + 1);
- }
- }
- return o;
- }
- if (o instanceof String) {
- for (int i = 0; i < columnName.length; i++) {
- return rs.getString(i + 1);
- }
- return o;
- }
- if (o instanceof Boolean) {
- for (int i = 0; i < columnName.length; i++) {
- return rs.getBoolean(i + 1);
- }
- return o;
- }
-
- // Util.getLogger().info("获取对象：" + o.toString());
- BeanInfo beanInfo = Introspector.getBeanInfo(o.getClass(), Object.class);
- PropertyDescriptor[] propertyDescriptors = beanInfo.getPropertyDescriptors();
- for (PropertyDescriptor propertyDescriptor : propertyDescriptors) {
- Class propertyType = propertyDescriptor.getPropertyType();
- Object value = null;
- String fieldName = propertyDescriptor.getName();
-
- if (Strings.isNullOrEmpty(fieldName)) {
- fieldName = propertyDescriptor.getDisplayName();
- }
- // Util.getLogger().info("获取类字段：" + fieldName);
- // Util.getLogger().info("获取类字段1：" + propertyDescriptor.getDisplayName());
- // Util.getLogger().info("获取的数据库数据：" + rs.getString(fieldName));
- Field declaredField = o.getClass().getDeclaredField(fieldName);
- if (enable) {
- value = ResultMapper.typeHandler.get(propertyType) == null ? null : ResultMapper.typeHandler.get(propertyType).getValue(rs, Util.toUnderlineCase(fieldName), declaredField);
- } else {
- value = ResultMapper.typeHandler.get(propertyType) == null ? null : ResultMapper.typeHandler.get(propertyType).getValue(rs, fieldName, declaredField);
- }
- propertyDescriptor.getWriteMethod().invoke(o, value);
- }
-
- } catch (Exception e) {
- e.printStackTrace();
- Util.getLogger().error("报错了，写入数据到实体类报错！\n" + Util.getErrString(e));
- }
- return o;
- }
-
- public Object getObject(RecordSet rs, Object o, Method method) {
- CaseConversion annotation = method.getAnnotation(CaseConversion.class);
- boolean enable = annotation == null ? true : annotation.value();
- String[] columnName = rs.getColumnName();
- String[] columnTypeName = rs.getColumnTypeName();
- int[] columnTypes = rs.getColumnType();
- if (columnTypeName == null) {
- columnTypeName = new String[columnTypes.length];
- for (int i = 0; i < columnTypes.length; i++) {
- int type = columnTypes[i];
- switch (type) {
- case -1:
- columnTypeName[i] = "TEXT";
- break;
- case 4:
- columnTypeName[i] = "INT";
- break;
- case 12:
- columnTypeName[i] = "VARCHAR";
- break;
- default:
- columnTypeName[i] = "VARCHAR";
- }
- }
- }
-
- try {
- if (o instanceof Map) {
- for (int i = 0; i < columnName.length; i++) {
- String columnType = "";
- if(i >= columnTypeName.length){
- columnType = "varchar";
- }else {
- columnType = columnTypeName[i];
- }
- if ("int".equalsIgnoreCase(columnType) || "long".equalsIgnoreCase(columnType) || "number".equalsIgnoreCase(columnType) || "MEDIUMINT".equalsIgnoreCase(columnType) || "TINYINT".equalsIgnoreCase(columnType) || "SMALLINT".equalsIgnoreCase(columnType) || "BIGINT".equalsIgnoreCase(columnType) || "INTEGER".equalsIgnoreCase(columnType)) {
- if (enable) {
- ((Map) o).put(Util.toCamelCase(columnName[i]), rs.getInt(i + 1));
- continue;
- }
- ((Map) o).put(columnName[i], rs.getInt(i + 1));
- continue;
- }
- if ("FLOAT".equalsIgnoreCase(columnType) || "DOUBLE".equalsIgnoreCase(columnType) || "DECIMAL".equalsIgnoreCase(columnType)) {
- if (enable) {
- ((Map) o).put(Util.toCamelCase(columnName[i]), rs.getFloat(i + 1));
- continue;
- }
- ((Map) o).put(columnName[i], rs.getFloat(i + 1));
- continue;
- }
- if (enable) {
- ((Map) o).put(Util.toCamelCase(columnName[i]), rs.getString(i + 1));
- continue;
- }
- ((Map) o).put(columnName[i], rs.getString(i + 1));
- continue;
- }
- return o;
- }
- if (o instanceof Collection) {
- throw new TypeNonsupportException("An unsupported return type!");
- }
- if (o instanceof Number) {
- for (int i = 0; i < columnName.length; i++) {
- String columnType = columnTypeName[i];
- if ("int".equalsIgnoreCase(columnType) || "long".equalsIgnoreCase(columnType) || "number".equalsIgnoreCase(columnType) || "MEDIUMINT".equalsIgnoreCase(columnType) || "TINYINT".equalsIgnoreCase(columnType) || "SMALLINT".equalsIgnoreCase(columnType) || "BIGINT".equalsIgnoreCase(columnType) || "INTEGER".equalsIgnoreCase(columnType)) {
- return rs.getInt(i + 1);
- }
- if ("FLOAT".equalsIgnoreCase(columnType) || "DOUBLE".equalsIgnoreCase(columnType) || "DECIMAL".equalsIgnoreCase(columnType)) {
- return rs.getFloat(i + 1);
- }
- }
- return o;
- }
- if (o instanceof String) {
- for (int i = 0; i < columnName.length; i++) {
- return rs.getString(i + 1);
- }
- return o;
- }
- if (o instanceof Boolean) {
- for (int i = 0; i < columnName.length; i++) {
- return rs.getBoolean(i + 1);
- }
- return o;
- }
-
- // Util.getLogger().info("获取对象：" + o.toString());
- BeanInfo beanInfo = Introspector.getBeanInfo(o.getClass(), Object.class);
- PropertyDescriptor[] propertyDescriptors = beanInfo.getPropertyDescriptors();
- for (PropertyDescriptor propertyDescriptor : propertyDescriptors) {
- Class propertyType = propertyDescriptor.getPropertyType();
- Object value = null;
- String fieldName = propertyDescriptor.getName();
-
- if (Strings.isNullOrEmpty(fieldName)) {
- fieldName = propertyDescriptor.getDisplayName();
- }
- // Util.getLogger().info("获取类字段：" + fieldName);
- // Util.getLogger().info("获取类字段1：" + propertyDescriptor.getDisplayName());
- // Util.getLogger().info("获取的数据库数据：" + rs.getString(fieldName));
- Field declaredField = o.getClass().getDeclaredField(fieldName);
- if (enable) {
- value = ResultMapper.typeHandler.get(propertyType) == null ? null : ResultMapper.typeHandler.get(propertyType).getValue(rs, Util.toUnderlineCase(fieldName), declaredField);
- } else {
- value = ResultMapper.typeHandler.get(propertyType) == null ? null : ResultMapper.typeHandler.get(propertyType).getValue(rs, fieldName, declaredField);
- }
- propertyDescriptor.getWriteMethod().invoke(o, value);
- }
-
- } catch (Exception e) {
- e.printStackTrace();
- Util.getLogger().error("报错了，写入数据到实体类报错！\n" + Util.getErrString(e));
- }
- return o;
- }
-
- public Class getRawType(Type type) {
- if (type instanceof Class) {
- return (Class) type;
- } else if (type instanceof ParameterizedType) {
- ParameterizedType parameterizedType = (ParameterizedType) type;
- Type rawType = parameterizedType.getRawType();
- return (Class) rawType;
- } else if (type instanceof GenericArrayType) {
- Type componentType = ((GenericArrayType) type).getGenericComponentType();
- return Array.newInstance(getRawType(componentType), 0).getClass();
- } else if (type instanceof TypeVariable) {
- return Object.class;
- } else if (type instanceof WildcardType) {
- return getRawType(((WildcardType) type).getUpperBounds()[0]);
- } else {
- String className = type == null ? "null" : type.getClass().getName();
- throw new IllegalArgumentException("Expected a Class, ParameterizedType, or GenericArrayType, but <" + type + "> is of type " + className);
- }
- }
+
+ private static final Map, TypeHandler> typeHandler = new HashMap<>();
+ private static final Map, Function> paramType = new HashMap<>();
+
+ private RecordsetUtil recordsetUtil = null;
+
+ static {
+ IntegerTypeHandler integerTypeHandler = new IntegerTypeHandler();
+ typeHandler.put(String.class, new StringTypeHandler());
+ typeHandler.put(Integer.class, integerTypeHandler);
+ typeHandler.put(int.class, integerTypeHandler);
+ typeHandler.put(byte.class, integerTypeHandler);
+ typeHandler.put(short.class, integerTypeHandler);
+ typeHandler.put(long.class, integerTypeHandler);
+ typeHandler.put(Long.class, integerTypeHandler);
+ typeHandler.put(Boolean.class, new BooleanTypeHandler());
+ typeHandler.put(boolean.class, new BooleanTypeHandler());
+ typeHandler.put(Date.class, new DataTypeHandler());
+ typeHandler.put(Float.class, new FloatTypeHandler());
+ typeHandler.put(float.class, new FloatTypeHandler());
+ }
+
+ static {
+ paramType.put(String.class, value -> value);
+ paramType.put(Integer.class, Integer::parseInt);
+ paramType.put(int.class, Integer::parseInt);
+ paramType.put(byte.class, Byte::parseByte);
+ paramType.put(short.class, Short::parseShort);
+ paramType.put(long.class, Long::parseLong);
+ paramType.put(Boolean.class, Boolean::parseBoolean);
+ paramType.put(boolean.class, Boolean::parseBoolean);
+ paramType.put(Float.class, Float::parseFloat);
+ paramType.put(float.class, Float::parseFloat);
+ }
+
+ public T mapperResult(RecordSet rs, Method method, Class tClass, RecordsetUtil recordsetUtil) {
+ this.recordsetUtil = recordsetUtil;
+ if (tClass.equals(void.class)) {
+ return null;
+ }
+ if (tClass.equals(RecordSet.class) || tClass.equals(RecordSetTrans.class)) {
+ return (T) rs;
+ }
+ try {
+ if (tClass.equals(List.class)) {
+ tClass = (Class) ArrayList.class;
+ }
+ if (tClass.equals(Map.class)) {
+ tClass = (Class) HashMap.class;
+ }
+ if (ResultMapper.typeHandler.containsKey(tClass)) {
+ rs.next();
+ return (T) ResultMapper.typeHandler.get(tClass).getValue(rs, 1, null);
+ }
+ T t = tClass.newInstance();
+ if (t instanceof Collection) {
+ // 集合求出泛型
+ // 获取返回值的类型
+ Type genericReturnType = method.getGenericReturnType();
+ Type actualTypeArgument = ((ParameterizedType) genericReturnType).getActualTypeArguments()[0];
+ Class rawType = this.getRawType(actualTypeArgument);
+ if (rawType.equals(Map.class)) {
+ rawType = HashMap.class;
+ }
+ while (rs.next()) {
+ Object o = null;
+ try {
+ Constructor constructor = rawType.getConstructor();
+ o = constructor.newInstance();
+ } catch (NoSuchMethodException | InvocationTargetException ignored) {
+ if (Number.class.isAssignableFrom(rawType)) {
+ Constructor constructor;
+ try {
+ constructor = rawType.getConstructor(String.class);
+ o = constructor.newInstance("-1");
+ } catch (NoSuchMethodException | InvocationTargetException e) {
+ throw new CustomerException("can not Initialization " + t.getClass() + " [" + rawType + "]", e);
+ }
+ }
+ }
+ if (o == null) {
+ throw new CustomerException("can not Initialization " + t.getClass() + " [" + rawType + "]");
+ }
+ Object object = getObject(rs, o, method);
+
+ ((Collection) t).add(object);
+ }
+ return t;
+ }
+ if (t instanceof Map) {
+ // map
+ Type genericReturnType = method.getGenericReturnType();
+ Type actualTypeArgument = ((ParameterizedType) genericReturnType).getActualTypeArguments()[0];
+ Class rawType = this.getRawType(actualTypeArgument);
+ if (rawType.equals(List.class)) {
+ rawType = ArrayList.class;
+ }
+ if (rawType.equals(Map.class)) {
+ rawType = HashMap.class;
+ }
+ // Object o = rawType.newInstance();
+ Object o = null;
+ try {
+ Constructor constructor = rawType.getConstructor();
+ o = constructor.newInstance();
+ } catch (NoSuchMethodException | InvocationTargetException ignored) {
+ if (Number.class.isAssignableFrom(rawType)) {
+ Constructor constructor;
+ try {
+ constructor = rawType.getConstructor(String.class);
+ o = constructor.newInstance("-1");
+ } catch (NoSuchMethodException | InvocationTargetException e) {
+ throw new CustomerException("can not Initialization " + t.getClass() + " [" + rawType + "]", e);
+ }
+ }
+ }
+ if (o == null) {
+ throw new CustomerException("can not Initialization " + t.getClass() + " [" + rawType + "]");
+ }
+ if (o instanceof Map || o instanceof Collection) {
+ throw new TypeNonsupportException("An unsupported return type!");
+ }
+ if (rs.next()) {
+ return (T) getObject(rs, t, method);
+ }
+ return null;
+ }
+ if (t.getClass().isArray()) {
+ throw new TypeNonsupportException("An unsupported return type!");
+ }
+ if (rs.next()) {
+ return (T) getObject(rs, t, method);
+ }
+ return null;
+ } catch (InstantiationException | IllegalAccessException e) {
+ e.printStackTrace();
+ }
+ return null;
+ }
+
+ public T mapperResult(RecordSetTrans rs, Method method, Class tClass, RecordsetUtil recordsetUtil) {
+ this.recordsetUtil = recordsetUtil;
+ if (tClass.equals(void.class)) {
+ return null;
+ }
+ if (tClass.equals(RecordSet.class) || tClass.equals(RecordSetTrans.class)) {
+ return (T) rs;
+ }
+ try {
+ if (tClass.equals(List.class)) {
+ tClass = (Class) ArrayList.class;
+ }
+ if (tClass.equals(Map.class)) {
+ tClass = (Class) HashMap.class;
+ }
+ if (ResultMapper.typeHandler.containsKey(tClass)) {
+ rs.next();
+ return (T) ResultMapper.typeHandler.get(tClass).getValue(rs, 1, null);
+ }
+ T t = tClass.newInstance();
+ if (t instanceof Collection) {
+ // 集合求出泛型
+ // 获取返回值的类型
+ Type genericReturnType = method.getGenericReturnType();
+ Type actualTypeArgument = ((ParameterizedType) genericReturnType).getActualTypeArguments()[0];
+ Class rawType = this.getRawType(actualTypeArgument);
+ if (rawType.equals(Map.class)) {
+ rawType = HashMap.class;
+ }
+ while (rs.next()) {
+ Object o = null;
+ try {
+ Constructor constructor = rawType.getConstructor();
+ o = constructor.newInstance();
+ } catch (NoSuchMethodException | InvocationTargetException ignored) {
+ if (Number.class.isAssignableFrom(rawType)) {
+ Constructor constructor;
+ try {
+ constructor = rawType.getConstructor(String.class);
+ o = constructor.newInstance("-1");
+ } catch (NoSuchMethodException | InvocationTargetException e) {
+ throw new CustomerException("can not Initialization " + t.getClass() + " [" + rawType + "]", e);
+ }
+ }
+ }
+ if (o == null) {
+ throw new CustomerException("can not Initialization " + t.getClass() + " [" + rawType + "]");
+ }
+ Object object = getObjectTrans(rs, o, method);
+ ((Collection) t).add(object);
+ }
+ return t;
+ }
+ if (t instanceof Map) {
+ // map
+ Type genericReturnType = method.getGenericReturnType();
+ Type actualTypeArgument = ((ParameterizedType) genericReturnType).getActualTypeArguments()[0];
+ Class rawType = this.getRawType(actualTypeArgument);
+ if (rawType.equals(List.class)) {
+ rawType = ArrayList.class;
+ }
+ if (rawType.equals(Map.class)) {
+ rawType = HashMap.class;
+ }
+ // Object o = rawType.newInstance();
+ Object o = null;
+ try {
+ Constructor constructor = rawType.getConstructor();
+ o = constructor.newInstance();
+ } catch (NoSuchMethodException | InvocationTargetException ignored) {
+ if (Number.class.isAssignableFrom(rawType)) {
+ Constructor constructor;
+ try {
+ constructor = rawType.getConstructor(String.class);
+ o = constructor.newInstance("-1");
+ } catch (NoSuchMethodException | InvocationTargetException e) {
+ throw new CustomerException("can not Initialization " + t.getClass() + " [" + rawType + "]", e);
+ }
+ }
+ }
+ if (o == null) {
+ throw new CustomerException("can not Initialization " + t.getClass() + " [" + rawType + "]");
+ }
+ if (o instanceof Map || o instanceof Collection) {
+ throw new TypeNonsupportException("An unsupported return type!");
+ }
+ if (rs.next()) {
+ return (T) getObjectTrans(rs, t, method);
+ }
+ return null;
+ }
+ if (t.getClass().isArray()) {
+ throw new TypeNonsupportException("An unsupported return type!");
+ }
+ if (rs.next()) {
+ return (T) getObjectTrans(rs, t, method);
+ }
+ return null;
+ } catch (InstantiationException | IllegalAccessException e) {
+ e.printStackTrace();
+ }
+ return null;
+ }
+
+ public Object getObjectTrans(RecordSetTrans rs, Object o, Method method) {
+ CaseConversion annotation = method.getAnnotation(CaseConversion.class);
+ boolean enable = annotation == null || annotation.value();
+ String[] columnName = rs.getColumnName();
+ String[] columnTypeName = rs.getColumnTypeName();
+ int[] columnTypes = rs.getColumnType();
+ if (columnTypeName == null) {
+ columnTypeName = new String[columnTypes.length];
+ for (int i = 0; i < columnTypes.length; i++) {
+ int type = columnTypes[i];
+ switch (type) {
+ case -1:
+ columnTypeName[i] = "TEXT";
+ break;
+ case 4:
+ columnTypeName[i] = "INT";
+ break;
+ case 12:
+ columnTypeName[i] = "VARCHAR";
+ break;
+ default:
+ columnTypeName[i] = "VARCHAR";
+ }
+ }
+ }
+
+ try {
+ if (o instanceof Map) {
+ for (int i = 0; i < columnName.length; i++) {
+ String columnType = columnTypeName[i];
+ if ("int".equalsIgnoreCase(columnType) || "long".equalsIgnoreCase(columnType) || "number".equalsIgnoreCase(columnType) || "MEDIUMINT".equalsIgnoreCase(columnType) || "TINYINT".equalsIgnoreCase(columnType) || "SMALLINT".equalsIgnoreCase(columnType) || "BIGINT".equalsIgnoreCase(columnType) || "INTEGER".equalsIgnoreCase(columnType)) {
+ if (enable) {
+ ((Map) o).put(Util.toCamelCase(columnName[i]), rs.getInt(i + 1));
+ continue;
+ }
+ ((Map) o).put(columnName[i].toLowerCase(), rs.getInt(i + 1));
+ continue;
+ }
+ if ("FLOAT".equalsIgnoreCase(columnType) || "DOUBLE".equalsIgnoreCase(columnType) || "DECIMAL".equalsIgnoreCase(columnType)) {
+ if (enable) {
+ ((Map) o).put(Util.toCamelCase(columnName[i]), rs.getFloat(i + 1));
+ continue;
+ }
+ ((Map) o).put(columnName[i].toLowerCase(), rs.getFloat(i + 1));
+ continue;
+ }
+ if (enable) {
+ ((Map) o).put(Util.toCamelCase(columnName[i]), rs.getString(i + 1));
+ continue;
+ }
+ ((Map) o).put(columnName[i].toLowerCase(), rs.getString(i + 1));
+ continue;
+ }
+ return o;
+ }
+ if (o instanceof Collection) {
+ throw new TypeNonsupportException("An unsupported return type!");
+ }
+ if (o instanceof Number) {
+ for (int i = 0; i < columnName.length; i++) {
+ String columnType = columnTypeName[i];
+ if ("int".equalsIgnoreCase(columnType) || "long".equalsIgnoreCase(columnType) || "number".equalsIgnoreCase(columnType) || "MEDIUMINT".equalsIgnoreCase(columnType) || "TINYINT".equalsIgnoreCase(columnType) || "SMALLINT".equalsIgnoreCase(columnType) || "BIGINT".equalsIgnoreCase(columnType) || "INTEGER".equalsIgnoreCase(columnType)) {
+ return rs.getInt(i + 1);
+ }
+ if ("FLOAT".equalsIgnoreCase(columnType) || "DOUBLE".equalsIgnoreCase(columnType) || "DECIMAL".equalsIgnoreCase(columnType)) {
+ return rs.getFloat(i + 1);
+ }
+ }
+ return o;
+ }
+ if (o instanceof String) {
+ for (int i = 0; i < columnName.length; i++) {
+ return rs.getString(i + 1);
+ }
+ return o;
+ }
+ if (o instanceof Boolean) {
+ for (int i = 0; i < columnName.length; i++) {
+ return rs.getBoolean(i + 1);
+ }
+ return o;
+ }
+
+ // Util.getLogger().info("获取对象：" + o.toString());
+ BeanInfo beanInfo = Introspector.getBeanInfo(o.getClass(), Object.class);
+ PropertyDescriptor[] propertyDescriptors = beanInfo.getPropertyDescriptors();
+ for (PropertyDescriptor propertyDescriptor : propertyDescriptors) {
+ Class propertyType = propertyDescriptor.getPropertyType();
+ Object value = null;
+ String fieldName = propertyDescriptor.getName();
+
+ if (Strings.isNullOrEmpty(fieldName)) {
+ fieldName = propertyDescriptor.getDisplayName();
+ }
+ Field declaredField = o.getClass().getDeclaredField(fieldName);
+ if (enable) {
+ value = ResultMapper.typeHandler.get(propertyType) == null ? null : ResultMapper.typeHandler.get(propertyType).getValue(rs, Util.toUnderlineCase(fieldName), declaredField);
+ } else {
+ value = ResultMapper.typeHandler.get(propertyType) == null ? null : ResultMapper.typeHandler.get(propertyType).getValue(rs, fieldName, declaredField);
+ }
+ propertyDescriptor.getWriteMethod().invoke(o, value);
+ }
+
+ } catch (Exception e) {
+ Util.getLogger().error("报错了，写入数据到实体类报错！\n" + Util.getErrString(e));
+ throw new CustomerException(e);
+ }
+ return o;
+ }
+
+ public Object getObject(RecordSet rs, Object o, Method method) {
+ CaseConversion annotation = method.getAnnotation(CaseConversion.class);
+ boolean enable = annotation == null || annotation.value();
+ String[] columnName = rs.getColumnName();
+ String[] columnTypeName = rs.getColumnTypeName();
+ int[] columnTypes = rs.getColumnType();
+ if (columnTypeName == null) {
+ columnTypeName = new String[columnTypes.length];
+ for (int i = 0; i < columnTypes.length; i++) {
+ int type = columnTypes[i];
+ switch (type) {
+ case -1:
+ columnTypeName[i] = "TEXT";
+ break;
+ case 4:
+ columnTypeName[i] = "INT";
+ break;
+ case 12:
+ columnTypeName[i] = "VARCHAR";
+ break;
+ default:
+ columnTypeName[i] = "VARCHAR";
+ }
+ }
+ }
+
+ try {
+ if (o instanceof Map) {
+ for (int i = 0; i < columnName.length; i++) {
+ String columnType = "";
+ if (i >= columnTypeName.length) {
+ columnType = "varchar";
+ } else {
+ columnType = columnTypeName[i];
+ }
+ if ("int".equalsIgnoreCase(columnType) || "long".equalsIgnoreCase(columnType) || "number".equalsIgnoreCase(columnType) || "MEDIUMINT".equalsIgnoreCase(columnType) || "TINYINT".equalsIgnoreCase(columnType) || "SMALLINT".equalsIgnoreCase(columnType) || "BIGINT".equalsIgnoreCase(columnType) || "INTEGER".equalsIgnoreCase(columnType)) {
+ if (enable) {
+ ((Map) o).put(Util.toCamelCase(columnName[i]), rs.getInt(i + 1));
+ continue;
+ }
+ ((Map) o).put(columnName[i].toLowerCase(), rs.getInt(i + 1));
+ continue;
+ }
+ if ("FLOAT".equalsIgnoreCase(columnType) || "DOUBLE".equalsIgnoreCase(columnType) || "DECIMAL".equalsIgnoreCase(columnType)) {
+ if (enable) {
+ ((Map) o).put(Util.toCamelCase(columnName[i]), rs.getFloat(i + 1));
+ continue;
+ }
+ ((Map) o).put(columnName[i].toLowerCase(), rs.getFloat(i + 1));
+ continue;
+ }
+ if (method.isAnnotationPresent(Associations.class)) {
+ Association association = searchAssociation(method, columnName[i], true);
+ if (association != null) {
+ if (association.column().equalsIgnoreCase(columnName[i])) {
+ Object cassociationValue = association(rs, association, method);
+ ((Map) o).put(association.property(), cassociationValue);
+ continue;
+ }
+ }
+ }
+
+ if (method.isAnnotationPresent(CollectionMappings.class)) {
+ CollectionMapping collectionMapping = searchCollection(method, columnName[i], true);
+ if (collectionMapping != null) {
+ if (collectionMapping.column().equals(columnName[i])) {
+ Object collection = collection(rs, collectionMapping, method);
+ String property = collectionMapping.property();
+ ((Map) o).put(property, collection);
+ continue;
+ }
+ }
+ }
+ if (enable) {
+ ((Map) o).put(Util.toCamelCase(columnName[i]), rs.getString(i + 1));
+ continue;
+ }
+ ((Map) o).put(columnName[i].toLowerCase(), rs.getString(i + 1));
+ }
+ return o;
+ }
+ if (o instanceof Collection) {
+ throw new TypeNonsupportException("An unsupported return type!");
+ }
+ if (o instanceof Number) {
+ for (int i = 0; i < columnName.length; i++) {
+ String columnType = columnTypeName[i];
+ if ("int".equalsIgnoreCase(columnType) || "long".equalsIgnoreCase(columnType) || "number".equalsIgnoreCase(columnType) || "MEDIUMINT".equalsIgnoreCase(columnType) || "TINYINT".equalsIgnoreCase(columnType) || "SMALLINT".equalsIgnoreCase(columnType) || "BIGINT".equalsIgnoreCase(columnType) || "INTEGER".equalsIgnoreCase(columnType)) {
+ return rs.getInt(i + 1);
+ }
+ if ("FLOAT".equalsIgnoreCase(columnType) || "DOUBLE".equalsIgnoreCase(columnType) || "DECIMAL".equalsIgnoreCase(columnType)) {
+ return rs.getFloat(i + 1);
+ }
+ }
+ return o;
+ }
+ if (o instanceof String) {
+ for (int i = 0; i < columnName.length; i++) {
+ return rs.getString(i + 1);
+ }
+ return o;
+ }
+ if (o instanceof Boolean) {
+ for (int i = 0; i < columnName.length; i++) {
+ return rs.getBoolean(i + 1);
+ }
+ return o;
+ }
+
+ // Util.getLogger().info("获取对象：" + o.toString());
+ BeanInfo beanInfo = Introspector.getBeanInfo(o.getClass(), Object.class);
+ PropertyDescriptor[] propertyDescriptors = beanInfo.getPropertyDescriptors();
+
+ for (PropertyDescriptor propertyDescriptor : propertyDescriptors) {
+ Class propertyType = propertyDescriptor.getPropertyType();
+ Object value = null;
+ String fieldName = propertyDescriptor.getName();
+
+ if (Strings.isNullOrEmpty(fieldName)) {
+ fieldName = propertyDescriptor.getDisplayName();
+ }
+ if (method.isAnnotationPresent(Associations.class)) {
+ Association association = searchAssociation(method, fieldName, false);
+ if (association != null) {
+ if (association.property().equals(fieldName)) {
+ Object cassociationValue = association(rs, association, method);
+ propertyDescriptor.getWriteMethod().invoke(o, cassociationValue);
+ continue;
+ }
+ }
+ }
+ if (method.isAnnotationPresent(CollectionMappings.class)) {
+ CollectionMapping collectionMapping = searchCollection(method, fieldName, false);
+ if (collectionMapping != null) {
+ if (fieldName.equals(collectionMapping.property()) && !"".equals(collectionMapping.property())) {
+ Object collection = collection(rs, collectionMapping, method);
+ propertyDescriptor.getWriteMethod().invoke(o, collection);
+ continue;
+ }
+ }
+ }
+ Field declaredField = o.getClass().getDeclaredField(fieldName);
+ if (enable) {
+ value = ResultMapper.typeHandler.get(propertyType) == null ? null : ResultMapper.typeHandler.get(propertyType).getValue(rs, Util.toUnderlineCase(fieldName), declaredField);
+ } else {
+ value = ResultMapper.typeHandler.get(propertyType) == null ? null : ResultMapper.typeHandler.get(propertyType).getValue(rs, fieldName, declaredField);
+ }
+ propertyDescriptor.getWriteMethod().invoke(o, value);
+ }
+
+
+ } catch (Exception e) {
+ Util.getLogger().error("报错了，写入数据到实体类报错！\n" + Util.getErrString(e));
+ throw new CustomerException(e.getMessage(), e);
+ }
+ return o;
+ }
+
+ public Class getRawType(Type type) {
+ if (type instanceof Class) {
+ return (Class) type;
+ } else if (type instanceof ParameterizedType) {
+ ParameterizedType parameterizedType = (ParameterizedType) type;
+ Type rawType = parameterizedType.getRawType();
+ return (Class) rawType;
+ } else if (type instanceof GenericArrayType) {
+ Type componentType = ((GenericArrayType) type).getGenericComponentType();
+ return Array.newInstance(getRawType(componentType), 0).getClass();
+ } else if (type instanceof TypeVariable) {
+ return Object.class;
+ } else if (type instanceof WildcardType) {
+ return getRawType(((WildcardType) type).getUpperBounds()[0]);
+ } else {
+ String className = type == null ? "null" : type.getClass().getName();
+ throw new IllegalArgumentException("Expected a Class, ParameterizedType, or GenericArrayType, but <" + type + "> is of type " + className);
+ }
+ }
+
+
+ private Association searchAssociation(Method method, String filedName, boolean isMap) {
+
+ Associations annotation = method.getAnnotation(Associations.class);
+ Association[] mappings = annotation.value();
+ Association mapping = null;
+ for (Association item : mappings) {
+ String property = isMap ? item.column() : item.property();
+ if (isMap ? filedName.equalsIgnoreCase(property) : filedName.equals(property)) {
+ mapping = item;
+ }
+ }
+ return mapping;
+ }
+
+ private Object association(RecordSet rs, Association annotation, Method method) {
+ Id id = annotation.id();
+ String column = annotation.column();
+ String columnValue = rs.getString(column);
+ if (Objects.isNull(columnValue) || "".equals(columnValue)) {
+ return null;
+ }
+ if (id.methodId() != -1) {
+ Class declaringClass = method.getDeclaringClass();
+ Method[] declaredMethods = declaringClass.getDeclaredMethods();
+ for (Method declaredMethod : declaredMethods) {
+ if (declaredMethod.isAnnotationPresent(AssociationMethod.class)) {
+ AssociationMethod collectionMethod = declaredMethod.getAnnotation(AssociationMethod.class);
+ int value = collectionMethod.value();
+ if (id.methodId() == value) {
+ return recordsetUtil.invoke(null, declaredMethod,
+ new Object[]{paramType.get(id.value()).apply(columnValue)}, declaredMethod.getName());
+ }
+ }
+ }
+ }
+ String selectMethod = annotation.select();
+ int i = selectMethod.lastIndexOf(".");
+ String selectClass = selectMethod.substring(0, i);
+ Class aClass = null;
+ try {
+ aClass = Class.forName(selectClass);
+ } catch (ClassNotFoundException e) {
+ throw new RuntimeException(e);
+ }
+ Method associationMethod = null;
+ try {
+ associationMethod = aClass.getMethod(selectMethod.substring(i + 1), id.value());
+ } catch (NoSuchMethodException e) {
+ throw new RuntimeException(e);
+ }
+ Class returnType = associationMethod.getReturnType();
+ if (List.class.isAssignableFrom(returnType)) {
+ throw new CustomerException("can not set result, association annotation not support result of list! Do you want to use @CollectionSql!");
+ }
+ return recordsetUtil.invoke(null, associationMethod,
+ new Object[]{paramType.get(id.value()).apply(columnValue)}, selectMethod);
+ }
+
+
+ private CollectionMapping searchCollection(Method method, String filedName, boolean isMap) {
+ CollectionMappings annotation = method.getAnnotation(CollectionMappings.class);
+ CollectionMapping[] mappings = annotation.value();
+ CollectionMapping mapping = null;
+ for (CollectionMapping item : mappings) {
+ String property = isMap ? item.column() : item.property();
+ if (isMap ? filedName.equalsIgnoreCase(property) : filedName.equals(property)) {
+ mapping = item;
+ }
+ }
+ return mapping;
+ }
+
+ private Object collection(RecordSet rs, CollectionMapping annotation, Method method) {
+ Id id = annotation.id();
+ String column = annotation.column();
+ String columnValue = rs.getString(column);
+ if (Objects.isNull(columnValue) || "".equals(columnValue)) {
+ return null;
+ }
+ if (id.methodId() != -1) {
+ Class declaringClass = method.getDeclaringClass();
+ Method[] declaredMethods = declaringClass.getDeclaredMethods();
+ for (Method declaredMethod : declaredMethods) {
+ if (declaredMethod.isAnnotationPresent(CollectionMethod.class)) {
+ CollectionMethod collectionMethod = declaredMethod.getAnnotation(CollectionMethod.class);
+ int value = collectionMethod.value();
+ if (id.methodId() == value) {
+ return recordsetUtil.invoke(null, declaredMethod,
+ new Object[]{paramType.get(id.value()).apply(columnValue)}, declaredMethod.getName());
+ }
+ }
+ }
+ }
+ String selectMethod = annotation.select();
+ int i = selectMethod.lastIndexOf(".");
+ String selectClass = selectMethod.substring(0, i);
+ Class aClass = null;
+ try {
+ aClass = Class.forName(selectClass);
+ } catch (ClassNotFoundException e) {
+ throw new RuntimeException(e);
+ }
+ Method associationMethod = null;
+ try {
+ associationMethod = aClass.getMethod(selectMethod.substring(i + 1), id.value());
+ } catch (NoSuchMethodException e) {
+ throw new RuntimeException(e);
+ }
+ Class returnType = associationMethod.getReturnType();
+ if (!List.class.isAssignableFrom(returnType)) {
+ throw new CustomerException("can not set result, CollectionSql annotation only support result of list! Do you want to use @Association!");
+ }
+ return recordsetUtil.invoke(null, associationMethod,
+ new Object[]{paramType.get(id.value()).apply(columnValue)}, selectMethod);
+ }
}
diff --git a/src/main/java/com/api/youhong/ai/pcn/racetrack/controller/RaceTrackController.java b/src/main/java/com/api/youhong/ai/pcn/racetrack/controller/RaceTrackController.java
index 5438a33..6b77b3b 100644
--- a/src/main/java/com/api/youhong/ai/pcn/racetrack/controller/RaceTrackController.java
+++ b/src/main/java/com/api/youhong/ai/pcn/racetrack/controller/RaceTrackController.java
@@ -44,4 +44,33 @@ public class RaceTrackController {
}
}
+
+ @GET
+ @Path("/get/stage-info")
+ @Produces(MediaType.APPLICATION_JSON)
+ @Consumes(MediaType.APPLICATION_JSON)
+ public String getStageInfo(@Context HttpServletRequest request, @Context HttpServletResponse response) {
+ User user = HrmUserVarify.getUser(request, response);
+ try {
+ return ApiResult.success(service.getStageInfo(user));
+ } catch (Exception e) {
+ log.info("race track get event list error!\n" + Util.getErrString(e));
+ return ApiResult.error("race track get event list error!");
+ }
+ }
+
+ @GET
+ @Path("/get/status")
+ @Produces(MediaType.APPLICATION_JSON)
+ @Consumes(MediaType.APPLICATION_JSON)
+ public String getUserStatus(@Context HttpServletRequest request, @Context HttpServletResponse response) {
+ try {
+ User user = HrmUserVarify.getUser(request, response);
+ return ApiResult.success(user.getStatus());
+ } catch (Exception e) {
+ log.info("get usr status fail!\n" + Util.getErrString(e));
+ return ApiResult.error("get usr status fail!");
+ }
+ }
+
}
diff --git a/src/main/java/com/api/youhong/ai/pcn/racetrack/dto/RaceTrackStage.java b/src/main/java/com/api/youhong/ai/pcn/racetrack/dto/RaceTrackStage.java
new file mode 100644
index 0000000..cf70d90
--- /dev/null
+++ b/src/main/java/com/api/youhong/ai/pcn/racetrack/dto/RaceTrackStage.java
@@ -0,0 +1,32 @@
+package com.api.youhong.ai.pcn.racetrack.dto;
+
+import lombok.Getter;
+import lombok.Setter;
+import lombok.ToString;
+
+/**
+ * 直道图数据库对象
+ *
+ * create: 2023/2/17 11:41
+ *
+ * @author youHong.ai
+ */
+@Setter
+@Getter
+@ToString
+public class RaceTrackStage {
+ /** id */
+ private int id;
+
+ /** 一阶段 */
+ private String oneStage;
+
+ /** 二阶段 */
+ private String towStage;
+
+ /** 三阶段 */
+ private String threeStage;
+ /** 四阶段 */
+ private String fourStage;
+
+}
diff --git a/src/main/java/com/api/youhong/ai/pcn/racetrack/mapper/RacetrackMapper.java b/src/main/java/com/api/youhong/ai/pcn/racetrack/mapper/RacetrackMapper.java
index f9b9867..bcfc4bd 100644
--- a/src/main/java/com/api/youhong/ai/pcn/racetrack/mapper/RacetrackMapper.java
+++ b/src/main/java/com/api/youhong/ai/pcn/racetrack/mapper/RacetrackMapper.java
@@ -4,6 +4,7 @@ import aiyh.utils.annotation.recordset.ParamMapper;
import aiyh.utils.annotation.recordset.Select;
import aiyh.utils.annotation.recordset.SqlMapper;
import com.api.youhong.ai.pcn.racetrack.dto.RaceTrackEvent;
+import com.api.youhong.ai.pcn.racetrack.dto.RaceTrackStage;
import java.util.List;
@@ -47,4 +48,19 @@ public interface RacetrackMapper {
*/
@Select("select companyworkyear from hrmresource where id = #{userId}")
String selectLengthOfEntryTime(@ParamMapper("userId") Integer userId);
+
+
+ @Select("select id, $t{raceTrackStageOneField} one_stage," +
+ "$t{raceTrackStageTwoField} tow_stage," +
+ "$t{raceTrackStageThreeField} three_stage ," +
+ "$t{raceTrackStageFourField} four_stage " +
+ "from $t{raceTrackStageTable} " +
+ "where $t{raceTrackStageUserField} = #{userId}")
+ RaceTrackStage selectStageInfoByUserId(@ParamMapper("userId") int uid,
+ @ParamMapper("raceTrackStageTable") String raceTrackStageTable,
+ @ParamMapper("raceTrackStageUserField") String raceTrackStageUserField,
+ @ParamMapper("raceTrackStageOneField") String raceTrackStageOneField,
+ @ParamMapper("raceTrackStageTwoField") String raceTrackStageTwoField,
+ @ParamMapper("raceTrackStageThreeField") String raceTrackStageThreeField,
+ @ParamMapper("raceTrackStageFourField") String raceTrackStageFourField);
}
diff --git a/src/main/java/com/api/youhong/ai/pcn/racetrack/service/RaceTrackService.java b/src/main/java/com/api/youhong/ai/pcn/racetrack/service/RaceTrackService.java
index 046d023..756cfb7 100644
--- a/src/main/java/com/api/youhong/ai/pcn/racetrack/service/RaceTrackService.java
+++ b/src/main/java/com/api/youhong/ai/pcn/racetrack/service/RaceTrackService.java
@@ -1,13 +1,18 @@
package com.api.youhong.ai.pcn.racetrack.service;
import aiyh.utils.Util;
-import cn.hutool.core.lang.Assert;
+import aiyh.utils.tool.Assert;
import com.api.youhong.ai.pcn.racetrack.dto.RaceTrackEvent;
+import com.api.youhong.ai.pcn.racetrack.dto.RaceTrackStage;
import com.api.youhong.ai.pcn.racetrack.mapper.RacetrackMapper;
+import com.api.youhong.ai.pcn.racetrack.vo.RaceTrackStageVo;
import com.api.youhong.ai.pcn.racetrack.vo.RaceTrackVo;
+import com.api.youhong.ai.pcn.racetrack.vo.StageInfoVo;
+import com.google.common.base.Strings;
import ebu7common.youhong.ai.bean.Builder;
import weaver.hrm.User;
+import java.util.ArrayList;
import java.util.List;
/**
@@ -54,6 +59,114 @@ public class RaceTrackService {
return Builder.builder(RaceTrackVo::new)
.with(RaceTrackVo::setEventList, raceTrackEventList)
.with(RaceTrackVo::setLengthOfEntryTime, lengthOfEntryTime)
+ .with(RaceTrackVo::setUserStatus, user.getStatus())
.build();
}
+
+ /**
+ * 获取阶段信息列表
+ *
+ * @param user 当前用户
+ * @return 阶段信息列表
+ */
+ public Object getStageInfo(User user) {
+ // 直到图数据表
+ String raceTrackStageTable = Util.getCusConfigValue("RACE_TRACK_STAGE_TABLE");
+ // 数据表用户字段
+ String raceTrackStageUserField = Util.getCusConfigValue("RACE_TRACK_STAGE_USER_FIELD");
+ // 一阶段列名
+ String raceTrackStageOneField = Util.getCusConfigValue("RACE_TRACK_STAGE_ONE_FIELD");
+ // 二阶段列名
+ String raceTrackStageTwoField = Util.getCusConfigValue("RACE_TRACK_STAGE_TWO_FIELD");
+ // 三阶段列名
+ String raceTrackStageThreeField = Util.getCusConfigValue("RACE_TRACK_STAGE_THREE_FIELD");
+ // 四阶段列名
+ String raceTrackStageFourField = Util.getCusConfigValue("RACE_TRACK_STAGE_Four_FIELD");
+ Assert.notBlank(raceTrackStageTable,
+ "race track stage info table can not be null! check configuration [RACE_TRACK_STAGE_TABLE] in uf_cus_dev_config table!");
+ Assert.notBlank(raceTrackStageUserField,
+ "race track stage info user field can not be null! check configuration [RACE_TRACK_STAGE_USER_FIELD] in uf_cus_dev_config table!");
+ Assert.notBlank(raceTrackStageOneField,
+ "race track stage info one stage filed can not be null! check configuration [RACE_TRACK_STAGE_ONE_FIELD] in uf_cus_dev_config table!");
+ Assert.notBlank(raceTrackStageTwoField,
+ "race track stage info tow stage field can not be null! check configuration [RACE_TRACK_STAGE_TWO_FIELD] in uf_cus_dev_config table!");
+ Assert.notBlank(raceTrackStageThreeField,
+ "race track stage info three stage field can not be null! check configuration [RACE_TRACK_STAGE_THREE_FIELD] in uf_cus_dev_config table!");
+ Assert.notBlank(raceTrackStageFourField,
+ "race track stage info four stage field can not be null! check configuration [RACE_TRACK_STAGE_Four_FIELD] in uf_cus_dev_config table!");
+ RaceTrackStage raceTrackStage = mapper.selectStageInfoByUserId(user.getUID(), raceTrackStageTable,
+ raceTrackStageUserField,
+ raceTrackStageOneField,
+ raceTrackStageTwoField,
+ raceTrackStageThreeField,
+ raceTrackStageFourField);
+ List stageInfoVos = pushRaceTrackVo(raceTrackStage);
+ return Builder.builder(RaceTrackStageVo::new)
+ .with(RaceTrackStageVo::setStageVoList, stageInfoVos)
+ .with(RaceTrackStageVo::setStatus, user.getStatus())
+ .build();
+ }
+
+
+ /**
+ * 添加阶段信息
+ *
+ * @param raceTrackStage 阶段信息
+ * @return 阶段信息列表
+ */
+ private List pushRaceTrackVo(RaceTrackStage raceTrackStage) {
+ List raceTrackVos = new ArrayList<>();
+ String oneStage = raceTrackStage.getOneStage();
+ String towStage = raceTrackStage.getTowStage();
+ String threeStage = raceTrackStage.getThreeStage();
+ String fourStage = raceTrackStage.getFourStage();
+ setRaceTrackValue(raceTrackVos, oneStage);
+ setRaceTrackValue(raceTrackVos, towStage);
+ setRaceTrackValue(raceTrackVos, threeStage);
+ setRaceTrackValue(raceTrackVos, fourStage);
+ for (int i = 0; i < raceTrackVos.size(); i++) {
+ StageInfoVo item = raceTrackVos.get(i);
+ if (i < raceTrackVos.size() - 1) {
+ if (item == null) {
+ raceTrackVos.set(i, Builder.builder(StageInfoVo::new)
+ .with(StageInfoVo::setFailedToArrive, true)
+ .build());
+ } else {
+ item.setActive(true);
+ }
+ } else {
+ if (item == null) {
+ item = Builder.builder(StageInfoVo::new)
+ .with(StageInfoVo::setFailedToArrive, true)
+ .build();
+ }
+ StageInfoVo nextItem = raceTrackVos.get(i + 1);
+ if (nextItem == null) {
+ item.setActive(true);
+ item.setFailedToArrive(false);
+ } else {
+ item.setPass(true);
+ }
+ }
+ }
+ return raceTrackVos;
+ }
+
+ /**
+ * 阶段信息值设置
+ *
+ * @param raceTrackVos 阶段信息列表
+ * @param stage 阶段字段值
+ */
+ private static void setRaceTrackValue(List raceTrackVos, String stage) {
+ if (Strings.isNullOrEmpty(stage)) {
+ raceTrackVos.add(null);
+ } else {
+ raceTrackVos.add(
+ Builder.builder(StageInfoVo::new)
+ .with(StageInfoVo::setTime, stage)
+ .build()
+ );
+ }
+ }
}
diff --git a/src/main/java/com/api/youhong/ai/pcn/racetrack/vo/RaceTrackStageVo.java b/src/main/java/com/api/youhong/ai/pcn/racetrack/vo/RaceTrackStageVo.java
new file mode 100644
index 0000000..0862caf
--- /dev/null
+++ b/src/main/java/com/api/youhong/ai/pcn/racetrack/vo/RaceTrackStageVo.java
@@ -0,0 +1,25 @@
+package com.api.youhong.ai.pcn.racetrack.vo;
+
+import lombok.Getter;
+import lombok.Setter;
+import lombok.ToString;
+
+import java.util.List;
+
+/**
+ * 直道图阶段信息
+ *
+ * create: 2023/2/17 11:06
+ *
+ * @author youHong.ai
+ */
+@Getter
+@Setter
+@ToString
+public class RaceTrackStageVo {
+ private int id;
+
+ private int status;
+
+ private List stageVoList;
+}
diff --git a/src/main/java/com/api/youhong/ai/pcn/racetrack/vo/RaceTrackVo.java b/src/main/java/com/api/youhong/ai/pcn/racetrack/vo/RaceTrackVo.java
index 70e1a1b..0cf965e 100644
--- a/src/main/java/com/api/youhong/ai/pcn/racetrack/vo/RaceTrackVo.java
+++ b/src/main/java/com/api/youhong/ai/pcn/racetrack/vo/RaceTrackVo.java
@@ -24,4 +24,7 @@ public class RaceTrackVo {
/** 入职时长 */
private String lengthOfEntryTime;
+
+
+ private int userStatus;
}
diff --git a/src/main/java/com/api/youhong/ai/pcn/racetrack/vo/StageInfoVo.java b/src/main/java/com/api/youhong/ai/pcn/racetrack/vo/StageInfoVo.java
new file mode 100644
index 0000000..053d84f
--- /dev/null
+++ b/src/main/java/com/api/youhong/ai/pcn/racetrack/vo/StageInfoVo.java
@@ -0,0 +1,26 @@
+package com.api.youhong.ai.pcn.racetrack.vo;
+
+import lombok.Getter;
+import lombok.Setter;
+import lombok.ToString;
+
+/**
+ *