From c2a5872d4ab060e0e19a14be271a4b50d5e6059e Mon Sep 17 00:00:00 2001
From: riku <risaku@163.com>
Date: 星期一, 21 六月 2021 11:04:30 +0800
Subject: [PATCH] 1. 新增走航检测污染溯源模型逻辑

---
 src/main/kotlin/com/flightfeather/uav/lightshare/service/impl/RealTimeDataServiceImpl.kt |   57 +++++++++++++++++++++++++++++++++++++++++++++++++++++----
 1 files changed, 53 insertions(+), 4 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 182ac66..287c2e1 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
@@ -1,40 +1,89 @@
 package com.flightfeather.uav.lightshare.service.impl
 
+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.mapper.RealTimeDataMapper
 import com.flightfeather.uav.lightshare.bean.BaseResponse
 import com.flightfeather.uav.lightshare.bean.DataHead
+import com.flightfeather.uav.lightshare.bean.DataImportResult
 import com.flightfeather.uav.lightshare.bean.DataVo
 import com.flightfeather.uav.lightshare.service.RealTimeDataService
 import com.flightfeather.uav.socket.bean.AirData
 import com.github.pagehelper.PageHelper
 import org.springframework.stereotype.Service
+import org.springframework.web.multipart.MultipartFile
 import tk.mybatis.mapper.entity.Example
+import java.io.ByteArrayInputStream
+import java.io.FileInputStream
+import java.io.InputStream
+import java.text.DateFormat
 import java.text.SimpleDateFormat
 
 @Service
 class RealTimeDataServiceImpl(val realTimeDataMapper: RealTimeDataMapper) : RealTimeDataService {
 
-    override fun getSecondData(deviceCode: String?, page: Int?, perPage: Int?): BaseResponse<List<DataVo>> {
+    private var dateFormatter = SimpleDateFormat("yyyy-MM-dd HH:mm:ss")
+    private val fileExchange = FileExchange()
+
+    override fun getSecondData(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) }
         val pageInfo = PageHelper.startPage<RealTimeData>(_page, _perPage)
         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) }
             }
-            orderBy("dataTime").desc()
+            orderBy("dataTime").apply {
+                // 褰撹姹傛帴鍙d笉浼犻�掕捣濮嬫椂闂达紝榛樿鑾峰彇鏈�鏂扮殑鏁版嵁
+                if (startTime == null && endTime == null) {
+                    desc()
+                }
+            }
         }).forEach {
             result.add(DataVo(
-                    SimpleDateFormat.getDateTimeInstance().format(it.dataTime),
+                    dateFormatter.format(it.dataTime),
                     it.deviceCode,
                     GsonUtils.parserJsonToArrayBeans(it.factors, AirData::class.java),
                     it.longitude.toDouble(), it.latitude.toDouble()
             ))
         }
-        result.reverse()
+        if (startTime == null && endTime == null) {
+            result.reverse()
+        }
         return BaseResponse(true, head = DataHead(pageInfo.pageNum, pageInfo.pages), data = result)
     }
+
+    override fun getNextData(deviceCode: String, updateTime: String, page: Int?, perPage: Int?): BaseResponse<List<DataVo>> {
+        val _perPage = perPage ?: 60
+        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()
+            ))
+        }
+        return BaseResponse(true, head = DataHead(pageInfo.pageNum, pageInfo.pages), data = result)
+    }
+
+    override fun importData(file: MultipartFile): BaseResponse<DataImportResult> {
+        val f = ByteArrayInputStream(file.bytes)
+        fileExchange.exchangeBoatData("0c0000000001", f).forEach {
+            realTimeDataMapper.insert(it)
+        }
+        return BaseResponse(true, data = DataImportResult(""))
+    }
 }
\ No newline at end of file

--
Gitblit v1.9.3