package cn.flightfeather.supervision.domain.ds1.repository
|
|
import cn.flightfeather.supervision.domain.ds1.entity.Task
|
import cn.flightfeather.supervision.domain.ds1.mapper.TaskMapper
|
import cn.flightfeather.supervision.lightshare.vo.AreaVo
|
import org.springframework.stereotype.Repository
|
import java.time.LocalDate
|
import java.time.ZoneId
|
import java.util.*
|
|
@Repository
|
class TaskRep(private val taskMapper: TaskMapper) {
|
|
private fun exampleTask(areaVo: AreaVo): Task?{
|
areaVo.starttime ?: return null
|
val mStart = LocalDate.parse(areaVo.starttime).withDayOfMonth(1).atStartOfDay()
|
val mEnd = mStart.plusMonths(1).minusSeconds(1)
|
return Task().apply {
|
provincecode = areaVo.provincecode
|
citycode = areaVo.citycode
|
districtcode = areaVo.districtcode
|
starttime = Date.from(mStart.atZone(ZoneId.systemDefault()).toInstant())
|
endtime = Date.from(mEnd.atZone(ZoneId.systemDefault()).toInstant())
|
}
|
}
|
/**
|
* 查找一个总任务
|
*/
|
fun findOneTask(areaVo: AreaVo): Task? {
|
val example = exampleTask(areaVo) ?: return null
|
return taskMapper.selectOne(example)
|
}
|
|
fun findTasks(areaVo: AreaVo): List<Task?> {
|
val example = exampleTask(areaVo) ?: return emptyList()
|
return taskMapper.select(example)
|
}
|
}
|