package com.flightfeather.uav.biz.sourcetrace
|
|
import com.flightfeather.uav.common.utils.DateUtil
|
import com.flightfeather.uav.domain.entity.Mission
|
import com.flightfeather.uav.domain.entity.RealTimeDataVehicle
|
import com.flightfeather.uav.domain.mapper.MissionMapper
|
import com.flightfeather.uav.domain.repository.MissionRep
|
import com.flightfeather.uav.domain.repository.SceneInfoRep
|
import com.flightfeather.uav.domain.repository.SourceTraceRep
|
import com.flightfeather.uav.lightshare.service.RealTimeDataService
|
import org.junit.Test
|
import org.junit.runner.RunWith
|
import org.springframework.beans.factory.annotation.Autowired
|
import org.springframework.boot.test.context.SpringBootTest
|
import org.springframework.test.context.junit4.SpringRunner
|
import tk.mybatis.mapper.entity.Example
|
import java.time.LocalDateTime
|
import java.time.ZoneId
|
|
@RunWith(SpringRunner::class)
|
@SpringBootTest
|
class SourceTraceControllerTest {
|
|
@Autowired
|
lateinit var sceneInfoRep: SceneInfoRep
|
|
@Autowired
|
lateinit var sourceTraceRep: SourceTraceRep
|
|
@Autowired
|
lateinit var missionMapper: MissionMapper
|
|
@Autowired
|
lateinit var missionRep: MissionRep
|
|
@Autowired
|
lateinit var realTimeDataService: RealTimeDataService
|
|
@Test
|
fun autoSourceTrace() {
|
val sourceTraceController = SourceTraceController(sceneInfoRep, sourceTraceRep, true)
|
val startTime = LocalDateTime.of(2025, 12, 11, 0, 0, 0).atZone(ZoneId.systemDefault()).toInstant()
|
val endTime = LocalDateTime.of(2025, 12, 11, 23, 59, 59).atZone(ZoneId.systemDefault()).toInstant()
|
val missions = missionMapper.selectByExample(Example(Mission::class.java).apply {
|
createCriteria().andBetween("startTime", startTime, endTime)
|
})
|
missions.forEach { m ->
|
val rtData = realTimeDataService.getSecondData(
|
m?.deviceType,
|
m?.deviceCode,
|
DateUtil.instance.dateToString(m?.startTime, DateUtil.DateStyle.YYYY_MM_DD_HH_MM_SS),
|
DateUtil.instance.dateToString(m?.endTime, DateUtil.DateStyle.YYYY_MM_DD_HH_MM_SS),
|
null,
|
1,
|
10000
|
)
|
rtData.data?.forEach { d ->
|
val rtdVehicle = d.toBaseRealTimeData(RealTimeDataVehicle::class.java)
|
// Thread.sleep(500)
|
sourceTraceController.addOneData(rtdVehicle)
|
}
|
sourceTraceController.initTask()
|
}
|
}
|
|
@Test
|
fun deleteSourceTrace() {
|
val startTime = LocalDateTime.of(2025, 12, 11, 0, 0, 0).atZone(ZoneId.systemDefault()).toInstant()
|
val endTime = LocalDateTime.of(2025, 12, 11, 23, 59, 59).atZone(ZoneId.systemDefault()).toInstant()
|
val missions = missionMapper.selectByExample(Example(Mission::class.java).apply {
|
createCriteria().andBetween("startTime", startTime, endTime)
|
})
|
missions.forEach { m ->
|
m ?: return@forEach
|
sourceTraceRep.delete(m)
|
}
|
}
|
|
}
|
