package cc.mrbird.febs.rabbit.producter;

import cc.mrbird.febs.mall.dto.ApiMemberChargeFailDto;
import cc.mrbird.febs.rabbit.enumerates.RabbitQueueEnum;
import com.alibaba.fastjson.JSON;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.AmqpException;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.core.MessagePostProcessor;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

/**
 * @author wzy
 * @date 2021-09-25
 **/
@Slf4j
@Component
public class AgentProducer implements RabbitTemplate.ConfirmCallback {

    /**
     * 配置中配置的RabbitTemplate的是prototype类型，不能直接注入
     */
    private RabbitTemplate rabbitTemplate;

    /**
     * 在构造方法上注入RabbitTemplate
     *
     * @param
     */
    @Autowired
    public AgentProducer(RabbitTemplate rabbitTemplate) {
        this.rabbitTemplate = rabbitTemplate;
        rabbitTemplate.setConfirmCallback(this);
    }

    @Override
    public void confirm(CorrelationData correlationData, boolean ack, String cause) {

    }

    /**
     * 购买VIP自动过期
     * @param apiMemberChargeFailDto
     */
    public void sendMemberChargeFailMsg(ApiMemberChargeFailDto apiMemberChargeFailDto) {
        log.info("发送购买VIP自动过期: {}", JSON.toJSONString(apiMemberChargeFailDto));
        rabbitTemplate.convertAndSend(
                RabbitQueueEnum.RUN_VIP_OPERATION_CHARGE_FAIL_TTL.getExchange(),
                RabbitQueueEnum.RUN_VIP_OPERATION_CHARGE_FAIL_TTL.getRoute(),
                apiMemberChargeFailDto,
                new MessagePostProcessor() {
                    @Override
                    public Message postProcessMessage(Message message) throws AmqpException {
                        message.getMessageProperties().setExpiration(String.valueOf(apiMemberChargeFailDto.getFailTime()));
                        return message;
                    }
                });
    }

    public void sendBuyVipSuccessMsg(Long id) {
        log.info("发送购买成功消息:{}",id);
        rabbitTemplate.convertAndSend(
                RabbitQueueEnum.RUN_VIP_OPERATION_CHARGE.getExchange(),
                RabbitQueueEnum.RUN_VIP_OPERATION_CHARGE.getRoute(),
                id);
    }

    public void sendNodeUpMsg(Long memberId) {
        log.info("发送节点升级消息:{}",memberId);
        rabbitTemplate.convertAndSend(
                RabbitQueueEnum.RUN_VIP_NODE_UP.getExchange(),
                RabbitQueueEnum.RUN_VIP_NODE_UP.getRoute(),
                memberId);
    }
}