设计模式之 策略模式

策略模式

package com.hq.cloud.structure.trade;

import com.xy.common.msg.Message;
import com.xy.common.msg.Msg;
import com.xy.common.verify.Asserts;
import com.xy.pay.core.joint.JointTradeFacePayService;
import com.xy.pay.core.structure.model.TradeFacePay;
import com.xy.pay.core.structure.result.TradeFacePayResult;
import com.xy.pay.dao.core.dict.JointProviderEnum;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.context.annotation.Primary;
import org.springframework.core.env.Environment;
import org.springframework.stereotype.Service;

import java.util.Map;
import java.util.Objects;
import java.util.concurrent.ConcurrentHashMap;

/**
 * @author yangjunxiong
 * @date 2019/4/29 11:34
 */
@Primary
@Service
public class TradeFacePayService implements JointTradeFacePayService<TradeFacePay>, ApplicationContextAware {

    @Autowired
    private Environment                           environment;
    private Map<String, JointTradeFacePayService> facePayServiceBeans = new ConcurrentHashMap<>();


    /**
     * 人脸支付 <br>
     *
     * @param facePay
     * @return
     */
    @Override
    public final Msg<TradeFacePayResult> facePay(TradeFacePay facePay) {

        Asserts.notNull(facePay);
        if (environment.acceptsProfiles("dev", "test")) {
            Msg msg = facePay.verify();
            if (msg.hasError()) {
                return new Message<>(msg);
            }
        }

        JointProviderEnum jointProvider = facePay.getAppInfo().getProvider();

        @SuppressWarnings("unchecked")
        Msg<TradeFacePayResult> msg = this.getImplBean(facePay).facePay(facePay);

        if (msg.hasError()) {
            return msg;
        }

        if (environment.acceptsProfiles("dev", "test")) {
            TradeFacePayResult facePayResult = msg.getData();
            Asserts.notNull(facePayResult);
            Asserts.state(facePayResult.getJointProvider() == jointProvider, "支付参数与返回的服务不一致");
            facePayResult.verifyThrow();
        }

        return msg;
    }


    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        applicationContext.getBeansOfType(JointTradeFacePayService.class).forEach((beanName, bean) -> {
            if (bean != this) {
                this.facePayServiceBeans.put(beanName, bean);
            }
        });
    }

    private JointTradeFacePayService getImplBean(TradeFacePay facePay) {
        JointProviderEnum jointProvider = Objects.requireNonNull(facePay.getAppInfo().getProvider());

        for (Map.Entry<String, JointTradeFacePayService> entry : this.facePayServiceBeans.entrySet()) {
            if (StringUtils.startsWithIgnoreCase(entry.getKey(), jointProvider.toString().replace("_", ""))) {
                return entry.getValue();
            }
        }

        JointTradeFacePayService facePayService = this.facePayServiceBeans.get(jointProvider.toString().toLowerCase());

        if (facePayService != null) {
            return facePayService;
        }

        facePayService = this.facePayServiceBeans.get(jointProvider.toString());

        if (facePayService != null) {
            return facePayService;
        }

        throw new UnsupportedOperationException("抱歉,系统暂不支持：" + jointProvider.getDescr());
    }


}