Стратегия на Golang

Паттерны проектирования / Поведенческие паттерны

import (
    "fmt"
)

//Strategy with integer operation
type Strategy interface {
    DoOperation(a int, b int) int
}

//AddStrategy is ConcreteStrategy
type AddStrategy struct{}

//DoOperation is integer addition function
func (s AddStrategy) DoOperation(a int, b int) int {
    return a + b
}

//SubtractStrategy is ConcreteStrategy
type SubtractStrategy struct{}

//DoOperation is integer subtraction function
func (s SubtractStrategy) DoOperation(a int, b int) int {
    return a - b
}

//Calc is Context
type Calc struct {
    _strategy Strategy
}

//Execute current strategy
func (c Calc) Execute(a int, b int) int {
    if c._strategy == nil {
        return 0
    }

    return c._strategy.DoOperation(a, b)
}

//SetStrategy changes the current strategy
func (c *Calc) SetStrategy(strategy Strategy) {
    c._strategy = strategy
}

//Client
calc := Calc{}
result1 := calc.Execute(5, 3)
//result1 is 0

calc.SetStrategy(AddStrategy{})
result2 := calc.Execute(5, 3)
//result2 is 8

calc.SetStrategy(SubtractStrategy{})
result3 := calc.Execute(5, 3)
//result3 is 2

fmt.Println(result1)
fmt.Println(result2)
fmt.Println(result3)

Определяет семейство алгоритмов, инкапсулируя их и позволяя подставлять один вместо другого. Можно менять алгоритм независимо от клиента, который им пользуется.

Задача. Магазин считает стоимость доставки: по весу, по расстоянию или по фиксированной цене. Если один метод держит все способы в цепочке условий, он растёт с каждым новым способом. Правка одного способа может сломать другие, и ни один способ нельзя проверить отдельно.

Идея. Каждый способ сделать работу становится отдельным объектом - стратегией. У всех стратегий одинаковый вид: те же входные данные и тот же тип результата. Объект, которому нужна эта работа, называют контекстом. Он хранит одну стратегию и вызывает её, не зная, какая именно ему досталась. Стратегию выбирает клиентский код и может заменить её в любой момент.

Как читать пример. Каждая конкретная стратегия - это один алгоритм, и все они принимают одинаковые входные данные. Клиентский код выбирает стратегию. Код, который пользуется стратегией, только вызывает её и не знает, какую получил. В большинстве примеров одни и те же два числа дают другой результат после каждой смены стратегии. Большинство из них отвечают и на случай, когда стратегии нет. В некоторых языках стратегия - просто функция.

Когда применять. Стратегия нужна, когда одну работу можно сделать несколькими способами, а выбор зависит от пользователя, данных или настроек. Для одного алгоритма, который не будет меняться, она ни к чему. Стратегия похожа на Состояние, но разница в том, кто делает выбор. Здесь алгоритм выбирает снаружи клиентский код, и стратегии ничего не знают друг о друге. В Состоянии объект сам переходит из состояния в состояние.

Частая ошибка. Заставить контекст проверять, какая стратегия у него в руках. Условия возвращаются, и паттерн теряет смысл. Контекст должен только вызывать стратегию.