TaxReceiver.java
2.61 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
package com.diligrp.tax.boot.receiver;
import com.diligrp.tax.central.context.Context;
import com.diligrp.tax.central.process.ProcessorChain;
import com.diligrp.tax.central.type.StatusType;
import com.diligrp.tax.central.type.SystemType;
import com.diligrp.tax.central.utils.JsonUtils;
import com.rabbitmq.client.Channel;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.annotation.Exchange;
import org.springframework.amqp.rabbit.annotation.Queue;
import org.springframework.amqp.rabbit.annotation.QueueBinding;
import org.springframework.amqp.rabbit.annotation.RabbitListener;
import org.springframework.stereotype.Component;
import java.io.IOException;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.nio.charset.StandardCharsets;
import java.util.Map;
import java.util.Optional;
import static com.diligrp.tax.boot.queue.TaxAutoPush.*;
/**
* @Author: zhangmeiyang
* @CreateTime: 2024-09-02 16:32
* @Version: todo
*/
@Component
@Slf4j
public class TaxReceiver {
@Resource
private Map<SystemType, ProcessorChain> processorChainMap;
@RabbitListener(bindings =
@QueueBinding(
value = @Queue(value = NORMAL_QUEUE, autoDelete = "false"),
exchange = @Exchange(value = NORMAL_EXCHANGE),
key = NORMAL_ROUTING),
ackMode = "MANUAL"
)
public void receiveMessage(Channel channel, Message message) throws IOException {
var content = new String(message.getBody(), StandardCharsets.UTF_8);
log.info("tax-agent收到消息:{}", content);
Context ctx = JsonUtils.fromJsonString(content, Context.class);
try {
handle(ctx);
channel.basicAck(message.getMessageProperties().getDeliveryTag(), false);
log.info("tax-agent消息处理成功:{}", content);
} catch (Exception e) {
channel.basicAck(message.getMessageProperties().getDeliveryTag(), false);
recordError(e, ctx);
log.error("tax-agent消息处理失败:", e);
}
}
private void handle(Context ctx) {
Context context = processorChainMap.get(SystemType.from(ctx.getSystemType())).startProcess(ctx);
context.setStatus(StatusType.SUCCESS.code);
//TODO write to db
}
private void recordError(Exception e, Context ctx) {
StringWriter sw = new StringWriter();
PrintWriter pw = new PrintWriter(sw);
e.printStackTrace(pw);
ctx.setError(sw.toString());
ctx.setStatus(StatusType.FAIL.code);
//TODO write error to db
}
}