From e5bdf2e02090357cbd580d54e6cd2406dd541760 Mon Sep 17 00:00:00 2001
From: feiyu02 <risaku@163.com>
Date: 星期五, 21 三月 2025 17:39:30 +0800
Subject: [PATCH] 1. 新增卫星遥测网格热力图计算逻辑（待完成）
---
src/main/kotlin/com/flightfeather/uav/domain/repository/SatelliteGridRep.kt | 132 ++++++++++++++++++++++++++++++++++++++++++--
1 files changed, 126 insertions(+), 6 deletions(-)
diff --git a/src/main/kotlin/com/flightfeather/uav/domain/repository/SatelliteGridRep.kt b/src/main/kotlin/com/flightfeather/uav/domain/repository/SatelliteGridRep.kt
index 2c0db63..a5439b5 100644
--- a/src/main/kotlin/com/flightfeather/uav/domain/repository/SatelliteGridRep.kt
+++ b/src/main/kotlin/com/flightfeather/uav/domain/repository/SatelliteGridRep.kt
@@ -1,15 +1,15 @@
package com.flightfeather.uav.domain.repository
-import com.flightfeather.uav.domain.entity.GridCell
-import com.flightfeather.uav.domain.entity.GridData
-import com.flightfeather.uav.domain.entity.GridDataDetail
-import com.flightfeather.uav.domain.entity.GridGroup
+import com.flightfeather.uav.domain.entity.*
+import com.flightfeather.uav.domain.mapper.GridAodDetailMapper
+import com.flightfeather.uav.domain.mapper.GridAodMapper
import com.flightfeather.uav.domain.mapper.GridCellMapper
import com.flightfeather.uav.domain.mapper.GridDataDetailMapper
import com.flightfeather.uav.domain.mapper.GridDataMapper
import com.flightfeather.uav.domain.mapper.GridGroupMapper
import com.flightfeather.uav.lightshare.bean.AreaVo
import org.springframework.stereotype.Repository
+import org.springframework.transaction.annotation.Transactional
import tk.mybatis.mapper.entity.Example
import java.time.LocalDateTime
@@ -24,17 +24,50 @@
private val gridCellMapper: GridCellMapper,
private val gridDataMapper: GridDataMapper,
private val gridDataDetailMapper: GridDataDetailMapper,
+ private val gridAodMapper: GridAodMapper,
+ private val gridAodDetailMapper: GridAodDetailMapper,
) {
- fun fetchGridGroup(areaVo: AreaVo): List<GridGroup?> {
+ fun fetchGridGroup(areaVo: AreaVo, type: String?): List<GridGroup?> {
return gridGroupMapper.selectByExample(Example(GridGroup::class.java).apply {
createCriteria()
.andEqualTo("provinceCode", areaVo.provinceCode).andEqualTo("provinceName", areaVo.provinceName)
.andEqualTo("cityCode", areaVo.cityCode).andEqualTo("cityName", areaVo.cityName)
.andEqualTo("districtCode", areaVo.districtCode).andEqualTo("districtName", areaVo.districtName)
.andEqualTo("townCode", areaVo.townCode).andEqualTo("townName", areaVo.townName)
+ .andEqualTo("type", type)
})
}
+
+ fun fetchGridGroup(id: Int): GridGroup? {
+ return gridGroupMapper.selectByPrimaryKey(id)
+ }
+
+ fun fetchGridGroup(gridGroup: GridGroup): List<GridGroup?> {
+ return gridGroupMapper.select(gridGroup)
+ }
+
+ fun insertGridGroup(gridGroup: GridGroup): Int {
+ return gridGroupMapper.insert(gridGroup)
+ }
+
+ @Transactional
+ fun deleteGridGroup(groupId: Int) {
+ gridCellMapper.delete(GridCell().apply { this.groupId = groupId })
+ gridCellMapper.selectByExample(
+ Example(GridCell::class.java).apply { orderBy("id").desc() }
+ ).takeIf { it.isNotEmpty() }?.get(0)?.id?.let { id ->
+ gridCellMapper.resetAutoIncrement(id + 1)
+ }
+ gridGroupMapper.deleteByPrimaryKey(groupId)
+ gridGroupMapper.selectByExample(
+ Example(GridGroup::class.java).apply { orderBy("id").desc() }
+ ).takeIf { it.isNotEmpty() }?.get(0)?.id?.let { id ->
+ gridGroupMapper.resetAutoIncrement(id + 1)
+ }
+ }
+
+ /*****************************************************************/
fun fetchGridCell(groupId: Int): List<GridCell?> {
return gridCellMapper.selectByExample(Example(GridCell::class.java).apply {
@@ -42,6 +75,12 @@
orderBy("id")
})
}
+
+ fun insertGridCell(gridCellList: List<GridCell?>): Int {
+ return gridCellMapper.insertList(gridCellList)
+ }
+
+ /*****************************************************************/
fun fetchGridData(groupId: Int, dataTime: LocalDateTime?, type: Int?): List<GridData?> {
return gridDataMapper.selectByExample(Example(GridData::class.java).apply {
@@ -51,7 +90,31 @@
})
}
- fun fetchGridDataDetail(dataId: Int, groupId: Int?, cellId: Int?): List<GridDataDetail?> {
+ fun fetchGridData(gridData: GridData): List<GridData?> {
+ return gridDataMapper.select(gridData)
+ }
+
+ fun fetchGridData(id: Int): GridData? {
+ return gridDataMapper.selectByPrimaryKey(id)
+ }
+
+ fun insertGridData(gridData: GridData): Int {
+ return gridDataMapper.insert(gridData)
+ }
+
+ fun insertGridDataDetail(gridDataDetails: List<GridDataDetail?>): Int {
+ return gridDataDetailMapper.insertList(gridDataDetails)
+ }
+
+ fun updateGridDataDetail(gridDataDetails: List<GridDataDetail?>): Int {
+ var res = 0
+ gridDataDetails.forEach {
+ res += gridDataDetailMapper.updateByPrimaryKey(it)
+ }
+ return res
+ }
+
+ fun fetchGridDataDetail(dataId: Int?, groupId: Int?, cellId: Int?): List<GridDataDetail?> {
return gridDataDetailMapper.selectByExample(Example(GridDataDetail::class.java).apply {
createCriteria().andEqualTo("dataId", dataId)
.andEqualTo("groupId", groupId)
@@ -59,4 +122,61 @@
orderBy("cellId")
})
}
+
+ fun insertGridDataAndDetail(data: GridData, gridDataDetails: List<GridDataDetail>) {
+ gridDataMapper.insert(data)
+ gridDataDetails.forEach {
+ it.dataId = data.id
+ it.groupId = data.groupId
+ }
+ gridDataDetailMapper.insertList(gridDataDetails)
+ }
+
+ fun updatePM25Batch(gridDataDetails: List<GridDataDetail>) {
+ gridDataDetailMapper.updatePM25Batch(gridDataDetails)
+ }
+
+ /*****************************************************************/
+
+ // aod 鐩稿叧鎿嶄綔
+ fun fetchGridAod(groupId: Int, dataTime: LocalDateTime?): List<GridAod?> {
+ return gridAodMapper.selectByExample(Example(GridAod::class.java).apply {
+ createCriteria().andEqualTo("groupId", groupId)
+ .andEqualTo("dataTime", dataTime)
+ })
+ }
+
+ fun fetchGridAodDetail(aodId: Int, groupId: Int?, cellId: Int?): List<GridAodDetail?> {
+ return gridAodDetailMapper.selectByExample(Example(GridAodDetail::class.java).apply {
+ createCriteria().andEqualTo("aodId", aodId)
+ .andEqualTo("groupId", groupId)
+ .andEqualTo("cellId", cellId)
+ orderBy("cellId")
+ })
+ }
+
+ fun insertGridAodAndDetail(aod: GridAod, gridAodDetails: List<GridAodDetail>) {
+ gridAodMapper.insert(aod)
+ gridAodDetails.forEach {
+ it.aodId = aod.id
+ it.groupId = aod.groupId
+ }
+ gridAodDetailMapper.insertList(gridAodDetails)
+ }
+
+ fun updateGridAodBatch(gridDataDetails: List<GridAodDetail>) {
+ gridDataDetails.forEach {
+ gridAodDetailMapper.updateByExample(it, Example(GridAodDetail::class.java).apply {
+ createCriteria().andEqualTo("aodId", it.aodId)
+ .andEqualTo("groupId", it.groupId)
+ .andEqualTo("cellId", it.cellId)
+ })
+ }
+ }
+
+ @Transactional
+ fun updateGridCellBatch(gridCellList: List<GridCell?>) {
+ gridCellList.forEach { gridCellMapper.updateByPrimaryKey(it) }
+ }
+
}
\ No newline at end of file
--
Gitblit v1.9.3