SpringBoot中实现简单策略模式

2023-08-06 09:22| 来源: 网络整理| 查看: 265

策略模式（Strategy Pattern）:一个类的行为或其算法可以在运行时更改,策略模式属于行为型模式。

策略模式简单Java实现: https://www.runoob.com/design-pattern/strategy-pattern.html

策略模式本质就是利用类型的多态,通过接口不同实现类来实现不同的功能

SpringBoot实现

场景:假设登录系统的用户有多种,例如:终身会员、年度会员、月度会员、普通用户,每种用户购买商品折扣不同

创建用户类型枚举 @Getter @AllArgsConstructor public enum UserTypeEnum { /** * 终身会员 * 年度会员 * 月度会员 * 普通用户 */ LIFETIME_MEMBER(2, BigDecimal.valueOf(0.2), "终身会员"), ANNUAL_MEMBER(3, BigDecimal.valueOf(0.5), "年度会员"), MONTHLY_MEMBER(4, BigDecimal.valueOf(0.75), "月度会员"), USER(5, BigDecimal.valueOf(1), "普通用户"); /** 用户类型code */ private final int code; /** 用户折扣 */ private final BigDecimal discount; /** 用户类型 */ private final String desc; } 创建用户Service接口

UserService接口定义了default方法,子类可以选择重写,不重写则使用默认方法

例如默认类型为:普通用户

默认购买方法为:购买金额 =金额 × 折扣

public interface UserService { /** * 获取用户类型 * @return 用户类型枚举 */ default UserTypeEnum getUserType() { return UserTypeEnum.USER; } /** * 默认扣除金额的方式 * 金额 × 折扣 * @param price 金额 * @return 折后金额 */ default BigDecimal calcPrice(String price) { return new BigDecimal(price).multiply(getUserType().getDiscount()); } }

创建实现类

重写getUserType()方法

LifetimeMemberUserServiceImpl

AnnualMemberUserServiceImpl

MonthlyMemberUserServiceImpl

不重写默认方法

NoramlUserServiceImpl

@Service public class AnnualMemberUserServiceImpl implements UserService { @Override public UserTypeEnum getUserType() { return UserTypeEnum.ANNUAL_MEMBER; } } 创建ShoppingService调用UserService

采用构造器注入,通过SpringBoot自动注入将所有类型为UserService的实现类注入进来并收集成一个key为用户类型,value为UserService的Map,此时通过当前登录用户的userType决定调用哪个具体UserService实现类,达到去除if else的目的

@Service public class ShoppingServiceImpl implements ShoppingService { @Resource private LoginService loginService; private final Map userServiceMap; public ShoppingServiceImpl(List userServices) { userServiceMap = userServices.stream() .collect(Collectors.toMap(UserService::getUserType, userService -> userService)); } @Override public String calcPrice(String price) { // 获取当前登录用户 User loginUser = loginService.getLoginUser(); // 主要业务流程 try { // 计算折扣 UserService userService = userServiceMap.get(loginUser.getUserType()); BigDecimal calcResult = userService.calcPrice(price); String result = calcResult.toPlainString(); // 假设扣除用户余额 System.out.println("用户扣除余额 = " + result); } catch (Exception e) { return "error"; } return "success"; } } 创建单元测试类或接口进行测试 @SpringBootTest class Springboot3ApplicationTests { @Resource private ShoppingService shoppingService; @Test void contextLoads() { String result = shoppingService.calcPrice("56"); } }

【本文地址】

公司简介

联系我们