策略模式实现多种支付方式

时间：2024-02-24 10:03:30

使用策略模式优雅的实现多种支付方式（支付宝、微信），或者多种支付场景（订单、维修金）的业务，且方便扩展。

下例是使用注解配合反射方式，扫描到所有的具体的支付策略并放到map集合中，然后根据前端传递来的支付类型参数，选择对应的支付策略，完成支付过程。

 

如上图：

PayStrategy是支付策略接口；

OrderPay（订单支付），RepairPay（维修金支付）是具体的支付策略（支付场景）；

AbstractPayService是实现PayStrategy接口的抽象类，规定了支付的流程，在其中选择具体的支付策略，为controller提供支付接口；

PayServiceImpl实现AbstractPayService中规定的支付流程的具体步骤，通过PayStrategyFactory获取实际的支付策略class，通过BeansUtil返回spring容器中的策略实现类bean；

PayStrategyFactory通过扫描被Pay注解注释的类，获得所有支付策略的类型和class；

Pay注解表名被注解的类是一个具体的支付策略，可以设置支付类型。

 

具体代码如下：

PayStrategy

package com.zhya.strategy.service;

/**
 * 支付策略
 *
 * @author zhangyang
 * @date 2018/11/14
 **/
public interface PayStrategy {

    /**
     * 支付前准备支付参数
     *
     * @param payFor
     * @return
     */
    boolean prePay(String payFor);

    /**
     * 支付后处理支付回调结果
     *
     * @param payFor
     * @param isPaySuccess
     */
    void afterPay(String payFor, boolean isPaySuccess);
}

 

Pay注解

package com.zhya.strategy.annotation;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

/**
 * 支付策略
 *
 * @author zhangyang
 * @date 2018/11/14
 **/
@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
public @interface Pay {
    /**
     * 支付类型
     *
     * @return
     */
    String value();
}

 

OrderPay订单支付

package com.zhya.strategy.service.impl;

import com.zhya.strategy.annotation.Pay;
import com.zhya.strategy.service.PayStrategy;
import org.springframework.stereotype.Service;

/**
 * 订单支付
 *
 * @author zhangyang
 * @date 2018/11/14
 **/
@Service
@Pay("order")
public class OrderPay implements PayStrategy {
    /**
     * 支付前准备支付参数
     *
     * @param payFor
     */
    @Override
    public boolean prePay(String payFor) {
        System.out.printf("-----------%s-----------\r\n", "订单支付");
        return true;
    }

    /**
     * 支付后处理支付回调结果
     *
     * @param payFor
     * @param isPaySuccess
     */
    @Override
    public void afterPay(String payFor, boolean isPaySuccess) {
        System.out.printf("-----------%s-----------\n", "订单支付回调");
    }
}

 

RepairPay维修金支付

package com.zhya.strategy.service.impl;

import com.zhya.strategy.annotation.Pay;
import com.zhya.strategy.service.PayStrategy;
import org.springframework.stereotype.Service;

/**
 * 维修金支付
 *
 * @author zhangyang
 * @date 2018/11/14
 **/
@Service
@Pay("repair")
public class RepairPay implements PayStrategy {
    /**
     * 支付前准备支付参数
     *
     * @param payFor
     */
    @Override
    public boolean prePay(String payFor) {
        System.out.printf("-----------%s-----------\r\n", "维修金支付");
        return true;
    }

    /**
     * 支付后处理支付回调结果
     *
     * @param payFor
     * @param isPaySuccess
     */
    @Override
    public void afterPay(String payFor, boolean isPaySuccess) {
        System.out.printf("-----------%s-----------\r\n", "维修金支付回调");
    }
}

 

BeansUtil

package com.zhya.strategy.util;

import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.stereotype.Component;

/**
 * bean工具类
 *
 * @author zhangyang
 * @date 2018/11/14
 **/
@Component
public class BeansUtil implements ApplicationContextAware {
    private static ApplicationContext applicationContext;

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        BeansUtil.applicationContext = applicationContext;
    }

    /**
     * 获取实例
     *
     * @param clazz
     * @return
     */
    public static Object getBean(Class clazz) {
        return applicationContext.getBean(clazz);
    }
}

 

PayStrategyFactory策略工厂

