package cn.flightfeather.supervision.domain.ds1.repository
|
|
import cn.flightfeather.supervision.common.exception.BizException
|
import cn.flightfeather.supervision.common.utils.UUIDGenerator
|
import cn.flightfeather.supervision.domain.ds1.entity.Scense
|
import cn.flightfeather.supervision.domain.ds1.mapper.ScenseMapper
|
import cn.flightfeather.supervision.domain.ds1.mapper.UserinfoMapper
|
import org.springframework.stereotype.Repository
|
import tk.mybatis.mapper.entity.Example
|
import java.util.*
|
|
/**
|
* 场景信息数据库相关操作
|
*/
|
@Repository
|
class SceneRep(
|
private val scenseMapper: ScenseMapper,
|
private val userinfoMapper: UserinfoMapper,
|
private val subTaskRep: SubTaskRep,
|
) {
|
|
fun insert(scene: Scense?):Int {
|
scene ?: return 0
|
if (scene.guid == null) scene.guid = UUIDGenerator.generate16ShortUUID()
|
scene.createdate = Date()
|
return scenseMapper.insertSelective(scene)
|
}
|
|
fun findScene(userId: String?): Scense? {
|
val user = userinfoMapper.selectByPrimaryKey(userId) ?: throw BizException("用户id不存在")
|
return scenseMapper.selectByPrimaryKey(user.dGuid)
|
}
|
|
fun findScene(sceneId: String? = null, sceneName: String? = null): Scense? {
|
return scenseMapper.selectOne(Scense().apply {
|
guid = sceneId
|
name = sceneName
|
})
|
}
|
|
fun findSceneList(scene: Scense): List<Scense?> {
|
return scenseMapper.select(scene)
|
}
|
|
fun findSceneList(nameList: List<String?>): List<Scense?> {
|
return scenseMapper.selectByExample(Example(Scense::class.java).apply {
|
createCriteria().andIn("name", nameList)
|
})
|
}
|
|
/**
|
* 查找场景
|
*/
|
fun findSceneList(topTaskId: String, sceneTypeId: Int? = null, townCode: String? = null): List<Scense?> {
|
return scenseMapper.getSceneByType(topTaskId, sceneTypeId, townCode)
|
}
|
|
fun findBySubTask(subTaskId: String): Scense? {
|
val subtask = subTaskRep.findOne(subTaskId)
|
return scenseMapper.selectByPrimaryKey(subtask?.scenseid)
|
}
|
}
|
