情景描述
有些时候，我们需要根据不同的参数，做不同的操作。情况较少的时候，我们用if-else无可厚非，然而当有多种情况时，用if-else写出的代码总感觉观赏性、拓展性、13格有些不够自信。这个时候我们就需要对我们的代码进行一下优化。我始终坚信着：代码是可鉴赏的艺术品，机器运行只是它的附属功能！
以RocketMQ消费者接收消息为例
盗取阿里云提供的资料描述
Topic 和 Tag 的定义如下：
Topic
消息主题，通过 Topic 对不同的业务消息进行分类。
Tag
消息标签，用来进一步区分某个 Topic 下的消息分类，消息从生产者发出即带上的属性。
Topic 和 Tag 的关系如下图所示。
image.png
举例
我们是一个电商平台，现在有一个下单之后，订单中台需要发出一个topic为"placeOrder"的消息，具体消息tag则代表：女装、日用品、电器、奢侈品、数码、图书、水果、五金、汽车配件.....一共用了18种不同的tag做了具体业务区分。现在有一个数据统计分析系统，需要对这个topic下所有的tag对应的消息进行接收，并实时地完成针对不同的子业务做不同操作的计算。
那么问题来了，如果我们用if-else来进行处理，你的代码是这样的
if("a".equals(tag)) {
// do(msg);
}else if("b".equals(tag)) {
// do(msg);
} // ...
...
有的同学或者说，改成switch-case呀，其实也是一样的，代码都太难看了。并且以后如果增加了其它的业务还需要继续添加if-else，并且每次消费消息时，都需要从第一个if开始逐个往下进行字符串比较。
改造一下
第一种方式
提供一个静态map，维护tag->具体操作枚举实例的对应关系
直接上代码，其中编程思想自行领悟，哈哈
// MQ消费者监听类
@Component
@RocketMQMessageListener(topic = "placeOrder", consumerGroup = "data-system-consumer")
public class PlaceOrderDataSystemListener implements RocketMQListener<MessageDTO> {
@Autowired
private MessageHandler messageHandler;
@Override
public void onMessage(MessageDTO rechargeRequest) {
messageHandler.hand(rechargeRequest);
}
}
public class MessageDTO {
private String tag;
// 具体业务需要处理的参数对象jsonString
private String dataJson;
// ...
// 协调处理类
@Component
public class MessageHandler {
@Autowired
private IBusinessService businessService;
private static Map<String,EHandler> handlerMap = new HashMap<>(32);
static {
handlerMap.put("tag1",EHandler.A_BUSINESS_HANDLER);
handlerMap.put("tag2",EHandler.B_BUSINESS_HANDLER);
handlerMap.put("tag3",EHandler.C_BUSINESS_HANDLER);
// ...
}
// 控制跟逻辑分离的编程思想
public boolean hand(MessageDTO rechargeRequest) {
String tag = rechargeRequest.getTag();
EHandler handler = handlerMap.get(tag);
// 这句代码写在这可以达到复用的目的，写到枚举hand方法中，则需要写多次
MessageData messageData = handler.getMessageDataFrom(rechargeRequest.getDataJson());
boolean handResult = handler.hand(messageData, businessService);
return handResult;
}
}
interface IBusinessService {
boolean businessAHand(MessageData messageData);
boolean businessBHand(MessageData messageData);
boolean businessCHand(MessageData messageData);
// ...
}
enum EHandler {
A_BUSINESS_HANDLER() {
@Override
boolean hand(MessageData messageData, IBusinessService businessService) {
boolean result = businessService.businessAHand(messageData);
return result;
}
@Override
MessageData getMessageDataFrom(String dataJson) {
AMessageData aMessageData = JSONObject.parseObject(dataJson, AMessageData.class);
return aMessageData;
}
},
B_BUSINESS_HANDLER() {
@Override
boolean hand(MessageData messageData, IBusinessService businessService) {
return businessService.businessBHand(messageData);
}
@Override
MessageData getMessageDataFrom(String dataJson) {
BMessageData bMessageData = JSONObject.parseObject(dataJson, BMessageData.class);
return bMessageData;
}
},
C_BUSINESS_HANDLER() {
@Override
boolean hand(MessageData messageData, IBusinessService businessService) {
return businessService.businessCHand(messageData);
}
@Override
MessageData getMessageDataFrom(String dataJson) {
CMessageData cMessageData = JSONObject.parseObject(dataJson, CMessageData.class);
return cMessageData;
}
};
abstract boolean hand(MessageData messageData, IBusinessService businessService);
abstract MessageData getMessageDataFrom(String dataJson);
}
interface MessageData {}
public class AMessageData implements MessageData{}
public class BMessageData implements MessageData{}
public class CMessageData implements MessageData{}
@Service
public class BusinessService implements IBusinessService {
@Override
public boolean businessAHand(MessageData messageData) {
// do
return true;
}
@Override
public boolean businessBHand(MessageData messageData) {
// do
return true;
}
@Override
public boolean businessCHand(MessageData messageData) {
// do
return true;
}
}
第二种方式
Java8 Function
提供一个不相关的代码如下(加减法判断具体操作)
public class SimpleTest {
private static Map<String, Function<? super Hand, ? extends Integer>> optMap = new HashMap<>();
static {
optMap.put("+",Hand::add);
optMap.put("-", Hand::subtraction);
}
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int a = sc.nextInt();
String c = sc.next();
int b = sc.nextInt();
//...
int result = new Hand(a,b).execute(optMap.get(c));
System.out.println(result);
}
}
class Hand {
public Hand(int a, int b) {
this.a = a;
this.b = b;
}
private Integer a;
private Integer b;
public Integer add() {
return a+b;
}
public Integer subtraction() {
return a - b;
}
public Integer execute(Function<? super Hand, ? extends Integer> function) {
return function.apply(this);
}
}
