underway-spring.git

			@@ -1,8 +1,16 @@
			package com.flightfeather.uav.common.utils

			import com.flightfeather.uav.common.exception.BizException
			import org.apache.poi.hssf.usermodel.HSSFRow
			import org.apache.poi.hssf.usermodel.HSSFWorkbook
			import org.apache.poi.ss.usermodel.Row
			import org.apache.poi.ss.util.CellRangeAddress
			import org.apache.poi.xssf.streaming.SXSSFWorkbook
			import org.apache.poi.xssf.usermodel.XSSFRow
			import org.apache.poi.xssf.usermodel.XSSFWorkbook
			import java.io.File
			import java.io.FileInputStream
			import java.io.InputStream
			import java.time.LocalDate
			import java.util.*
			import kotlin.math.max
			@@ -16,16 +24,64 @@
			private const val isLog = false

			class MyCell(
			var text: String,
			var rowSpan: Int = 1,
			var colSpan: Int = 1,
			var fontColor: Short? = null
			var text: String,
			var rowSpan: Int = 1,
			var colSpan: Int = 1,
			var fontColor: Short? = null,
			)

			/**
			* 读取文件
			* @param file 要读取的文件
			* @param headerCheck 文件首行检查回调
			* @param onRow 单行内容回调，从第二行开始
			*/
			fun readXLXS(file: File, headerCheck: (header: XSSFRow) -> Boolean, onRow: (row: Row) -> Unit): Boolean {
			if (!file.exists() \|\| !file.isFile) throw BizException("it's not a normal file!")
			if (!file.extension.equals("xls", ignoreCase = true) \|\| !file.extension.equals("xlsx", ignoreCase = true)) {
			throw BizException("file's extension name should be xls or xlsx!")
			}
			return readXLXS(FileInputStream(file), headerCheck, onRow)
			}

			/**
			* 包括 .xlsx 文件
			*/
			fun readXLXS(input: InputStream, headerCheck: (header: XSSFRow) -> Boolean, onRow: (row: Row) -> Unit): Boolean {
			val workbook = XSSFWorkbook(input)
			val sheet1 = workbook.getSheetAt(0)
			val header = sheet1.getRow(sheet1.topRow.toInt())
			return if (headerCheck(header)) {
			// 获取迭代器并去除第一行标题
			val iterator = sheet1.rowIterator().also { it.next() }
			iterator.forEach { onRow(it) }
			true
			} else {
			false
			}
			}

			/**
			* 包括 .xls .csv 文件
			*/
			fun readXLS(input: InputStream, headerCheck: (header: HSSFRow) -> Boolean, onRow: (row: Row) -> Unit): Boolean {
			val workbook = HSSFWorkbook(input)
			val sheet1 = workbook.getSheetAt(0)
			val header = sheet1.getRow(sheet1.topRow.toInt())
			return if (headerCheck(header)) {
			// 获取迭代器并去除第一行标题
			val iterator = sheet1.rowIterator().also { it.next() }
			iterator.forEach { onRow(it) }
			true
			} else {
			false
			}
			}

			/**
			* 自动处理行合并数据
			*/
			fun write2(heads: List<String>, contents: List<Array<Any>>, workbook: SXSSFWorkbook, sheetName:String) {
			fun write2(heads: List<String>, contents: List<Array<Any>>, workbook: SXSSFWorkbook, sheetName: String) {

			val sheet = workbook.createSheet(sheetName)

			@@ -52,31 +108,31 @@
			var rowspan = maxRow//合并的行的跨度

			val c =
			if (cell is Array<*>) {
			//当数据为数组时，需要根据最大行数重新计算该单元格的行跨度
			arrayMap[i] = cell
			rowspan = maxRow / if (cell.size == 0) 1 else cell.size
			val c = cell[0]
			if (c is MyCell) {
			rowspan = c.rowSpan
			}
			if (rowspan > 1) {
			log("合并1-1：$rowIndex;${rowIndex + rowspan - 1};$i")
			sheet.addMergedRegion(CellRangeAddress(rowIndex, rowIndex + rowspan - 1, i, i))
			}
			if (cell.isEmpty()) {
			""
			} else {
			cell[0]
			}
			} else {
			//当数据不是数组时，需要按最大行数合并单元格
			if (rowspan > 1) {
			log("合并1-2：$rowIndex;${rowIndex + rowspan - 1};$i")
			sheet.addMergedRegion(CellRangeAddress(rowIndex, rowIndex + rowspan - 1, i, i))
			}
			cell
			if (cell is Array<*>) {
			//当数据为数组时，需要根据最大行数重新计算该单元格的行跨度
			arrayMap[i] = cell
			rowspan = maxRow / if (cell.size == 0) 1 else cell.size
			val c = cell[0]
			if (c is MyCell) {
			rowspan = c.rowSpan
			}
			if (rowspan > 1) {
			log("合并1-1：$rowIndex;${rowIndex + rowspan - 1};$i")
			sheet.addMergedRegion(CellRangeAddress(rowIndex, rowIndex + rowspan - 1, i, i))
			}
			if (cell.isEmpty()) {
			""
			} else {
			cell[0]
			}
			} else {
			//当数据不是数组时，需要按最大行数合并单元格
			if (rowspan > 1) {
			log("合并1-2：$rowIndex;${rowIndex + rowspan - 1};$i")
			sheet.addMergedRegion(CellRangeAddress(rowIndex, rowIndex + rowspan - 1, i, i))
			}
			cell
			}
			when (c) {
			is MyCell -> {
			rows.createCell(i).setCellValue(c.text)
			@@ -107,7 +163,7 @@
			if (c is MyCell) {
			rowspan = c.rowSpan
			}
			val _c = array[i-1]
			val _c = array[i - 1]
			if (_c is MyCell) {
			lastRowSpan = _c.rowSpan
			}
			@@ -153,9 +209,15 @@
			/**
			* 自动处理行合并数据
			*/
			fun write(heads: List<Array<String>>, contents: List<Array<Any>>, workbook: SXSSFWorkbook, sheetName: String = "sheet1", row: Int = 0): Int {
			fun write(
			heads: List<Array<String>>,
			contents: List<Array<Any>>,
			workbook: SXSSFWorkbook,
			sheetName: String = "sheet1",
			row: Int = 0,
			): Int {

			val sheet = workbook.getSheet(sheetName)?: workbook.createSheet(sheetName)
			val sheet = workbook.getSheet(sheetName) ?: workbook.createSheet(sheetName)
			// println("sheet: $sheetName")

			var rowIndex = row
			@@ -195,7 +257,10 @@
			}
			if (rowspan > 1 \|\| colSpan > 1) {
			log("合并1-1：$rowIndex;${rowIndex + rowspan - 1};$col")
			sheet.addMergedRegion(CellRangeAddress(rowIndex, rowIndex + rowspan - 1, col, col + colSpan - 1))
			sheet.addMergedRegion(CellRangeAddress(rowIndex,
			rowIndex + rowspan - 1,
			col,
			col + colSpan - 1))
			}
			if (cell.isEmpty()) {
			""
			@@ -210,7 +275,10 @@
			}
			if (rowspan > 1 \|\| colSpan > 1) {
			log("合并1-2：$rowIndex;${rowIndex + rowspan - 1};$col")
			sheet.addMergedRegion(CellRangeAddress(rowIndex, rowIndex + rowspan - 1, col, col + colSpan - 1))
			sheet.addMergedRegion(CellRangeAddress(rowIndex,
			rowIndex + rowspan - 1,
			col,
			col + colSpan - 1))
			}
			cell
			}
			@@ -272,7 +340,10 @@
			}
			if (rowspan > 1 \|\| colSpan > 1) {
			log("合并2：${index};${index + rowspan - 1};${map.key}")
			sheet.addMergedRegion(CellRangeAddress(index, index + rowspan - 1, map.key, map.key + colSpan - 1))
			sheet.addMergedRegion(CellRangeAddress(index,
			index + rowspan - 1,
			map.key,
			map.key + colSpan - 1))
			}

			when (c) {