类型枚举

  1. import lombok.Getter;
  3. @Getter
  4. @AllArgsConstructor
  5. public enum EventTypeEnum {
  6.     WE_CHAT(0),
  7.     ALI_PAY(1);
  8.     private int value;
  9. }

支付接口

  1. /**
  2.  * @author yuhui_cai
  3.  */
  4. public interface IPayService {
  5.     /**
  6.      * 逻辑
  7.      * @return
  8.      */
  9.     boolean dealEvent();
  11.     /**
  12.      * 获取事件类型
  13.      * @return
  14.      */
  15.     int getType();
  16. }

阿里支付实现类

  1. import com.test.demo.textdemo.strategy.enumeration.EventTypeEnum;
  2. import com.test.demo.textdemo.strategy.service.IPayService;
  3. import org.springframework.stereotype.Service;
  5. /**
  6.  * @author yuhui_cai
  7.  */
  8. @Service
  9. public class AliPayServiceImpl implements IPayService {
  10.     @Override
  11.     public boolean dealEvent() {
  12.         System.out.println("aliPay");
  13.         return true;
  14.     }
  16.     @Override
  17.     public int getType() {
  18.         return EventTypeEnum.ALI_PAY.getValue();
  19.     }
  20. }

微信支付实现类

  1. import com.test.demo.textdemo.strategy.enumeration.EventTypeEnum;
  2. import com.test.demo.textdemo.strategy.service.IPayService;
  3. import org.springframework.stereotype.Service;
  5. /**
  6.  * @author yuhui_cai
  7.  */
  8. @Service
  9. public class WeChatServiceImpl implements IPayService {
  10.     @Override
  11.     public boolean dealEvent() {
  12.         System.out.println("weixin");
  13.         return true;
  14.     }
  16.     @Override
  17.     public int getType() {
  18.         return EventTypeEnum.WE_CHAT.getValue();
  19.     }
  20. }

策略类

  1. import com.test.demo.textdemo.strategy.service.IPayService;
  2. import org.springframework.stereotype.Service;
  4. import java.util.HashMap;
  5. import java.util.List;
  6. import java.util.Map;
  8. @Service
  9. public class StrategyService {
  10.     Map<Integer, IPayService> eventServiceMap = new HashMap<>();
  12.     public StrategyService(List<IPayService> iStrategyServices) {
  13.         for (IPayService iStrategyService : iStrategyServices) {
  14.             eventServiceMap.put(iStrategyService.getType(), iStrategyService);
  15.         }
  16.     }
  18.     public boolean dealEvent(int eventType) {
  19.         IPayService iStrategyService = eventServiceMap.get(eventType);
  20.         return iStrategyService.dealEvent();
  21.     }
  22. }

单元测试

  1. @SpringBootTest(classes = TextDemoApplication.class)
  2. @RunWith(SpringRunner.class)
  3. public class TestDemo2 {
  4.     @Autowired
  5.     private StrategyService strategyService;
  7.     @Test
  8.     public void test(){
  9.         strategyService.dealEvent(0);
  10.         strategyService.dealEvent(1);
  11.     }
  12. }

结果

  1. 2021-05-08 09:06:22.625  WARN 2408 --- [           main] c.b.m.core.metadata.TableInfoHelper      : Can not find table primary key in Class: "com.test.demo.textdemo.entity.PmsSystemDatabase".
  2. 2021-05-08 09:06:23.345  INFO 2408 --- [           main] o.s.s.concurrent.ThreadPoolTaskExecutor  : Initializing ExecutorService 'applicationTaskExecutor'
  3. 2021-05-08 09:06:23.779  INFO 2408 --- [           main] yuhui.TestDemo2                          : Started TestDemo2 in 3.612 seconds (JVM running for 4.746)
  4. weixin
  5. aliPay
  6. 2021-05-08 09:06:24.014  INFO 2408 --- [extShutdownHook] o.s.s.concurrent.ThreadPoolTaskExecutor  : Shutting down ExecutorService 'applicationTaskExecutor'
  7. 2021-05-08 09:06:24.015  INFO 2408 --- [extShutdownHook] com.alibaba.druid.pool.DruidDataSource   : {dataSource-1} closing ...
  8. 2021-05-08 09:06:24.017  INFO 2408 --- [extShutdownHook] com.alibaba.druid.pool.DruidDataSource   : {dataSource-1} closed
  10. Process finished with exit code 0