feiyu02
2024-12-16 b3e273990fda27b68dfcf2c7f9d78311a5279885
src/test/kotlin/com/flightfeather/uav/lightshare/service/impl/RealTimeDataServiceImplTest.kt
@@ -1,26 +1,17 @@
package com.flightfeather.uav.lightshare.service.impl
import com.flightfeather.uav.common.utils.DateUtil
import com.flightfeather.uav.common.utils.ExcelUtil
import com.flightfeather.uav.dataprocess.DataProcess
import com.flightfeather.uav.domain.entity.RealTimeData
import com.flightfeather.uav.domain.entity.RealTimeDataVehicle
import com.flightfeather.uav.domain.mapper.RealTimeDataVehicleMapper
import com.flightfeather.uav.lightshare.service.RealTimeDataService
import com.github.pagehelper.PageHelper
import org.apache.poi.xssf.streaming.SXSSFWorkbook
import com.flightfeather.uav.repository.impl.AirDataRepositoryImpl
import org.junit.Test
import org.junit.Assert.*
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.io.File
import java.io.FileOutputStream
import java.util.*
import kotlin.math.atan
import kotlin.math.cos
import kotlin.math.sin
@RunWith(SpringRunner::class)
@SpringBootTest
@@ -28,6 +19,12 @@
    @Autowired
    lateinit var realTimeDataService: RealTimeDataService
    @Autowired
    lateinit var realTimeDataVehicleMapper: RealTimeDataVehicleMapper
    @Autowired
    lateinit var airDataRepository: AirDataRepositoryImpl
    @Test
    fun outToExcel() {
@@ -137,4 +134,31 @@
        val r = realTimeDataService.dataCalibration()
        println(r.data)
    }
    @Test
    fun dataTransform() {
        var page = 1
        var total = -1
        var count = 0
        while (total == -1 || page <= total) {
            println("------车载数据转换start------")
            val res = realTimeDataService.getOriginData("0a0000000001", "2021-01-01 00:00:00", "2021-10-25 15:37:21", page, 50000)
            res.head?.let {
                total = it.totalPage
            }
            if (page == 1) {
                println("总页数:$total")
            }
            println("当前页数:$page")
            res.data?.forEach {vo ->
                val d = RealTimeDataVehicle()
                airDataRepository.dataTransform(vo, d)
                realTimeDataVehicleMapper.insert(d)
                count++
            }
            page++
        }
    }
}