使用Spring AMQP實現RPC異步調用
示列
服務器端
應用啟動類代碼,
import org.springframework.context.annotation.AnnotationConfigApplicationContext;
import org.springframework.context.annotation.ComponentScan;
import java.util.concurrent.TimeUnit;
@ComponentScan
public class Application {
public static void main(String[] args) throws Exception{
AnnotationConfigApplicationContext context = new AnnotationConfigApplicationContext(Application.class);
System.out.println("===server startup======");
TimeUnit.SECONDS.sleep(120);
context.close();
}
}
配置類:
監聽了sms
隊列,這個隊列將會是客戶端請求消息發送到的隊列,配置了適配器,適配器中去調用服務,適配器返回的值就是服務端返回給客戶端的RPC調用的結果
import org.springframework.amqp.core.AcknowledgeMode;
import org.springframework.amqp.rabbit.connection.CachingConnectionFactory;
import org.springframework.amqp.rabbit.connection.ConnectionFactory;
import org.springframework.amqp.rabbit.core.RabbitAdmin;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.rabbit.listener.SimpleMessageListenerContainer;
import org.springframework.amqp.rabbit.listener.adapter.MessageListenerAdapter;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
@Configuration
public class MQConfig {
@Bean
public ConnectionFactory connectionFactory(){
CachingConnectionFactory factory = new CachingConnectionFactory();
factory.setUri("amqp://zhihao.miao:123456@192.168.1.131:5672");
return factory;
}
@Bean
public RabbitAdmin rabbitAdmin(ConnectionFactory connectionFactory){
RabbitAdmin rabbitAdmin = new RabbitAdmin(connectionFactory);
return rabbitAdmin;
}
@Bean
public RabbitTemplate rabbitTemplate(ConnectionFactory connectionFactory){
RabbitTemplate rabbitTemplate = new RabbitTemplate(connectionFactory);
return rabbitTemplate;
}
@Bean
public SimpleMessageListenerContainer messageListenerContainer(ConnectionFactory connectionFactory){
SimpleMessageListenerContainer container = new SimpleMessageListenerContainer();
container.setConnectionFactory(connectionFactory);
container.setQueueNames("sms");
container.setAcknowledgeMode(AcknowledgeMode.NONE);
//使用適配器的方式
container.setMessageListener(new MessageListenerAdapter(new SendSMSHandler()));
return container;
}
}
處理器,處理器中調用具體的服務,我們此列子中處理器方法返回的值是boolean類型
import java.util.concurrent.TimeUnit;
public class SendSMSHandler {
public boolean handleMessage(byte[] body){
String _body = new String(body);
System.out.println(_body);
String[] sms = _body.split(":");
String phone = sms[0];
String content = sms[1];
boolean is = SendSMSTool.sendSMS(phone,content);
try {
TimeUnit.SECONDS.sleep(6);
} catch (InterruptedException e) {
e.printStackTrace();
}
return is;
}
}
服務接口
public class SendSMSTool {
public static boolean sendSMS(String phone,String content){
System.out.println("發送短信內容:【"+content+"】到手機號:"+phone);
return phone.length() > 6;
}
}
服務端步驟
- 消息處理方法,一定要有返回值,這個返回值就是就是server回復客戶端的結果。比如我們
SendSMSHandler.handleMessage
方法返回的值。
客戶端
應用啟動類:
import org.springframework.amqp.core.Message;
import org.springframework.amqp.core.MessageProperties;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.rabbit.support.CorrelationData;
import org.springframework.context.annotation.AnnotationConfigApplicationContext;
import org.springframework.context.annotation.ComponentScan;
import java.util.UUID;
import java.util.concurrent.TimeUnit;
@ComponentScan
public class Application {
public static void main(String[] args) throws Exception{
AnnotationConfigApplicationContext context = new AnnotationConfigApplicationContext(Application.class);
RabbitTemplate rabbitTemplate = context.getBean(RabbitTemplate.class);
//設置超時時間,單位是ms
rabbitTemplate.setReplyTimeout(10000);
String phone = "15634344321";
String content ="周年慶,五折優惠";
MessageProperties messageProperties = new MessageProperties();
Message message = new Message((phone+":"+content).getBytes(),messageProperties);
//rabbitTemplate.send("","sms",message);
Message reply = rabbitTemplate.sendAndReceive("","sms",message,
new CorrelationData(UUID.randomUUID().toString()));
System.out.println(reply);
System.out.println("message,body:"+new String(reply.getBody()));
System.out.println("message,properties:"+reply.getMessageProperties());
TimeUnit.SECONDS.sleep(30);
context.close();
}
}
配置類代碼:
import org.springframework.amqp.rabbit.connection.CachingConnectionFactory;
import org.springframework.amqp.rabbit.connection.ConnectionFactory;
import org.springframework.amqp.rabbit.core.RabbitAdmin;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
@Configuration
public class MQConfig {
@Bean
public ConnectionFactory connectionFactory(){
CachingConnectionFactory factory = new CachingConnectionFactory();
factory.setUri("amqp://zhihao.miao:123456@192.168.1.131:5672");
return factory;
}
@Bean
public RabbitAdmin rabbitAdmin(ConnectionFactory connectionFactory){
RabbitAdmin rabbitAdmin = new RabbitAdmin(connectionFactory);
return rabbitAdmin;
}
@Bean
public RabbitTemplate rabbitTemplate(ConnectionFactory connectionFactory){
RabbitTemplate rabbitTemplate = new RabbitTemplate(connectionFactory);
return rabbitTemplate;
}
}
如果服務端睡眠6s,則客戶端通過sendAndReceive
方法接收到的Message
對象為空,怎樣設置呢?
客戶端通過設置rabbitTemplate.setReplyTimeout(10000);
就可以了。
客戶端步驟
- 使用
sendAndReceive
方法發送消息,該方法返回一個Message對象,該對象就是server返回的結果 -
sendAndReceive
如果超過5s還沒有收到結果,則返回null,這個超時時間可以通過rabbitTemplate.setReplyTimeout()
來進行設置 - server端返回的結果一定要注意,和
MessageConverter
有關,默認的org.springframework.amqp.support.converter.SimpleMessageConverter
會把基本的數據類型轉換成Serializable
對象,這樣的話,client端接收的也是序列化的java對象,所以,需要合理設置MessageConverter
。
示列代碼中服務端返回給客戶端的是Boolean類型,
啟動服務端客戶端代碼:
服務器打印控制臺打印:
15634344321:周年慶,五折優惠
發送短信內容:【周年慶,五折優惠】到手機號:15634344321
客戶端控制臺打印:
message,body:????sr??java.lang.Boolean? r???????Z??valuexp?
message,properties:MessageProperties [headers={}, timestamp=null, messageId=null, userId=null, receivedUserId=null, appId=null, clusterId=null, type=null, correlationId=null, correlationIdString=null, replyTo=null, contentType=application/x-java-serialized-object, contentEncoding=null, contentLength=0, deliveryMode=null, receivedDeliveryMode=PERSISTENT, expiration=null, priority=0, redelivered=false, receivedExchange=, receivedRoutingKey=amq.rabbitmq.reply-to.g2dkAB5yYWJiaXRAaVpicDFqY3d4N3NmYjFud3pyZWh5NloAAFu0AAAABwI=.kHL9zxtdQmtcxl0mQF8zrg==, receivedDelay=null, deliveryTag=1, messageCount=null, consumerTag=null, consumerQueue=null]
我們發現客戶端接收到的數據亂碼,將服務端的處理器的返回值改寫成String類型的,
import java.util.concurrent.TimeUnit;
public class SendSMSHandler {
public String handleMessage(byte[] body){
String _body = new String(body);
System.out.println(_body);
String[] sms = _body.split(":");
String phone = sms[0];
String content = sms[1];
boolean is = SendSMSTool.sendSMS(phone,content);
try {
TimeUnit.SECONDS.sleep(6);
} catch (InterruptedException e) {
e.printStackTrace();
}
return is ? "success":"false";
}
}
此時發現客戶端接收的消息數據沒有亂碼,原因何在?我們總結一下就是服務器端處理器返回給客戶端boolean類型,那么返回的消息數據就亂碼,如果返回的是String類型,那么返回的消息數據就不會亂碼。
之前我們學習了org.springframework.amqp.support.converter.MessageConverter
接口,當客戶端向服務端發送消息的時候會進行消息類型轉換,調用了fromMessage
方法,而當服務器返回給客戶端的時候會將服務端的對象轉換成Message
對象,很明顯調用的是toMessage
方法。
我們知道org.springframework.amqp.support.converter.MessageConverter
接口的默認實現是org.springframework.amqp.support.converter.SimpleMessageConverter
,而toMessage
方法的實現是在其繼承的對象AbstractMessageConverter
中,
我們看到其AbstractMessageConverter.toMessage
方法的實現邏輯是:
@Override
public final Message toMessage(Object object, MessageProperties messageProperties)
throws MessageConversionException {
if (messageProperties == null) {
messageProperties = new MessageProperties();
}
//將對象轉換成Message對象
Message message = createMessage(object, messageProperties);
messageProperties = message.getMessageProperties();
if (this.createMessageIds && messageProperties.getMessageId() == null) {
messageProperties.setMessageId(UUID.randomUUID().toString());
}
return message;
}
其createMessage
方法就是將對象轉換成Message
對象,
/**
* Creates an AMQP Message from the provided Object.
*/
@Override
protected Message createMessage(Object object, MessageProperties messageProperties) throws MessageConversionException {
byte[] bytes = null;
if (object instanceof byte[]) {
bytes = (byte[]) object;
messageProperties.setContentType(MessageProperties.CONTENT_TYPE_BYTES);
}
else if (object instanceof String) {
try {
bytes = ((String) object).getBytes(this.defaultCharset);
}
catch (UnsupportedEncodingException e) {
throw new MessageConversionException(
"failed to convert to Message content", e);
}
messageProperties.setContentType(MessageProperties.CONTENT_TYPE_TEXT_PLAIN);
messageProperties.setContentEncoding(this.defaultCharset);
}
//因為boolean類型實現Serializable接口,所以會將其序列化
else if (object instanceof Serializable) {
try {
bytes = SerializationUtils.serialize(object);
}
catch (IllegalArgumentException e) {
throw new MessageConversionException(
"failed to convert to serialized Message content", e);
}
messageProperties.setContentType(MessageProperties.CONTENT_TYPE_SERIALIZED_OBJECT);
}
if (bytes != null) {
messageProperties.setContentLength(bytes.length);
}
return new Message(bytes, messageProperties);
}
我們在程序中將序列化對象直接轉換成字符串所以亂碼,而返回的是String類型的情形的時候先將字符串轉換成相應的字節數組,然后返回new Message(bytes, messageProperties);
就不會亂碼。
繼續探討,當我們服務端返回的是一個對象的時候,客戶端會返回空
返回的對象:
public class SendStatus {
private String phone;
private String result;
public String getPhone() {
return phone;
}
public void setPhone(String phone) {
this.phone = phone;
}
public String getResult() {
return result;
}
public void setResult(String result) {
this.result = result;
}
}
將該對象返回:
public class SendSMSHandler {
public SendStatus handleMessage(byte[] body){
String _body = new String(body);
System.out.println(_body);
String[] sms = _body.split(":");
String phone = sms[0];
String content = sms[1];
boolean is = SendSMSTool.sendSMS(phone,content);
SendStatus sendStatus = new SendStatus();
sendStatus.setPhone(phone);
sendStatus.setResult(is ? "SUCCESS":"FAILURE");
return sendStatus;
}
}
服務端控制臺:
15634344321:周年慶,五折優惠
發送短信內容:【周年慶,五折優惠】到手機號:15634344321
客戶端:
message,body:
message,properties:MessageProperties [headers={}, timestamp=null, messageId=null, userId=null, receivedUserId=null, appId=null, clusterId=null, type=null, correlationId=null, correlationIdString=null, replyTo=null, contentType=application/octet-stream, contentEncoding=null, contentLength=0, deliveryMode=null, receivedDeliveryMode=PERSISTENT, expiration=null, priority=0, redelivered=false, receivedExchange=, receivedRoutingKey=amq.rabbitmq.reply-to.g2dkAB5yYWJiaXRAaVpicDFqY3d4N3NmYjFud3pyZWh5NloAAFxBAAAABwI=.01fOGW/nvS2nz6gKza+cjg==, receivedDelay=null, deliveryTag=1, messageCount=null, consumerTag=null, consumerQueue=null]
原因何在,因為我們定義的SendStatus
不走createMessage
中的所有if分支,最后返回的是null,怎么解決呢,要么自己去定義一個org.springframework.amqp.support.converter.MessageConverter
實現,要么換一個默認的org.springframework.amqp.support.converter.MessageConverter
實現。
改造后的示列
使用AMQP自帶的消息類型轉換器Jackson2JsonMessageConverter
服務端
應用啟動類,
import org.springframework.context.annotation.AnnotationConfigApplicationContext;
import org.springframework.context.annotation.ComponentScan;
import java.util.concurrent.TimeUnit;
@ComponentScan
public class Application {
public static void main(String[] args) throws Exception{
AnnotationConfigApplicationContext context = new AnnotationConfigApplicationContext(Application.class);
System.out.println("===server startup======");
TimeUnit.SECONDS.sleep(120);
context.close();
}
}
配置類,添加自定義的消息轉換器
import org.springframework.amqp.core.AcknowledgeMode;
import org.springframework.amqp.rabbit.connection.CachingConnectionFactory;
import org.springframework.amqp.rabbit.connection.ConnectionFactory;
import org.springframework.amqp.rabbit.core.RabbitAdmin;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.rabbit.listener.SimpleMessageListenerContainer;
import org.springframework.amqp.rabbit.listener.adapter.MessageListenerAdapter;
import org.springframework.amqp.support.converter.Jackson2JsonMessageConverter;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
@Configuration
public class MQConfig {
@Bean
public ConnectionFactory connectionFactory(){
CachingConnectionFactory factory = new CachingConnectionFactory();
factory.setUri("amqp://zhihao.miao:123456@192.168.1.131:5672");
return factory;
}
@Bean
public RabbitAdmin rabbitAdmin(ConnectionFactory connectionFactory){
RabbitAdmin rabbitAdmin = new RabbitAdmin(connectionFactory);
return rabbitAdmin;
}
@Bean
public RabbitTemplate rabbitTemplate(ConnectionFactory connectionFactory){
RabbitTemplate rabbitTemplate = new RabbitTemplate(connectionFactory);
return rabbitTemplate;
}
@Bean
public SimpleMessageListenerContainer messageListenerContainer(ConnectionFactory connectionFactory){
SimpleMessageListenerContainer container = new SimpleMessageListenerContainer();
container.setConnectionFactory(connectionFactory);
container.setQueueNames("sms");
container.setAcknowledgeMode(AcknowledgeMode.NONE);
//使用適配器的方式
container.setMessageListener(new MessageListenerAdapter(new SendSMSHandler(),new Jackson2JsonMessageConverter()));
return container;
}
}
處理器handler,返回自定義的SendStatus類型
import java.util.concurrent.TimeUnit;
public class SendSMSHandler {
public SendStatus handleMessage(byte[] body){
String _body = new String(body);
System.out.println(_body);
String[] sms = _body.split(":");
String phone = sms[0];
String content = sms[1];
boolean is = SendSMSTool.sendSMS(phone,content);
SendStatus sendStatus = new SendStatus();
sendStatus.setPhone(phone);
sendStatus.setResult(is ? "SUCCESS":"FAILURE");
try {
TimeUnit.SECONDS.sleep(6);
} catch (InterruptedException e) {
e.printStackTrace();
}
return sendStatus;
}
}
接口服務,
public class SendSMSTool {
public static boolean sendSMS(String phone,String content){
System.out.println("發送短信內容:【"+content+"】到手機號:"+phone);
return phone.length() > 6;
}
}
服務端返回的對象,
public class SendStatus {
private String phone;
private String result;
public String getPhone() {
return phone;
}
public void setPhone(String phone) {
this.phone = phone;
}
public String getResult() {
return result;
}
public void setResult(String result) {
this.result = result;
}
}
客戶端
應用啟動類,
import org.springframework.amqp.core.Message;
import org.springframework.amqp.core.MessageProperties;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.rabbit.support.CorrelationData;
import org.springframework.context.annotation.AnnotationConfigApplicationContext;
import org.springframework.context.annotation.ComponentScan;
import java.util.UUID;
import java.util.concurrent.TimeUnit;
@ComponentScan
public class Application {
public static void main(String[] args) throws Exception{
AnnotationConfigApplicationContext context = new AnnotationConfigApplicationContext(Application.class);
RabbitTemplate rabbitTemplate = context.getBean(RabbitTemplate.class);
//設置超時時間,單位是ms
rabbitTemplate.setReplyTimeout(10000);
String phone = "15634344321";
String content ="周年慶,五折優惠";
MessageProperties messageProperties = new MessageProperties();
Message message = new Message((phone+":"+content).getBytes(),messageProperties);
//rabbitTemplate.send("","sms",message);
Message reply = rabbitTemplate.sendAndReceive("","sms",message,
new CorrelationData(UUID.randomUUID().toString()));
System.out.println(reply);
System.out.println("message,body:"+new String(reply.getBody()));
System.out.println("message,properties:"+reply.getMessageProperties());
TimeUnit.SECONDS.sleep(30);
context.close();
}
}
配置類
import org.springframework.amqp.rabbit.connection.CachingConnectionFactory;
import org.springframework.amqp.rabbit.connection.ConnectionFactory;
import org.springframework.amqp.rabbit.core.RabbitAdmin;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
@Configuration
public class MQConfig {
@Bean
public ConnectionFactory connectionFactory(){
CachingConnectionFactory factory = new CachingConnectionFactory();
factory.setUri("amqp://zhihao.miao:123456@192.168.1.131:5672");
return factory;
}
@Bean
public RabbitAdmin rabbitAdmin(ConnectionFactory connectionFactory){
RabbitAdmin rabbitAdmin = new RabbitAdmin(connectionFactory);
return rabbitAdmin;
}
@Bean
public RabbitTemplate rabbitTemplate(ConnectionFactory connectionFactory){
RabbitTemplate rabbitTemplate = new RabbitTemplate(connectionFactory);
return rabbitTemplate;
}
}
啟動服務器客戶端,客戶端返回
message,body:{"phone":"15634344321","result":"SUCCESS"}
message,properties:MessageProperties [headers={__TypeId__=rpc.server.SendStatus}, timestamp=null, messageId=null, userId=null, receivedUserId=null, appId=null, clusterId=null, type=null, correlationId=null, correlationIdString=null, replyTo=null, contentType=application/json, contentEncoding=UTF-8, contentLength=0, deliveryMode=null, receivedDeliveryMode=PERSISTENT, expiration=null, priority=0, redelivered=false, receivedExchange=, receivedRoutingKey=amq.rabbitmq.reply-to
返回了SendStatus的JSON格式,因為使用了Jackson2JsonMessageConverter
消息類型轉換器。