介绍#

策略模式是一种行为型设计模式。

在策略模式定义了一系列算法或策略，并将每个算法封装在独立的类中，使得它们可以互相替换。通过使用策略模式，可以在运行时根据需要选择不同的算法，而不需要修改客户端代码。

在策略模式中，我们创建表示各种策略的对象和一个行为随着策略对象改变而改变的 context 对象。策略对象改变 context 对象的执行算法。

策略模式平常我们多用来消除if-else switch等多重判断的代码，可以有效地应对代码复杂性。

下述代码对应的业务，根据对应的优惠类型，对价格作出相应的优惠。

1
package cn.meowrain;
2

3
import java.util.Objects;
4

5
public class DiscountTest {
6
    public static void main(String[] args) {
7
        Double result = DiscountTest.discount("1", 100.00);
8
        System.out.println(result);
9
    }
10

11
    public static Double discount(String type, Double price) {
12
        if (Objects.equals(type, "1")) {
13
            return price * 0.8;
14
        } else if (Objects.equals(type, "2")) {
15
            return price * 0.6;
16
        } else if (Objects.equals(type, "3")) {
17
            return price * 0.5;
18
        } else {
19
            return price;
20
        }
21
    }
22
}

但是我们很快就能发现问题了，这还是个案例，if else代码块就这么多了，真实的业务会多少if else可想而知了

我们可以应用策略模式解决这个问题： 1.将不同的优惠类型定义为不同的策略算法实现类。 2. 保证开闭原则，增加程序的健壮性以及可扩展性。

1
package cn.meowrain;
2

3
import java.util.HashMap;
4
import java.util.Map;
5

6
interface DiscountStrategy {
7
    Double discount(Double price);
8
}
9

10
/**
11
 * Implements 80% discount logic.
12
 */
13
class Discount80Strategy implements DiscountStrategy {
14
    static {
15
        DiscountStrategyFactory.registry("1", new Discount80Strategy());
16
    }
17

18
    @Override
19
    public Double discount(Double price) {
20
        return price * 0.8; // Bug 1: Incorrect operator `_0.8`
21
    }
22
}
23

24
/**
25
 * Implements 60% discount logic.
26
 */
27
class Discount60Strategy implements DiscountStrategy {
28
    static {
29
        DiscountStrategyFactory.registry("2", new Discount60Strategy());
30
    }
31

32
    @Override
33
    public Double discount(Double price) {
34
        return price * 0.6; // Bug 2: Incorrect operator `_ 0.6`
35
    }
36
}
37

38
/**
39
 * Implements 50% discount logic.
40
 */
41
class Discount50Strategy implements DiscountStrategy {
42
    static {
43
        DiscountStrategyFactory.registry("3", new Discount50Strategy());
44
    }
45

46
    @Override
47
    public Double discount(Double price) {
48
        return price * 0.5;
49
    }
50
}
51

52
class DiscountStrategyFactory {
53
    private static final Map<String, DiscountStrategy> strategyMap = new HashMap<>();
54

55
    public static void registry(String type, DiscountStrategy strategy) {
56
        strategyMap.put(type, strategy);
57
    }
58

59
    public static DiscountStrategy getStrategy(String type) {
60
        return strategyMap.get(type);
61
    }
62
}
63

64
public class DiscountTest2 {
65
    public static void main(String[] args) {
66
        new Discount80Strategy();
67
        new Discount60Strategy();
68
        new Discount50Strategy();
69
        Double result1 = DiscountStrategyFactory.getStrategy("1").discount(100.00);
70
        System.out.println("80% Discount: " + result1);
71

72
        Double result2 = DiscountStrategyFactory.getStrategy("2").discount(100.00);
73
        System.out.println("60% Discount: " + result2);
74
    }
75
}

使用spring

1
package org.example.discount;
2

3
import org.springframework.stereotype.Component;
4

5
@Component
6
public class Discount90Strategy implements DiscountStrategy {
7
    @Override
8
    public Double discount(Double price) {
9
        return price * 0.9;
10
    }
11

12
    @Override
13
    public String mark() {
14
        return "1";
15
    }
16
}

1
package org.example.discount;
2

3
import org.springframework.stereotype.Component;
4

5
@Component
6
public class Discount80Strategy implements DiscountStrategy {
7
    @Override
8
    public Double discount( Double price) {
9
        return price * 0.8;
10
    }
11

12
    @Override
13
    public String mark() {
14
        return "2";
15
    }
16
}

1
package org.example.discount;
2

3
import org.springframework.stereotype.Component;
4

5
@Component
6
public class Discount50Strategy implements DiscountStrategy {
7
    @Override
8
    public Double discount(Double price) {
9
        return price * 0.5;
10
    }
11

12
    @Override
13
    public String mark() {
14
        return "3";
15
    }
16
}

1
package org.example.discount;
2

3
import org.springframework.beans.factory.InitializingBean;
4
import org.springframework.beans.factory.annotation.Autowired;
5
import org.springframework.context.ApplicationContext;
6
import org.springframework.stereotype.Component;
7

