2026.4.7

feiyu02

9 天以前 85ef942e7195abeb71466b7159c3ee30161e1e54

 src/main/kotlin/cn/flightfeather/supervision/business/import/SceneImport.kt

@@ -3,8 +3,11 @@
import cn.flightfeather.supervision.common.exception.BizException
import cn.flightfeather.supervision.common.utils.Constant
import cn.flightfeather.supervision.common.utils.ExcelUtil
import cn.flightfeather.supervision.domain.ds1.entity.Scense
import cn.flightfeather.supervision.domain.ds1.entity.*
import cn.flightfeather.supervision.domain.ds1.repository.RegionRep
import cn.flightfeather.supervision.lightshare.vo.SceneDetail
import com.google.gson.Gson
import org.apache.poi.ss.usermodel.Cell
import org.apache.poi.ss.usermodel.CellType
import org.springframework.stereotype.Component
import java.io.InputStream
@@ -17,9 +20,10 @@
class SceneImport(
    private val regionRep: RegionRep,
) {
    // 场景共有信息
    private val headers =
        listOf(
            ExcelUtil.MyHeader("编号（若不填写，则按照顺序自动添加）", CellType.NUMERIC),
            ExcelUtil.MyHeader("编号（可省略）", CellType.NUMERIC, true),
            ExcelUtil.MyHeader("名称", CellType.STRING),
            ExcelUtil.MyHeader("类型", CellType.STRING),
            ExcelUtil.MyHeader("地址", CellType.STRING),
@@ -33,56 +37,116 @@
            ExcelUtil.MyHeader("联系电话", CellType.STRING)
        )

    // 工地信息
    private val headersType1 = listOf(
        ExcelUtil.MyHeader("工程类型", CellType.STRING),
        ExcelUtil.MyHeader("建设状态", CellType.STRING),
        ExcelUtil.MyHeader("施工阶段", CellType.STRING),
        ExcelUtil.MyHeader("开工时间", CellType.STRING),
        ExcelUtil.MyHeader("完工时间", CellType.STRING),
        ExcelUtil.MyHeader("剩余工期", CellType.STRING),
        ExcelUtil.MyHeader("占地面积", CellType.NUMERIC),
        ExcelUtil.MyHeader("施工面积", CellType.NUMERIC),
        ExcelUtil.MyHeader("每月施工面积", CellType.NUMERIC),
        ExcelUtil.MyHeader("业主单位", CellType.STRING),
        ExcelUtil.MyHeader("业主单位联系人", CellType.STRING),
        ExcelUtil.MyHeader("业主单位联系电话", CellType.STRING),
        ExcelUtil.MyHeader("施工单位", CellType.STRING),
        ExcelUtil.MyHeader("施工单位联系人", CellType.STRING),
        ExcelUtil.MyHeader("施工单位联系电话", CellType.STRING),
        ExcelUtil.MyHeader("安全员", CellType.STRING),
        ExcelUtil.MyHeader("安全员电话", CellType.STRING),
    )

    /**
     * 通过文件读取场景信息
     */
    fun readFromFile(file: InputStream): List<Scense> {
        val result = mutableListOf<Scense>()
    fun readFromFile(file: InputStream): List<SceneDetail> {
        val result = mutableListOf<SceneDetail>()
        try {
            ExcelUtil.readXLXS(file, headers) {
                val data = Scense().apply {
                    guid = null
                    name = it.getCell(1)?.stringCellValue?.trim()
                    typeid = null
                    type = it.getCell(2)?.stringCellValue?.trim()
                    scensesubtypeid = null
                    scensesubtype = null
                    location = it.getCell(3)?.stringCellValue?.trim()
                    longitude = it.getCell(4)?.numericCellValue?.toBigDecimal()
                    latitude = it.getCell(5)?.numericCellValue?.toBigDecimal()
                    provincecode = null
                    provincename = it.getCell(6)?.stringCellValue?.trim()
                    citycode = null
                    cityname = it.getCell(7)?.stringCellValue?.trim()
                    districtcode = null
                    districtname = it.getCell(8)?.stringCellValue?.trim()
                    towncode = null
                    townname = it.getCell(9)?.stringCellValue?.trim()
                    contactst = it.getCell(11)?.stringCellValue?.trim()
                    contactswx = null
                    contacts = it.getCell(10)?.stringCellValue?.trim()
                    createdate = Date()
                    updatedate = Date()
                    extension1 = "1"
                    extension2 = null
                    index = it.getCell(5)?.stringCellValue?.trim()?.toInt()
                    remark = null
                val data = SceneDetail().apply {
                    scense = Scense().apply {
                        guid = null
                        name = it.getCell(1)?.stringCellValue?.trim()
                        typeid = null
                        type = it.getCell(2)?.stringCellValue?.trim()
                        scensesubtypeid = null
                        scensesubtype = null
                        location = it.getCell(3)?.stringCellValue?.trim()
                        longitude = it.getCell(4)?.numericCellValue?.toBigDecimal()
                        latitude = it.getCell(5)?.numericCellValue?.toBigDecimal()
                        provincecode = null
                        provincename = it.getCell(6)?.stringCellValue?.trim()
                        citycode = null
                        cityname = it.getCell(7)?.stringCellValue?.trim()
                        districtcode = null
                        districtname = it.getCell(8)?.stringCellValue?.trim()
                        towncode = null
                        townname = it.getCell(9)?.stringCellValue?.trim()
                        contactst = it.getCell(11)?.stringCellValue?.trim()
                        contactswx = null
                        contacts = it.getCell(10)?.stringCellValue?.trim()
                        createdate = Date()
                        updatedate = Date()
                        extension1 = "1"
                        extension2 = null
                        index = it.getCell(0)?.numericCellValue?.toInt()
                        remark = null
                    }
                    if (scense?.name.isNullOrBlank()) throw BizException(errorStr(it.rowNum + 1, 2, "场景名称不能空白"))
                    scense?.typeid = Constant.SceneType.getByName(scense?.type)?.value?.toByte()
                        ?: throw BizException(errorStr(it.rowNum + 1, 3, "场景类型不存在"))
                    if (scense?.location.isNullOrBlank()) throw BizException(errorStr(it.rowNum + 1, 4, "场景地址不能空白"))
                    scense?.longitude ?: throw BizException(errorStr(it.rowNum + 1, 5, "经度不能空白"))
                    scense?.latitude ?: throw BizException(errorStr(it.rowNum + 1, 6, "纬度不能空白"))
                    scense?.provincecode = regionRep.findProvince(scense?.provincename)?.provincecode
                        ?: throw BizException(errorStr(it.rowNum + 1, 7, "省份错误或系统未曾配置"))
                    scense?.citycode = regionRep.findCity(scense?.cityname)?.citycode
                        ?: throw BizException(errorStr(it.rowNum + 1, 8, "城市错误或系统未曾配置"))
                    scense?.districtcode = regionRep.findDistrict(scense?.districtname)?.districtcode
                        ?: throw BizException(errorStr(it.rowNum + 1, 9, "区县错误或系统未曾配置"))
                    scense?.towncode = regionRep.findTown(scense?.townname)?.towncode
                        ?: throw BizException(errorStr(it.rowNum + 1, 10, "街镇错误或系统未曾配置"))

                    when (scense?.typeid.toString()) {
                        Constant.SceneType.TYPE1.value -> {
                            subScene = SceneConstructionSite().apply {
                                csProjectType = getCellValueToString(it.getCell(12))
                                csStatus = getCellValueToString(it.getCell(13))
                                siExtension1 = getCellValueToString(it.getCell(14))
                                csStartTime = getCellValueToString(it.getCell(15))
                                csEndTime = getCellValueToString(it.getCell(16))
                                csLeftTime = getCellValueToString(it.getCell(17))
                                csFloorSpace = getCellValueToString(it.getCell(18))
                                csConstructionArea = getCellValueToString(it.getCell(19))
                                csConstructionAreaPerMonth = getCellValueToString(it.getCell(20))
                                csEmployerUnit = getCellValueToString(it.getCell(21))
                                csEmployerContacts = getCellValueToString(it.getCell(22))
                                csEmployerContactsTel = getCellValueToString(it.getCell(23))
                                csConstructionUnit = getCellValueToString(it.getCell(24))
                                csConstructionContacts = getCellValueToString(it.getCell(25))
                                csConstructionContactsTel = getCellValueToString(it.getCell(26))
                                csSecurityOfficer = getCellValueToString(it.getCell(27))
                                csSecurityOfficerTel = getCellValueToString(it.getCell(28))
                            }
                        }
                        Constant.SceneType.TYPE2.value -> {

                        }
                        Constant.SceneType.TYPE3.value -> {

                        }
                        Constant.SceneType.TYPE14.value -> {

                        }
                    }
                }
                data.typeid = Constant.SceneType.getByName(data.type)?.value?.toByte()
                    ?: throw BizException(errorStr(it.rowNum + 1, 2, "场景类型不存在"))
                data.provincecode = regionRep.findProvince(data.provincename)?.provincecode
                    ?: throw BizException(errorStr(it.rowNum + 1, 6, "省份错误或系统未曾配置"))
                data.citycode = regionRep.findCity(data.cityname)?.citycode
                    ?: throw BizException(errorStr(it.rowNum + 1, 7, "城市错误或系统未曾配置"))
                data.districtcode = regionRep.findDistrict(data.districtname)?.districtcode
                    ?: throw BizException(errorStr(it.rowNum + 1, 8, "区县错误或系统未曾配置"))
                data.towncode = regionRep.findTown(data.townname)?.towncode
                    ?: throw BizException(errorStr(it.rowNum + 1, 9, "街镇错误或系统未曾配置"))

                result.add(data)
            }
        } catch (e: IllegalStateException) {
            throw BizException("excel文件内容错误，数据转换失败！", e)
            throw BizException(e.message, e)
        }
        return result
    }
@@ -90,4 +154,16 @@
    private fun errorStr(row: Int, col: Int, str: String): String {
        return "第${row}行，第${col}列单元格错误，${str}"
    }

    private fun getCellValueToString(cell: Cell?): String {
        cell ?: return ""
        return when (cell.cellType) {
            CellType.STRING -> cell.stringCellValue.trim()
            CellType.NUMERIC -> cell.numericCellValue.toString().trim()
            CellType.BLANK -> cell.stringCellValue.trim()
            CellType.BOOLEAN -> cell.booleanCellValue.toString().trim()
            CellType.ERROR -> cell.errorCellValue.toString().trim()
            else -> throw BizException(errorStr(cell.rowIndex + 1, cell.columnIndex + 1, "单元格格式不是常见格式，无法识别"))
        }
    }
}