条件付きのロジックをTempleteMethodパターンで置換する

コードが複雑になる要因の一つとしてif文やswitch文による条件付きロジックがあります。以下のような給与を計算する実装例を考えてみます。(employeeType が文字列、という話はおいておきます)

switch 文があると呼び出し側の実装が複雑になります。switch 文を呼び出し元の実装からリファクタリングします。

• main.go

package main

import "fmt"

func main() {
    employeeType := "engineer"

    var amount int
    switch employeeType {
    case "engineer":
        amount = 1
    case "salesman":
        amount = 10
    default:
        amount = -1
    }

    fmt.Println("amount", amount)
}

TemplateMethodパターン

条件付きのロジックをSteragyパターンで置換する の記事ではStrategyパターンを使って呼び出し側の構造体にアルゴリズムを実装することができました。今回はTemplateMethodパターンで実装します。

Strategyパターンと本質的には同様で、TemplateMethodパターンでは抽象クラスやインターフェースを使って実装します。呼び出し元の main.go はStrategyパターンと変更はありません。 - model.go

package main

type Employee interface {
    payAmount() int
}

func NewEmployee(employeeType string) Employee {
    switch employeeType {
    case "engineer":
        return Engineer{}
    case "salesman":
        return Salesman{}
    default:
        return dummy{}
    }
}

type Engineer struct{}

func (e Engineer) payAmount() int {
    return 1
}

type Salesman struct{}

func (s Salesman) payAmount() int {
    return 10
}

type dummy struct{}

func (d dummy) payAmount() int {
    return -1
}

• main.go

package main

import "fmt"

func main() {
    e := NewEmployee("engineer")
    fmt.Println("amount", e.payAmount())
}

• 出力結果

$ go run .
amount 1

呼び出し元からは Employee インターフェースのメソッドが呼び出されます。実装の詳細は Employee インターフェースを実装している各構造体のメソッドです。本例では Engineer 構造体のメソッドとなります。