wePush/src/main/java/com/kimgo/wepush/service/HeartBeatService.java

package com.kimgo.wepush.service;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
import com.kimgo.wepush.exceptions.DatabaseQueryException;
import com.kimgo.wepush.mapper.DeviceInfoDAOMapper;
import com.kimgo.wepush.model.DeviceInfo;
import com.kimgo.wepush.model.DeviceInfoDAO;
import com.kimgo.wepush.response.ServerResponseEntity;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Service;

@Service
public class HeartBeatService {
    private final Logger logger = LoggerFactory.getLogger(HeartBeatService.class);
    private final long TIME_OUT_PERIOD = 600;
    private final TokenService tokenService;
    private final DeviceInfoDAOMapper deviceInfoDAOMapper;

    public HeartBeatService(TokenService tokenService, DeviceInfoDAOMapper deviceInfoDAOMapper) {
        this.tokenService = tokenService;
        this.deviceInfoDAOMapper = deviceInfoDAOMapper;
    }

    public ServerResponseEntity handleHeartbeatSignal(String accessToken, DeviceInfo deviceInfo) {
        String correctAccessToken = tokenService.getApiAccessToken();
        logger.info("accessToken: {} correctAccessToken: {}", accessToken, correctAccessToken);
        if (!correctAccessToken.equals(accessToken)) {
            return ServerResponseEntity.fail("Invalid accessToken");
        }
        try {
            DeviceInfoDAO deviceInfoDAO = getDeviceInfoBySN(deviceInfo.getSN());
            if (deviceInfoDAO == null) {
                // 设备信息不存在，可能是无效的设备
                logger.info("device not find,start add device to mysql.");
                boolean isSuccess = addDeviceInfo(deviceInfo);
                if (isSuccess) {
                    return ServerResponseEntity.success();
                } else {
                    logger.info("add to mysql error");
                    return ServerResponseEntity.fail("server error,please check log.");
                }
            }
        } catch (DatabaseQueryException e) {
            // 处理查询失败的情况
            logger.error("Database query failed.", e);
            return ServerResponseEntity.fail("Database query failed");
        } catch (Exception e) {
            logger.error("update client status error.");
            return ServerResponseEntity.fail("Error updating client status");
        }
        try {
            updateClientStatus(deviceInfo);
            return ServerResponseEntity.success();
        } catch (Exception e) {
            logger.error("update client status error.");
            // 处理更新状态时的异常
            return ServerResponseEntity.fail("Error updating client status");
        }
    }


    public void updateClientStatus(DeviceInfo deviceInfo) {
        long currentTimeMillis = System.currentTimeMillis();
        // 更新该客户端的最后活跃时间

        UpdateWrapper<DeviceInfoDAO> updateWrapper = new UpdateWrapper<>();
        updateWrapper.eq("sn", deviceInfo.getSN());
        updateWrapper.set("last_online_time", currentTimeMillis);

        int result = deviceInfoDAOMapper.update(null, updateWrapper);
        if (result > 0) {
            logger.debug("Update successful");
        } else {
            logger.warn("Update failed: No rows affected");
        }
    }

    public boolean addDeviceInfo(DeviceInfo deviceInfo) {
        logger.info("start add device. device info: {}", deviceInfo);
        try {
            DeviceInfoDAO deviceInfoDAO = convertToDeviceInfoDAO(deviceInfo);
            int result = deviceInfoDAOMapper.insert(deviceInfoDAO);
            logger.info("result: {}", result);
            return result > 0;
        } catch (Exception e) {
            logger.error("Error adding device info", e);
            return false;
        }
    }

    /**
     * 根据 androidId 查询 DeviceInfo。
     *
     * @param SN 要查询的 SN。
     * @return 与 androidId 对应的 DeviceInfoDAO 对象，如果没有找到，则返回 null。
     */
    public DeviceInfoDAO getDeviceInfoBySN(String SN) throws DatabaseQueryException {
        QueryWrapper<DeviceInfoDAO> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("sn", SN);
        try {
            DeviceInfoDAO deviceInfoDAO = deviceInfoDAOMapper.selectOne(queryWrapper);
            if (deviceInfoDAO != null) {
                logger.info("Data is queried in the database based on the SN");
                return deviceInfoDAO;
            }
            return null; // 数据不存在
        } catch (Exception e) {
            logger.error("Error querying device info by SN", e);
            throw new DatabaseQueryException("Error querying device info by SN", e);
        }
    }


    /**
     * 将 DeviceInfo 转换为 DeviceInfoDAO。
     *
     * @param deviceInfo 要转换的 DeviceInfo 对象。
     * @return 转换后的 DeviceInfoDAO 对象。
     */
    private DeviceInfoDAO convertToDeviceInfoDAO(DeviceInfo deviceInfo) {
        logger.info("start convert to device info dao");
        // 这里需要实现将 DeviceInfo 转换为 DeviceInfoDAO 的逻辑
        // 示例代码（根据实际字段调整）：
        DeviceInfoDAO deviceInfoDAO = new DeviceInfoDAO();
        deviceInfoDAO.setIsMonitored(1);
        deviceInfoDAO.setDeviceBrand(deviceInfo.getDeviceBrand());
        deviceInfoDAO.setSN(deviceInfo.getSN());
        deviceInfoDAO.setDeviceModel(deviceInfo.getDeviceModel());
        deviceInfoDAO.setAndroidVersion(deviceInfo.getAndroidVersion());
        deviceInfoDAO.setTimeOutPeriod(TIME_OUT_PERIOD);
        deviceInfoDAO.setLastOnlineTime(System.currentTimeMillis());
        deviceInfoDAO.setStatus(1);
        return deviceInfoDAO;
    }
}