8
import java.util.HashMap;
9
import java.util.Map;
10

11
@Component
12
public class DiscountFactory implements InitializingBean {
13
    @Autowired
14
    private ApplicationContext context;
15

16

17
    private final Map<String, DiscountStrategy> discountStrategies = new HashMap<>();
18

19
    public DiscountStrategy chooseStrategy(String type) {
20
        return discountStrategies.get(type);
21
    }
22

23
    @Override
24
    public void afterPropertiesSet() throws Exception {
25
        Map<String, DiscountStrategy> beans = context.getBeansOfType(DiscountStrategy.class);
26
        beans.forEach((k, v) -> discountStrategies.put(v.mark(), v));
27
    }
28
}

1
package org.example.discount;
2

3
public interface DiscountStrategy {
4
    Double discount(Double price);
5

6
    String mark();
7
}

1
package org.example;
2

3
import org.example.config.AppConfig;
4
import org.example.discount.DiscountFactory;
5
import org.example.discount.DiscountStrategy;
6
import lombok.extern.slf4j.Slf4j;
7

8
import org.junit.Test;
9
import org.junit.runner.RunWith;
10
import org.springframework.beans.factory.annotation.Autowired;
11
import org.springframework.boot.test.context.SpringBootTest;
12

13
import org.springframework.test.context.junit4.SpringRunner;
14

15
@RunWith(SpringRunner.class) // Add this annotation
16
@SpringBootTest(classes = AppConfig.class)
17
@Slf4j
18
public class DiscountTest {
19
    @Autowired
20
    private DiscountFactory discountFactory;
21

22
    @Test
23
    public void test() {
24
        DiscountStrategy strategy = discountFactory.chooseStrategy("2");
25
        Double result = strategy.discount(1000.0);
26
        log.info(String.valueOf(result));
27
    }
28
}

1
<?xml version="1.0" encoding="UTF-8"?>
2
<project xmlns="http://maven.apache.org/POM/4.0.0"
3
         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
4
         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
5
    <modelVersion>4.0.0</modelVersion>
6

7
    <parent>
8
        <groupId>org.springframework.boot</groupId>
9
        <artifactId>spring-boot-starter-parent</artifactId>
10
        <version>3.2.5</version>
11
        <relativePath/>
12
    </parent>
13

14
    <groupId>org.example</groupId>
15
    <artifactId>learn_juc</artifactId>
16
    <version>1.0-SNAPSHOT</version>
17

18
    <properties>
19
        <java.version>17</java.version>
20
        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
21
    </properties>
22

23
    <dependencies>
24
        <dependency>
25
            <groupId>org.springframework.boot</groupId>
26
            <artifactId>spring-boot-starter</artifactId>
27
        </dependency>
28
        <dependency>
29
            <groupId>org.slf4j</groupId>
30
            <artifactId>slf4j-api</artifactId>
31
        </dependency>
32
        <dependency>
33
            <groupId>ch.qos.logback</groupId>
34
            <artifactId>logback-classic</artifactId>
35
        </dependency>
36

37
        <dependency>
38
            <groupId>org.projectlombok</groupId>
39
            <artifactId>lombok</artifactId>
40
            <optional>true</optional>
41
        </dependency>
42

43
        <dependency>
44
            <groupId>org.springframework.boot</groupId>
45
            <artifactId>spring-boot-starter-test</artifactId>
46
            <scope>test</scope>
47
        </dependency>
48
        <dependency>
49
            <groupId>junit</groupId>
50
            <artifactId>junit</artifactId>
51
            <scope>test</scope>
52
        </dependency>
53
    </dependencies>
54

55
    <build>
56
        <plugins>
57
            <plugin>
58
                <groupId>org.springframework.boot</groupId>
59
                <artifactId>spring-boot-maven-plugin</artifactId>
60
                <configuration>
61
                    <excludes>
62
                        <exclude>
63
                            <groupId>org.projectlombok</groupId>
64
                            <artifactId>lombok</artifactId>
65
                        </exclude>
66
                    </excludes>
67
                </configuration>
68
            </plugin>
69
        </plugins>
70
    </build>
71

72
</project>

优点#

提高灵活性和可维护性：通过将算法的实现与使用分离开来，当需要修改或添加新算法时，只需定义新的策略类并将其传递给环境类即可，无需修改环境类代码。

提高代码复用性：算法被封装在独立的策略类中，使得这些算法可以被多个不同的客户（环境类）复用。

动态切换算法：允许在程序运行时根据需要动态地改变和选择算法，从而实现不同的功能和行为，使程序更灵活。

算法实现与使用分离使代码更清晰：客户端代码仅需关注如何选择和使用不同的算法，而不必关心算法的具体实现细节，使代码更简洁、易于理解和扩展。

避免大量条件语句：当需要根据不同条件选择不同算法时，策略模式可以避免使用复杂的 if-else 或 switch 语句，使代码结构更清晰，更易于维护。