From 665a2e1098fb52800ac7624d4a32dfeb6ce15151 Mon Sep 17 00:00:00 2001 From: riku <risaku@163.com> Date: 星期三, 10 十一月 2021 11:03:36 +0800 Subject: [PATCH] 1. 网格化均值计算逻辑完成 --- src/main/kotlin/com/flightfeather/uav/lightshare/service/impl/RealTimeDataServiceImpl.kt | 230 ++++++++++++++++++++++++++++++++++++++++++++++++--------- 1 files changed, 193 insertions(+), 37 deletions(-) diff --git a/src/main/kotlin/com/flightfeather/uav/lightshare/service/impl/RealTimeDataServiceImpl.kt b/src/main/kotlin/com/flightfeather/uav/lightshare/service/impl/RealTimeDataServiceImpl.kt index 75a581e..0336692 100644 --- a/src/main/kotlin/com/flightfeather/uav/lightshare/service/impl/RealTimeDataServiceImpl.kt +++ b/src/main/kotlin/com/flightfeather/uav/lightshare/service/impl/RealTimeDataServiceImpl.kt @@ -4,13 +4,15 @@ import com.flightfeather.uav.common.utils.ExcelUtil import com.flightfeather.uav.common.utils.FileExchange import com.flightfeather.uav.common.utils.GsonUtils -import com.flightfeather.uav.domain.entity.RealTimeData -import com.flightfeather.uav.domain.entity.toRowContent -import com.flightfeather.uav.domain.entity.toRowTitle -import com.flightfeather.uav.domain.mapper.RealTimeDataMapper +import com.flightfeather.uav.dataprocess.AverageUtil +import com.flightfeather.uav.domain.entity.* +import com.flightfeather.uav.domain.mapper.* import com.flightfeather.uav.lightshare.bean.* import com.flightfeather.uav.lightshare.service.RealTimeDataService +import com.flightfeather.uav.model.epw.EPWDataPrep +import com.flightfeather.uav.repository.AirDataRepository import com.flightfeather.uav.socket.bean.AirData +import com.flightfeather.uav.socket.eunm.UWDeviceType import com.github.pagehelper.PageHelper import org.apache.poi.xssf.streaming.SXSSFWorkbook import org.springframework.stereotype.Service @@ -18,13 +20,24 @@ import tk.mybatis.mapper.entity.Example import java.io.ByteArrayInputStream import java.text.SimpleDateFormat +import java.time.LocalDateTime +import java.time.format.DateTimeFormatter import java.util.* import javax.servlet.http.HttpServletResponse +import javax.swing.text.DateFormatter @Service -class RealTimeDataServiceImpl(val realTimeDataMapper: RealTimeDataMapper) : RealTimeDataService { +class RealTimeDataServiceImpl( + private val realTimeDataMapper: RealTimeDataMapper, + private val airDataRepository: AirDataRepository, + private val realTimeDataVehicleMapper: RealTimeDataVehicleMapper, + private val realTimeDataUavMapper: RealTimeDataUavMapper, + private val realTimeDataGridMapper: RealTimeDataGridMapper, + private val realTimeDataGridMinMapper: RealTimeDataGridMinMapper +) : RealTimeDataService { private var dateFormatter = SimpleDateFormat("yyyy-MM-dd HH:mm:ss") + private var dateFormatter2 = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm") private val fileExchange = FileExchange() override fun getSecondData(deviceCode: String?, startTime: String?, endTime: String?, page: Int?, perPage: Int?): BaseResponse<List<DataVo>> { @@ -32,35 +45,70 @@ val _page = page ?: 1 val sTime = startTime?.let { dateFormatter.parse(it) } val eTime = endTime?.let { dateFormatter.parse(it) } - val pageInfo = PageHelper.startPage<RealTimeData>(_page, _perPage) +// var pageInfo = PageHelper.startPage<BaseRealTimeData>(_page, _perPage) + var pageNum = 1 + var pages = 0 val result = mutableListOf<DataVo>() - realTimeDataMapper.selectByExample(Example(RealTimeData::class.java).apply { - createCriteria().apply { - deviceCode?.let { andEqualTo("deviceCode", it) } - sTime?.let { andGreaterThanOrEqualTo("dataTime", it) } - eTime?.let { andLessThanOrEqualTo("dataTime", it) } + + when (UWDeviceType.getType(deviceCode)) { + UWDeviceType.VEHICLE -> { + val pageInfo = PageHelper.startPage<RealTimeDataVehicle>(_page, _perPage) + realTimeDataVehicleMapper.selectByExample(Example(RealTimeDataVehicle::class.java).apply { + getSecondDataExample(this, deviceCode, sTime, eTime) + }).forEach { result.add(it.toDataVo()) } + pageNum = pageInfo.pageNum + pages = pageInfo.pages } - orderBy("dataTime").apply { - // 褰撹姹傛帴鍙d笉浼犻�掕捣濮嬫椂闂达紝榛樿鑾峰彇鏈�鏂扮殑鏁版嵁 - if (startTime == null && endTime == null) { - desc() + UWDeviceType.UAV -> { + val pageInfo = PageHelper.startPage<RealTimeDataUav>(_page, _perPage) + realTimeDataUavMapper.selectByExample(Example(RealTimeDataUav::class.java).apply { + getSecondDataExample(this, deviceCode, sTime, eTime) + }).forEach { result.add(it.toDataVo()) } + pageNum = pageInfo.pageNum + pages = pageInfo.pages + } + UWDeviceType.GRID -> { + val pageInfo = PageHelper.startPage<RealTimeDataGrid>(_page, _perPage) + realTimeDataGridMapper.selectByExample(Example(RealTimeDataGrid::class.java).apply { + getSecondDataExample(this, deviceCode, sTime, eTime) + }).forEach { result.add(it.toDataVo()) } + pageNum = pageInfo.pageNum + pages = pageInfo.pages + } + else -> { + // 浠庡師濮嬫暟鎹〃涓幏鍙栨暟鎹� + val pageInfo = PageHelper.startPage<RealTimeData>(_page, _perPage) + realTimeDataMapper.selectByExample(Example(RealTimeData::class.java).apply { + getSecondDataExample(this, deviceCode, sTime, eTime) + }).forEach { + if (it.longitude == null || it.latitude == null) { + return@forEach + } + result.add(it.toDataVo()) } + pageNum = pageInfo.pageNum + pages = pageInfo.pages } - }).forEach { - if (it.longitude == null || it.latitude == null) { - return@forEach - } - result.add(DataVo( - dateFormatter.format(it.dataTime), - it.deviceCode, - GsonUtils.parserJsonToArrayBeans(it.factors, AirData::class.java), - it.longitude?.toDouble(), it.latitude?.toDouble() - )) } + if (startTime == null && endTime == null) { result.reverse() } - return BaseResponse(true, head = DataHead(pageInfo.pageNum, pageInfo.pages), data = result) + return BaseResponse(true, head = DataHead(pageNum, pages), data = result) + } + + private fun getSecondDataExample(example: Example, deviceCode: String?, sTime: Date?, eTime: Date?) { + example.createCriteria().apply { + deviceCode?.let { andEqualTo("deviceCode", it) } + sTime?.let { andGreaterThanOrEqualTo("dataTime", it) } + eTime?.let { andLessThanOrEqualTo("dataTime", it) } + } + example.orderBy("dataTime").apply { + // 褰撹姹傛帴鍙d笉浼犻�掕捣濮嬫椂闂达紝榛樿鑾峰彇鏈�鏂扮殑鏁版嵁 + if (sTime == null && eTime == null) { + desc() + } + } } override fun getNextData(deviceCode: String, updateTime: String, page: Int?, perPage: Int?): BaseResponse<List<DataVo>> { @@ -68,19 +116,36 @@ val _page = page ?: 1 val pageInfo = PageHelper.startPage<RealTimeData>(_page, _perPage) val result = mutableListOf<DataVo>() - realTimeDataMapper.selectByExample(Example(RealTimeData::class.java).apply { - createCriteria().andEqualTo("deviceCode", deviceCode) - .andGreaterThan("dataTime", updateTime) - orderBy("dataTime") - }).forEach { - result.add(DataVo( - dateFormatter.format(it.dataTime), - it.deviceCode, - GsonUtils.parserJsonToArrayBeans(it.factors, AirData::class.java), - it.longitude.toDouble(), it.latitude.toDouble() - )) + + when (UWDeviceType.getType(deviceCode)) { + UWDeviceType.VEHICLE -> { + realTimeDataVehicleMapper.selectByExample(Example(RealTimeDataVehicle::class.java).apply { + getNextDataExample(this, deviceCode, updateTime) + }).forEach { result.add(it.toDataVo()) } + } + UWDeviceType.UAV -> { + realTimeDataUavMapper.selectByExample(Example(RealTimeDataUav::class.java).apply { + getNextDataExample(this, deviceCode, updateTime) + }).forEach { result.add(it.toDataVo()) } + } + UWDeviceType.GRID -> { + realTimeDataGridMapper.selectByExample(Example(RealTimeDataGrid::class.java).apply { + getNextDataExample(this, deviceCode, updateTime) + }).forEach { result.add(it.toDataVo()) } + } + else -> { + realTimeDataMapper.selectByExample(Example(RealTimeData::class.java).apply { + getNextDataExample(this, deviceCode, updateTime) + }).forEach { result.add(it.toDataVo()) } + } } return BaseResponse(true, head = DataHead(pageInfo.pageNum, pageInfo.pages), data = result) + } + + private fun getNextDataExample(example: Example, deviceCode: String, updateTime: String) { + example.createCriteria().andEqualTo("deviceCode", deviceCode) + .andGreaterThan("dataTime", updateTime) + example.orderBy("dataTime") } override fun importData(file: MultipartFile): BaseResponse<DataImportResult> { @@ -149,4 +214,95 @@ } return contents } + + private fun getOriginData(deviceCode: String?, startTime: String?, endTime: String?, page: Int?, perPage: Int?): BaseResponse<List<DataVo>> { + val _perPage = perPage ?: 60 + val _page = page ?: 1 + val sTime = startTime?.let { dateFormatter.parse(it) } + val eTime = endTime?.let { dateFormatter.parse(it) } +// var pageInfo = PageHelper.startPage<BaseRealTimeData>(_page, _perPage) + var pageNum = 1 + var pages = 0 + val result = mutableListOf<DataVo>() + + // 浠庡師濮嬫暟鎹〃涓幏鍙栨暟鎹� + val pageInfo = PageHelper.startPage<RealTimeData>(_page, _perPage) + realTimeDataMapper.selectByExample(Example(RealTimeData::class.java).apply { + getSecondDataExample(this, deviceCode, sTime, eTime) + }).forEach { + if (it.longitude == null || it.latitude == null) { + return@forEach + } + result.add(it.toDataVo()) + } + pageNum = pageInfo.pageNum + pages = pageInfo.pages + + if (startTime == null && endTime == null) { + result.reverse() + } + return BaseResponse(true, head = DataHead(pageNum, pages), data = result) + } + + override fun dataPreprocessing(): BaseResponse<String> { + val epwDataPrep = EPWDataPrep() + var page = 1 + var total = -1 + var count = 0 + + while (total == -1 || page <= total) { + println("------鏁版嵁棰勫鐞唖tart------") + val res = getOriginData("0d0000000001", "2021-07-05 19:47:01", "2021-11-05 00:00:00", page, 50000) + res.head?.let { + total = it.totalPage + } + if (page == 1) { + println("鎬婚〉鏁帮細$total") + } + println("褰撳墠椤垫暟锛�$page") + val dataList = res.data ?: emptyList() + val result = epwDataPrep.mDataPrep2(dataList) + count += airDataRepository.savePrepData2(result) + page++ + } + + return BaseResponse(count > 0, data = "鎻掑叆鏁版嵁: ${count}鏉�") + } + + override fun averageData(): BaseResponse<String> { + var page = 1 + var total = -1 + var count = 0 + + val minFormatter = SimpleDateFormat("yyyy-MM-dd HH:mm") + val averageUtil = AverageUtil<RealTimeDataGrid, RealTimeDataGridMin>({d -> + minFormatter.format(d.dataTime) + },{list -> + list.avg() + }) + + while (total == -1 || page <= total) { + println("------鍧囧�艰绠梥tart------") + + val p = PageHelper.startPage<RealTimeDataGrid>(page, 50000) + val res = realTimeDataGridMapper.selectByExample(Example(RealTimeDataGrid::class.java).apply { + createCriteria().andBetween("dataTime", "2021-06-01 00:00:00", "2021-11-05 00:00:00") + }) + + total = p.pages + + if (page == 1) { + println("鎬婚〉鏁帮細$total") + } + println("褰撳墠椤垫暟锛�$page") + averageUtil.avg(res).forEach { + realTimeDataGridMinMapper.insert(it) + count++ + } + + page++ + } + + return BaseResponse(count > 0, data = "鎻掑叆鏁版嵁: ${count}鏉�") + } } \ No newline at end of file -- Gitblit v1.9.3