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