BirthdayNoticeService.java 3.05 KB
package com.pipihelper.project.feishu.service.anniversary;

import com.pipihelper.project.feishu.dto.FeiShuConfig;
import com.pipihelper.project.feishu.dto.msg.FeiShuMsgDTO;
import com.pipihelper.project.feishu.dto.employee.FeiShuEmployeeDTO;
import com.pipihelper.project.feishu.service.FeiShuApiService;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.time.LocalDate;
import java.util.List;
import java.util.stream.Collectors;

/**
 * 飞书生日提醒
 */
@Service
@Slf4j
public class BirthdayNoticeService extends AbstractNoticeService {

    @Autowired
    private FeiShuApiService feiShuApiService;

    @Resource
    private FeiShuConfig feiShuConfig;

    @Override
    public void notice() {
        // 1. 获取当天生日人
        List<FeiShuEmployeeDTO> feiShuEmployeeDTOList = feiShuApiService.getEmployeeList(feiShuConfig.getAnniversaryApp());
        List<FeiShuEmployeeDTO> birthdayEmployees = feiShuEmployeeDTOList.stream()
                .filter(feiShuEmployeeDTO -> StringUtils.isNotEmpty(feiShuEmployeeDTO.getSystemFields().getBirthday()))
                .filter(feiShuEmployeeDTO -> {
                    LocalDate birthday = LocalDate.parse(feiShuEmployeeDTO.getSystemFields().getBirthday());
                    // 使用plusYears可以避免2月29日的特殊情况.当年若无2月29日,生日则为2月28日
                    long years = LocalDate.now().getYear() - birthday.getYear();
                    return birthday.plusYears(years).equals(LocalDate.now());
                })
                .map(feiShuEmployeeDTO -> {
                    // 给员工自己发消息
                    feiShuApiService.sendMsg(generateEmployeeMsgCard(feiShuEmployeeDTO), "open_id", feiShuConfig.getAnniversaryApp());
                    // 给领导发送消息
                    sendMsgToLeader(generateLeaderMsgCard(feiShuEmployeeDTO), feiShuEmployeeDTO);

                    return feiShuEmployeeDTO;
                }).collect(Collectors.toList());
        log.info("{} 当日生日人: {}", LocalDate.now(), birthdayEmployees);

    }

    @Override
    public FeiShuMsgDTO generateEmployeeMsgCard(FeiShuEmployeeDTO feiShuEmployeeDTO) {
        FeiShuMsgDTO feiShuMsgDTO = new FeiShuMsgDTO();
        feiShuMsgDTO.setMsgType("interactive");
        feiShuMsgDTO.setContent(String.format(getInteractiveCardStr("/templates/birthday-employee.json"), feiShuEmployeeDTO.getSystemFields().getName()));
        feiShuMsgDTO.setReceiveId(feiShuEmployeeDTO.getUserId());
        return feiShuMsgDTO;
    }

    @Override
    public FeiShuMsgDTO generateLeaderMsgCard(FeiShuEmployeeDTO feiShuEmployeeDTO) {
        FeiShuMsgDTO feiShuMsgDTO = new FeiShuMsgDTO();
        feiShuMsgDTO.setMsgType("interactive");
        feiShuMsgDTO.setContent(String.format(getInteractiveCardStr("/templates/birthday-leader.json"), feiShuEmployeeDTO.getSystemFields().getName()));
        return feiShuMsgDTO;
    }
}