package com.zhya.strategy.util;

import com.zhya.strategy.annotation.Pay;
import com.zhya.strategy.service.PayStrategy;
import org.reflections.Reflections;
import org.springframework.util.StringUtils;

import java.util.HashMap;
import java.util.Map;
import java.util.Set;

/**
 * 支付策略工厂
 *
 * @author zhangyang
 * @date 2018/11/14
 **/
public class PayStrategyFactory {
    // 单例开始

    /**
     * 私有化构造函数，单例
     */
    private PayStrategyFactory() {

    }

    private static class Builder {
        private static final PayStrategyFactory payStrategyFactory = new PayStrategyFactory();
    }

    public static PayStrategyFactory getInstance() {
        return Builder.payStrategyFactory;
    }
    // 单例结束

    private static final String PAY_STRATEGY_IMPLEMENTATION_PACKAGE = "com.zhya.strategy.service.impl";
    private static final Map<String, Class> STRATEGY_MAP = new HashMap<>();

    // 获取所有支付策略
    static {
        Reflections reflections = new Reflections(PAY_STRATEGY_IMPLEMENTATION_PACKAGE);
        Set<Class<?>> classSet = reflections.getTypesAnnotatedWith(Pay.class);
        classSet.forEach(aClass -> {
            Pay payAnnotation = aClass.getAnnotation(Pay.class);
            STRATEGY_MAP.put(payAnnotation.value(), aClass);
        });
    }

    /**
     * 根据支付策略类型获取支付策略bean
     *
     * @param type
     * @return
     */
    public static PayStrategy getStrategy(String type) {
        // 反射获取支付策略实现类clazz
        Class clazz = STRATEGY_MAP.get(type);
        if (StringUtils.isEmpty(clazz)) {
            return null;
        }

        // 通过applicationContext获取bean
        return (PayStrategy) BeansUtil.getBean(clazz);
    }
}

 

AbstractPayService提供支付服务的抽象类

package com.zhya.strategy.service;

/**
 * 支付service
 *
 * @author zhangyang
 * @date 2018/11/14
 **/
public abstract class AbstractPayService implements PayStrategy {
    protected String payType;

    /**
     * 支付
     *
     * @param payType
     * @param payFor
     */
    public void pay(String payType, String payFor) {
        this.payType = payType;
        boolean isPrepared = prePay(payFor);

        if (isPrepared) {
            System.out.println("------------支付请求已提交------------");
        } else {
            System.out.println("------------支付请求提交失败------------");
            return;
        }
        afterPay(payFor, true);
    }
}

　　

PayServiceImpl支付服务实现类

package com.zhya.strategy.service.impl;

import com.zhya.strategy.service.AbstractPayService;
import com.zhya.strategy.service.PayStrategy;
import com.zhya.strategy.util.PayStrategyFactory;
import org.springframework.stereotype.Service;

/**
 * 支付service
 *
 * @author zhangyang
 * @date 2018/11/14
 **/
@Service
public class PayServiceImpl extends AbstractPayService {
    /**
     * 支付前准备支付参数
     *
     * @param payFor
     */
    @Override
    public boolean prePay(String payFor) {
        PayStrategy payStrategy = PayStrategyFactory.getStrategy(this.payType);
        if (payStrategy == null) {
            System.out.printf("没有%s类型的支付策略...\r\n", this.payType);
            return false;
        }
        return payStrategy.prePay(payFor);
    }

    /**
     * 支付后处理支付回调结果
     *
     * @param payFor
     * @param isPaySuccess
     */
    @Override
    public void afterPay(String payFor, boolean isPaySuccess) {
        PayStrategy payStrategy = PayStrategyFactory.getStrategy(this.payType);
        payStrategy.afterPay(payFor, true);
    }
}

　　

PayController

package com.zhya.strategy.controller;

import com.zhya.strategy.service.AbstractPayService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

/**
 * 支付
 *
 * @author zhangyang
 * @date 2018/11/14
 **/
@RestController
@RequestMapping("pay")
public class PayController {
    @Autowired
    private AbstractPayService payService;

    @GetMapping
    public void pay(@RequestParam String payType, @RequestParam String payFor) {
        payService.pay(payType, payFor);
    }
}

　　

 

相关文章