啥? 满屏 If-Else，试试 策略模式 KO 这些 “头皮屑“

啥? 满屏 If-Else，试试 策略模式 KO 这些 “头皮屑”

使用场景

• 不同项目组对接，真实和MOCK 接口切换
• 支付策略 微信，支付宝，银联
• Seata 事务信息存储 支持 file/db/redis 策略选择
• ShardingSphere 支持 内存/单机/集群 模式 管理元数据 不同策略
• and so on …

编码环境

plugins {
    id 'java'
    id 'org.springframework.boot' version '2.6.13'
    id 'io.spring.dependency-management' version '1.1.0'
}

group = 'com.lab'
version = '0.0.1-SNAPSHOT'
sourceCompatibility = '17'

configurations {
    compileOnly {
        extendsFrom annotationProcessor
    }
}

repositories {
    mavenLocal()
    maven { url 'https://maven.aliyun.com/repository/public' }
    maven { url 'https://maven.aliyun.com/repository/spring' }
    mavenCentral()
}

dependencies {
    implementation 'org.springframework.boot:spring-boot-starter-web'

    implementation 'com.alibaba.cola:cola-component-domain-starter:4.3.1'
    implementation 'com.alibaba.cola:cola-component-extension-starter:4.3.1'

    annotationProcessor 'org.springframework.boot:spring-boot-configuration-processor'

    compileOnly 'org.projectlombok:lombok'
    annotationProcessor 'org.projectlombok:lombok'

    testCompileOnly 'org.projectlombok:lombok'
    testAnnotationProcessor 'org.projectlombok:lombok'

    testImplementation 'org.springframework.boot:spring-boot-starter-test'
}

tasks.named('test') {
    useJUnitPlatform()
}

Round one 手工打造

业务场景

参谋策划了一些计谋，供将军指挥杀敌

杀敌策略接口

package com.lab.strategy.fight;

public interface FightService {
    void fightEnemy();

    FightEnum fightEnum();
}

参谋提供的杀敌计谋

package com.lab.strategy.fight;

import lombok.AllArgsConstructor;
import lombok.Getter;

@Getter
@AllArgsConstructor
public enum FightEnum {
    HONEY_TRAP("美人计"),
    COLLATERAL("借刀杀人"),
    SELF_INJURY("苦肉计");

    private final String desc;
    private final String label = "杀敌之法";
}

参谋提供的每个计谋具体细节

• 美人计

package com.lab.strategy.fight;

import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;

@Slf4j
@Service
public class HoneyTrapServiceImpl implements FightService {
    @Override
    public void fightEnemy() {
        log.info("使用‘美人计’取得胜利");
    }

    @Override
    public FightEnum fightEnum() {
        return FightEnum.HONEY_TRAP;
    }
}

• 借刀杀人

package com.lab.strategy.fight;

import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;

@Slf4j
@Service
public class CollateralServiceImpl implements FightService {
    @Override
    public void fightEnemy() {
        log.info("使用’借刀杀人‘取得胜利");
    }

    @Override
    public FightEnum fightEnum() {
        return FightEnum.COLLATERAL;
    }
}

• 苦肉计

package com.lab.strategy.fight;

import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;

@Slf4j
@Service
public class SelfInjuryServiceImpl implements FightService {
    @Override
    public void fightEnemy() {
        log.info("使用’苦肉计‘取得胜利");
    }

    @Override
    public FightEnum fightEnum() {
        return FightEnum.SELF_INJURY;
    }
}

收集所有计谋，随时使用

package com.lab.strategy.fight;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.util.List;
import java.util.Map;
import java.util.function.Function;
import java.util.stream.Collectors;

@Component
public class FightFactory {
    private static Map<FightEnum, FightService> serviceMap;

    @Autowired
    private FightFactory(List<FightService> fightServiceList) {
        FightFactory.serviceMap = fightServiceList.stream()
                .collect(Collectors.toMap(FightService::fightEnum, Function.identity()));
    }

    public static FightService getFightService(FightEnum fightEnum) {
        return serviceMap.get(fightEnum);
    }
}

将军使用计谋杀敌

@DisplayName("测试-枚举工厂策略模式")
@Test
void testFight() {
    FightService fightService = FightFactory.getFightService(FightEnum.HONEY_TRAP);
    fightService.fightEnemy();

    fightService = FightFactory.getFightService(FightEnum.SELF_INJURY);
    fightService.fightEnemy();

    fightService = FightFactory.getFightService(FightEnum.COLLATERAL);
    fightService.fightEnemy();
}

Round two COLA扩展组件

业务场景

Seata 注册服务，提供了 Nacos/Etcd3/Consul 供您选择

定义服务注册类型

package com.lab.strategy.seata;

public class RegistryType {
    public static final String NACOS = "Nacos";
    public static final String ETCD3 = "Etcd3";
}

注册服务提供参数信息

package com.lab.strategy.seata;

import lombok.Data;
import lombok.NoArgsConstructor;
import lombok.experimental.Accessors;

@Accessors(chain = true)
@Data
@NoArgsConstructor
public class RegistryCmd {
    private String serverAddr;
    private String username;
    private String password;
}

服务注册 接口

接口名必须包含 ExtPt，否则报错

package com.lab.strategy.seata;

import com.alibaba.cola.extension.ExtensionPointI;

public interface SeataRegistryExtPtService extends ExtensionPointI {
    String registry(RegistryCmd registryCmd);
}

选择 Nacos 服务注册

package com.lab.strategy.seata;

import com.alibaba.cola.extension.Extension;
import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;

@Extension(bizId = "lab", useCase = "seata", scenario = RegistryType.NACOS)
@Slf4j
public class SeataNacosRegistryServiceImpl implements SeataRegistryExtPtService {
    @SneakyThrows
    @Override
    public String registry(RegistryCmd registryCmd) {
        log.info("选择 Nacos 作为 Seata 注册中心");
        return new ObjectMapper().writeValueAsString(registryCmd);
    }
}

选择 Etcd3 服务注册

package com.lab.strategy.seata;

import com.alibaba.cola.extension.Extension;
import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;

@Extension(bizId = "lab", useCase = "seata", scenario = RegistryType.ETCD3)
@Slf4j
public class SeataEtcd3RegistryServiceImpl implements SeataRegistryExtPtService {
    @SneakyThrows
    @Override
    public String registry(RegistryCmd registryCmd) {
        log.info("选择 Etcd3 作为 Seata 注册中心");
        return new ObjectMapper().writeValueAsString(registryCmd);
    }
}

开始服务注册

@DisplayName("测试-COLA扩展组件")
@Test
void testCOLAExtension() {
    ExtensionExecutor executor = DomainFactory.create(ExtensionExecutor.class);

    RegistryCmd registryCmd = new RegistryCmd();
    registryCmd.setServerAddr("192.168.1.168:8848").setUsername("nacos").setPassword("nacos");

    BizScenario bizScenario = BizScenario.valueOf("lab", "seata", RegistryType.NACOS);
    String ret = executor.execute(SeataRegistryExtPtService.class, bizScenario, ex -> ex.registry(registryCmd));
    log.info(ret);

    registryCmd.setServerAddr("http://192.168.1.169:2379").setUsername("root").setPassword("root");
    bizScenario = BizScenario.valueOf("lab", "seata", RegistryType.ETCD3);
    ret = executor.execute(SeataRegistryExtPtService.class, bizScenario, ex -> ex.registry(registryCmd));
    log.info(ret);
}

组件扫描

@SpringBootApplication(scanBasePackages = {"com.alibaba.cola", "com.lab"})