fire_point/src/main/java/com/xkrs/controller/PushController.java

package com.xkrs.controller;

import com.xkrs.common.encapsulation.PromptMessageEnum;
import com.xkrs.dao.SysUserDao;
import com.xkrs.model.entity.SysUserEntity;
import com.xkrs.model.qo.SysUserPushAccountQo;
import com.xkrs.sms.PushHelper;
import com.xkrs.utils.ListUtils;
import org.apache.hc.core5.util.TextUtils;
import org.springframework.context.i18n.LocaleContextHolder;
import org.springframework.web.bind.annotation.*;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.List;
import java.util.Locale;
import java.util.Optional;

import static com.xkrs.common.encapsulation.OutputEncapsulation.outputEncapsulationObject;

@RestController
@RequestMapping(value = "/push")
public class PushController {

    private final Locale locale = LocaleContextHolder.getLocale();

    @Resource
    private SysUserDao sysUserDao;

    @Resource
    private PushHelper pushHelper;

    @PostMapping("/bindUserPushInfo")
    public String bindUserPushInfo(@RequestBody SysUserPushAccountQo sysUserPushAccountQo) {
        Integer id = sysUserPushAccountQo.getId();
        String userAccount = sysUserPushAccountQo.getUserAccount();
        String regID = sysUserPushAccountQo.getRegID();
        if (null == id) {
            return outputEncapsulationObject(PromptMessageEnum.PROCESS_FAIL, "绑定失败，id == null", locale);
        }
        Optional<SysUserEntity> targetSysUserOptional = sysUserDao.findById(id);
        if (targetSysUserOptional.isEmpty()) {
            return outputEncapsulationObject(PromptMessageEnum.PROCESS_FAIL, "绑定失败，id不存在", locale);
        }
        if (TextUtils.isEmpty(userAccount)) {
            return outputEncapsulationObject(PromptMessageEnum.PROCESS_FAIL, "绑定失败，userAccount == null", locale);
        }
        if (TextUtils.isEmpty(regID)) {
            return outputEncapsulationObject(PromptMessageEnum.PROCESS_FAIL, "绑定失败，regID == null", locale);
        }
        SysUserEntity targetSysUser = targetSysUserOptional.get();
        // 更新 userAccount
        targetSysUser.setPushUserAccount(userAccount);
        // 更新 regID
        String pushRegIDs = targetSysUser.getPushRegID();
        if (TextUtils.isEmpty(pushRegIDs)) {
            targetSysUser.setPushRegID(regID);
        } else {
            List<String> pushRegIDList = new ArrayList<>();
            if (pushRegIDs.contains(",")) {
                pushRegIDList.addAll(ListUtils.toStringList(pushRegIDs, ","));
            } else {
                pushRegIDList.add(pushRegIDs);
            }
            if (!pushRegIDList.contains(regID)) {
                pushRegIDList.add(regID);
            }
            String newPushRegIDs = ListUtils.fromStringList(pushRegIDList, ",");
            targetSysUser.setPushRegID(newPushRegIDs);
        }
        sysUserDao.save(targetSysUser);
        return outputEncapsulationObject(PromptMessageEnum.SUCCESS, "绑定成功", locale);
    }

    @GetMapping("/testPush")
    public String testPush(@RequestParam(value = "userAccount") String userAccount) {
        //推送
        try {
            List<String> userAccountList = new ArrayList<>();
            userAccountList.add(userAccount);
            pushHelper.dispatchPushMessage(userAccountList, null);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return outputEncapsulationObject(PromptMessageEnum.SUCCESS, "OKK", locale);
    }

}