Optimize code

Author: tangllty

tang-commons/src/main/java/com/tang/commons/annotation/poi/Excel.java

@@ -0,0 +1,68 @@
+package com.tang.commons.annotation.poi
+
+import org.springframework.core.annotation.AliasFor
+
+import com.tang.commons.enumeration.poi.CellType
+import com.tang.commons.enumeration.poi.Type
+
+/**
+ * Excel 注解
+ *
+ * @author Tang
+ */
+@Target(AnnotationTarget.FIELD)
+@Retention(AnnotationRetention.RUNTIME)
+@MustBeDocumented
+annotation class Excel(
+
+    /**
+     * Alias for [name]
+     */
+    @get:AliasFor("name") val value: String = "",
+
+    /**
+     * 列名称
+     */
+    @get:AliasFor("value") val name: String = "",
+
+    /**
+     * 顺序
+     */
+    val sort: Int = 0,
+
+    /**
+     * 单元格类型
+     */
+    val cellType: CellType = CellType.STRING,
+
+    /**
+     * 日期格式
+     */
+    val dateFormat: String = "yyyy-MM-dd HH:mm:ss",
+
+    /**
+     * 字段类型
+     */
+    val type: Type = Type.ALL,
+
+    /**
+     * 字典类型
+     */
+    val dictType: String = "",
+
+    /**
+     * 列宽
+     */
+    val width: Int = 10,
+
+    /**
+     * 行高
+     */
+    val height: Int = 16,
+
+    /**
+     * 标题行高
+     */
+    val titleHeight: Int = 16
+
+)

tang-commons/src/main/kotlin/com/tang/commons/annotation/poi/Excel.kt

@@ -0,0 +1,44 @@
+package com.tang.commons.enumeration.poi
+
+/**
+ * Excel 单元格类型
+ *
+ * @author Tang
+ */
+enum class CellType(val code: Int) {
+
+    /**
+     * 字符串
+     */
+    STRING(0),
+
+    /**
+     * 数字
+     */
+    NUMBER(1),
+
+    /**
+     * 日期
+     */
+    DATE(2);
+
+    companion object {
+
+        /**
+         * 根据 code 返回枚举类型
+         *
+         * @param code 单元格类型
+         * @return [CellType] 单元格类型
+         */
+        fun getCellType(code: Int): CellType? {
+            for (cellType in CellType.entries) {
+                if (cellType.code == code) {
+                    return cellType
+                }
+            }
+            return null
+        }
+
+    }
+
+}

tang-commons/src/main/kotlin/com/tang/commons/enumeration/poi/CellType.kt

@@ -0,0 +1,44 @@
+package com.tang.commons.enumeration.poi
+
+/**
+ * Excel 字段类型
+ *
+ * @author Tang
+ */
+enum class Type(val code: Int) {
+
+    /**
+     * 导入导出
+     */
+    ALL(0),
+
+    /**
+     * 导入
+     */
+    IMPORT(1),
+
+    /**
+     * 导出
+     */
+    EXPORT(2);
+
+    companion object {
+
+        /**
+         * 根据 code 返回枚举类型
+         *
+         * @param code 导出或导入
+         * @return [Type] 导出或导入
+         */
+        fun getType(code: Int): Type? {
+            for (type in Type.entries) {
+                if (type.code == code) {
+                    return type
+                }
+            }
+            return null
+        }
+
+    }
+
+}

tang-commons/src/main/kotlin/com/tang/commons/enumeration/poi/Type.kt

@@ -4,6 +4,8 @@ import com.google.common.base.CaseFormat
 import com.tang.commons.annotation.poi.Excel
 import com.tang.commons.constants.ContentType
 import com.tang.commons.constants.FileType
+import com.tang.commons.enumeration.poi.Type
+import com.tang.commons.enumeration.poi.CellType as ExcelCellType
 import com.tang.commons.exception.file.FileTypeMismatchException
 import com.tang.commons.utils.Assert
 import com.tang.commons.utils.DictUtils
@@ -70,7 +72,7 @@ object ExcelUtils {
             val cellNumIndex = AtomicInteger()
             fields.forEach { (field: Field, excel: Excel) ->
                 val cell = row.getCell(cellNumIndex.getAndIncrement())
-                if (excel.type == Excel.Type.EXPORT) {
+                if (excel.type == Type.EXPORT) {
                     return@forEach
                 }
                 ReflectionUtils.makeAccessible(field)
@@ -289,7 +291,7 @@ object ExcelUtils {
         ReflectionUtils.makeAccessible(field)
         val stringValue = Objects.toString(field.get(clazz), "")
         when (excel.cellType) {
-            Excel.CellType.STRING -> {
+            ExcelCellType.STRING -> {
                 val dictType = excel.dictType
                 if (StringUtils.isBlank(dictType)) {
                     cell.setCellValue(stringValue)
@@ -304,13 +306,13 @@ object ExcelUtils {
                 cell.setCellValue(DictUtils.getDictLabel(dictType, stringValue))
                 cell.sheet.addValidationData(validation)
             }
-            Excel.CellType.NUMBER -> {
+            ExcelCellType.NUMBER -> {
                 if (StringUtils.isBlank(stringValue)) {
                     return
                 }
                 cell.setCellValue(stringValue.toDouble())
             }
-            Excel.CellType.DATE -> {
+            ExcelCellType.DATE -> {
                 if (StringUtils.isBlank(stringValue)) {
                     return
                 }

tang-commons/src/main/kotlin/com/tang/commons/utils/poi/ExcelUtils.kt

@@ -6,9 +6,9 @@
 import org.hibernate.validator.constraints.Length;
 
 import com.tang.commons.annotation.poi.Excel;
-import com.tang.commons.annotation.poi.Excel.CellType;
-import com.tang.commons.annotation.poi.Excel.Type;
 import com.tang.commons.base.entity.BaseEntity;
+import com.tang.commons.enumeration.poi.CellType;
+import com.tang.commons.enumeration.poi.Type;
 
 import jakarta.validation.constraints.Email;
 import jakarta.validation.constraints.Max;