src/main/kotlin/cn/flightfeather/supervision/domain/ds2/repository/UserMapRep.kt
@@ -7,6 +7,8 @@
import cn.flightfeather.supervision.domain.ds2.entity.UserinfoTZ
import cn.flightfeather.supervision.domain.ds2.mapper.UserMapMapper
import org.springframework.stereotype.Repository
import tk.mybatis.mapper.entity.Example
import java.util.*
@Repository
class UserMapRep(
@@ -21,18 +23,32 @@
     * @return
     */
    fun findFromSupervision(userinfoTZ: UserinfoTZ?): Scense? {
        return findFromSupervision(userinfoTZ?.guid)
    }
    fun findFromSupervision(guid: String?): Scense? {
        val userMap = UserMap().apply {
            tzUserId = userinfoTZ?.guid
            tzUserId = guid
        }
        // 找到两个系统用户的对应关系
        val uMap = userMapMapper.selectOne(userMap)
        if (uMap == null) {
            println("${userinfoTZ?.guid}, ${userinfoTZ?.realname}")
            return null
        }
        val uMap = userMapMapper.selectOne(userMap) ?: return null
        // 找到飞羽监管中的用户信息
        val userInfoSp = userinfoMapper.selectByPrimaryKey(uMap.svUserId)
        // 找到飞羽监管中的场景信息
        return scenseMapper.selectByPrimaryKey(userInfoSp.dGuid)
    }
    /**
     * 通过飞羽监管用户id查询飞羽环境用户id
     */
    fun findBySVUserId(idList: List<String?>): List<UserMap?> {
        return userMapMapper.selectByExample(Example(UserMap::class.java).apply {
            createCriteria().andIn("svUserId", idList)
        })
    }
    fun insert(userMap: UserMap): Int {
        if (userMap.umCreateTime == null) userMap.umCreateTime = Date()
        return userMapMapper.insert(userMap)
    }
}