IO、並び替え

検索…

IO、並び替え

IO and sorting

​この章のすべてのコードはここにあります​
​前の章 新しいエンドポイント /leagueを追加することで、アプリケーションを繰り返し続けました。途中で、JSONの処理方法、埋め込みタイプ、ルーティングについて学習しました。
私たちの製品の所有者は、サーバーが再起動されたときにソフトウェアがスコアを失うことにいくらか混乱しています。これは、ストアの実装がインメモリであるためです。彼女はまた、「/league」エンドポイントが勝ちの数で順序付けられたプレーヤーを返す必要があると解釈しなかったことに不満を持っています！
これまでのコード
1
// server.go
2
package main
3
​
4
import (
5
    "encoding/json"
6
    "fmt"
7
    "net/http"
8
)
9
​
10
// PlayerStore stores score information about players
11
type PlayerStore interface {
12
    GetPlayerScore(name string) int
13
    RecordWin(name string)
14
    GetLeague() []Player
15
}
16
​
17
// Player stores a name with a number of wins
18
type Player struct {
19
    Name string
20
    Wins int
21
}
22
​
23
// PlayerServer is a HTTP interface for player information
24
type PlayerServer struct {
25
    store PlayerStore
26
    http.Handler
27
}
28
​
29
const jsonContentType = "application/json"
30
​
31
// NewPlayerServer creates a PlayerServer with routing configured
32
func NewPlayerServer(store PlayerStore) *PlayerServer {
33
    p := new(PlayerServer)
34
​
35
    p.store = store
36
​
37
    router := http.NewServeMux()
38
    router.Handle("/league", http.HandlerFunc(p.leagueHandler))
39
    router.Handle("/players/", http.HandlerFunc(p.playersHandler))
40
​
41
    p.Handler = router
42
​
43
    return p
44
}
45
​
46
func (p *PlayerServer) leagueHandler(w http.ResponseWriter, r *http.Request) {
47
    w.Header().Set("content-type", jsonContentType)
48
    json.NewEncoder(w).Encode(p.store.GetLeague())
49
}
50
​
51
func (p *PlayerServer) playersHandler(w http.ResponseWriter, r *http.Request) {
52
    player := r.URL.Path[len("/players/"):]
53
​
54
    switch r.Method {
55
    case http.MethodPost:
56
        p.processWin(w, player)
57
    case http.MethodGet:
58
        p.showScore(w, player)
59
    }
60
}
61
​
62
func (p *PlayerServer) showScore(w http.ResponseWriter, player string) {
63
    score := p.store.GetPlayerScore(player)
64
​
65
    if score == 0 {
66
        w.WriteHeader(http.StatusNotFound)
67
    }
68
​
69
    fmt.Fprint(w, score)
70
}
71
​
72
func (p *PlayerServer) processWin(w http.ResponseWriter, player string) {
73
    p.store.RecordWin(player)
74
    w.WriteHeader(http.StatusAccepted)
75
}
Copied!
1
// InMemoryPlayerStore.go
2
package main
3
​
4
func NewInMemoryPlayerStore() *InMemoryPlayerStore {
5
    return &InMemoryPlayerStore{map[string]int{}}
6
}
7
​
8
type InMemoryPlayerStore struct {
9
    store map[string]int
10
}
11
​
12
func (i *InMemoryPlayerStore) GetLeague() []Player {
13
    var league []Player
14
    for name, wins := range i.store {
15
        league = append(league, Player{name, wins})
16
    }
17
    return league
18
}
19
​
20
func (i *InMemoryPlayerStore) RecordWin(name string) {
21
    i.store[name]++
22
}
23
​
24
func (i *InMemoryPlayerStore) GetPlayerScore(name string) int {
25
    return i.store[name]
26
}
Copied!
1
// main.go
2
package main
3
​
4
import (
5
    "log"
6
    "net/http"
7
)
8
​
9
func main() {
10
    server := NewPlayerServer(NewInMemoryPlayerStore())
11
​
12
    if err := http.ListenAndServe(":5000", server); err != nil {
13
        log.Fatalf("could not listen on port 5000 %v", err)
14
    }
15
}
Copied!
章の上部にあるリンクで対応するテストを見つけることができます。
データを保存する
これを使用できるデータベースは数十ありますが、ここでは非常にシンプルなアプローチを採用します。このアプリケーションのデータをJSONとしてファイルに保存します。
これにより、データの移植性が非常に高くなり、実装が比較的簡単になります。
特にうまくスケーリングしませんが、これはプロトタイプなので、今のところは問題ありません。私たちの状況が変化し、それが適切でなくなった場合、私たちが使用したPlayerStore抽象化のため、それを別のものと交換するのは簡単です。
新しいストアを開発するときに統合テストに合格し続けるために、今のところInMemoryPlayerStoreを保持します。新しい実装が統合テストに合格するのに十分であると確信したら、それを入れ替えてから、InMemoryPlayerStoreを削除します。
最初にテストを書く
これで、データの読み取り（io.Reader）、データの書き込み（io.Writer）、および標準ライブラリを使用してこれらの関数をテストすることなく標準ライブラリをテストする方法について、標準ライブラリに関連するインターフェースに精通しているはずです。実際のファイルを使用する必要があります。
この作業を完了するには、 PlayerStoreを実装する必要があるため、実装する必要のあるメソッドを呼び出すストアのテストを記述します。GetLeagueから始めます。
1
func TestFileSystemStore(t *testing.T) {
2
​
3
    t.Run("/league from a reader", func(t *testing.T) {
4
        database := strings.NewReader(`[
5
            {"Name": "Cleo", "Wins": 10},
6
            {"Name": "Chris", "Wins": 33}]`)
7
​
8
        store := FileSystemPlayerStore{database}
9
​
10
        got := store.GetLeague()
11
​
12
        want := []Player{
13
            {"Cleo", 10},
14
            {"Chris", 33},
15
        }
16
​
17
        assertLeague(t, got, want)
18
    })
19
}
Copied!
私たちはReaderを返すstrings.NewReaderを使用しています。 これは、FileSystemPlayerStoreがデータを読み取るために使用するものです。mainでは、Readerでもあるファイルを開きます。
テストを実行してみます
1
# github.com/quii/learn-go-with-tests/json-and-io/v7
2
./FileSystemStore_test.go:15:12: undefined: FileSystemPlayerStore
Copied!
テストを実行するための最小限のコードを記述し、失敗したテスト出力を確認します
新しいファイルでFileSystemPlayerStoreを定義しましょう
1
type FileSystemPlayerStore struct {}
Copied!
再試行
1
# github.com/quii/learn-go-with-tests/json-and-io/v7
2
./FileSystemStore_test.go:15:28: too many values in struct initializer
3
./FileSystemStore_test.go:17:15: store.GetLeague undefined (type FileSystemPlayerStore has no field or method GetLeague)
Copied!
Readerを渡していますが、予期しておらず、まだGetLeagueが定義されていないため、問題があります。
1
type FileSystemPlayerStore struct {
2
    database io.Reader
3
}
4
​
5
func (f *FileSystemPlayerStore) GetLeague() []Player {
6
    return nil
7
}
Copied!
もう一回試してみる...
1
=== RUN   TestFileSystemStore//league_from_a_reader
2
    --- FAIL: TestFileSystemStore//league_from_a_reader (0.00s)
3
        FileSystemStore_test.go:24: got [] want [{Cleo 10} {Chris 33}]
Copied!
成功させるのに十分なコードを書く
以前にリーダーからJSONを読み取りました
1
func (f *FileSystemPlayerStore) GetLeague() []Player {
2
    var league []Player
3
    json.NewDecoder(f.database).Decode(&league)
4
    return league
5
}
Copied!
テストは成功するはずです。
Refactor
これは以前に行ったことです！ サーバーのテストコードは、応答からJSONをデコードする必要がありました。
この関数をDRYにしてみましょう。
league.goと呼ばれる新しいファイルを作成し、これを中に入れます。
1
func NewLeague(rdr io.Reader) ([]Player, error) {
2
    var league []Player
3
    err := json.NewDecoder(rdr).Decode(&league)
4
    if err != nil {
5
        err = fmt.Errorf("problem parsing league, %v", err)
6
    }
7
​
8
    return league, err
9
}
Copied!
これを実装と、server_test.goのテストヘルパーgetLeagueFromResponseで呼び出します
1
func (f *FileSystemPlayerStore) GetLeague() []Player {
2
    league, _ := NewLeague(f.database)
3
    return league
4
}
Copied!
構文解析エラーに対処するための戦略はまだありませんが、続けましょう。
問題を探す
私たちの実装に欠陥があります。まず、io.Readerの定義を思い出してみましょう。
1
type Reader interface {
2
    Read(p []byte) (n int, err error)
3
}
Copied!
私たちのファイルを使用すると、最後までバイト単位で読み取ることを想像できます。 もう一度「読み取りRead」しようとするとどうなりますか？
現在のテストの最後に以下を追加します。
1
// read again
2
got = store.GetLeague()
3
assertLeague(t, got, want)
Copied!
これは成功させたいのですが、テストを実行すると成功しません。
問題は、Readerが最後に到達したため、これ以上読むものがないことです。最初に戻るように指示する方法が必要です。
​ReadSeekerは、標準ライブラリにあるもう1つのインターフェースで、役立ちます。
1
type ReadSeeker interface {
2
    Reader
3
    Seeker
4
}
Copied!
埋め込みを覚えていますか？これは、Readerと[Seeker]（https://golang.org/pkg/io/#Seeker）で構成されるインターフェースです。​
1
type Seeker interface {
2
    Seek(offset int64, whence int) (int64, error)
3
}
Copied!
これはいいですね、代わりにこのインターフェイスを取るようにFileSystemPlayerStoreを変更できますか？
1
type FileSystemPlayerStore struct {
2
    database io.ReadSeeker
3
}
4
​
5
func (f *FileSystemPlayerStore) GetLeague() []Player {
6
    f.database.Seek(0, 0)
7
    league, _ := NewLeague(f.database)
8
    return league
9
}
Copied!
テストを実行してみてください。テストは成功しました！ テストで使用したstring.NewReaderは、ReadSeekerも実装しているため、他の変更を行う必要はありませんでした。
次に、GetPlayerScoreを実装します。
最初にテストを書く
1
t.Run("get player score", func(t *testing.T) {
2
    database := strings.NewReader(`[
3
        {"Name": "Cleo", "Wins": 10},
4
        {"Name": "Chris", "Wins": 33}]`)
5
​
6
    store := FileSystemPlayerStore{database}
7
​
8
    got := store.GetPlayerScore("Chris")
9
​
10
    want := 33
11
​
12
    if got != want {
13
        t.Errorf("got %d want %d", got, want)
14
    }
15
})
Copied!
テストを実行してみます
1
./FileSystemStore_test.go:38:15: store.GetPlayerScore undefined (type FileSystemPlayerStore has no field or method GetPlayerScore)
Copied!
テストを実行するための最小限のコードを記述し、失敗したテスト出力を確認します
テストをコンパイルするには、メソッドを新しい型に追加する必要があります。
1
func (f *FileSystemPlayerStore) GetPlayerScore(name string) int {
2
    return 0
3
}
Copied!
これでコンパイルされ、テストは失敗します
1
=== RUN   TestFileSystemStore/get_player_score
2
    --- FAIL: TestFileSystemStore//get_player_score (0.00s)
3
        FileSystemStore_test.go:43: got 0 want 33
Copied!
成功させるのに十分なコードを書く
リーグを繰り返してプレーヤーを見つけ、スコアを返すことができます。
1
func (f *FileSystemPlayerStore) GetPlayerScore(name string) int {
2
​
3
    var wins int
4
​
5
    for _, player := range f.GetLeague() {
6
        if player.Name == name {
7
            wins = player.Wins
8
            break
9
        }
10
    }
11
​
12
    return wins
13
}
Copied!
リファクタリング♪
何十ものテストヘルパーリファクタリングを確認したので、これを機能させるためにこれをあなたにお任せします。
1
t.Run("/get player score", func(t *testing.T) {
2
    database := strings.NewReader(`[
3
        {"Name": "Cleo", "Wins": 10},
4
        {"Name": "Chris", "Wins": 33}]`)
5
​
6
    store := FileSystemPlayerStore{database}
7
​
8
    got := store.GetPlayerScore("Chris")
9
    want := 33
10
    assertScoreEquals(t, got, want)
11
})
Copied!
最後に、RecordWinでスコアの記録を開始する必要があります。
最初にテストを書く
私たちのアプローチは、書き込みに対してかなり近視眼的です。ファイル内のJSONの「行」を1つだけ更新することは簡単にはできません。すべての書き込みで、データベースの whole 新しい表現を保存する必要があります。
どうやって書くの？通常はWriterを使用しますが、すでにReadSeekerがあります。潜在的に2つの依存関係が存在する可能性がありますが、標準ライブラリにはすでにReadWriteSeeker用のインターフェースがあり、ファイルで必要なすべてのことを実行できます。
タイプを更新しましょう
1
type FileSystemPlayerStore struct {
2
    database io.ReadWriteSeeker
3
}
Copied!
コンパイルされるかどうかを確認します
1
./FileSystemStore_test.go:15:34: cannot use database (type *strings.Reader) as type io.ReadWriteSeeker in field value:
2
    *strings.Reader does not implement io.ReadWriteSeeker (missing Write method)
3
./FileSystemStore_test.go:36:34: cannot use database (type *strings.Reader) as type io.ReadWriteSeeker in field value:
4
    *strings.Reader does not implement io.ReadWriteSeeker (missing Write method)
Copied!
strings.ReaderがReadWriteSeekerを実装しないことはそれほど驚くべきことではないので、何をしますか？
2つの選択肢があります
テストごとに一時ファイルを作成します。*os.FileはReadWriteSeekerを実装します。これの長所は、それが統合テストになり、実際にファイルシステムからの読み取りと書き込みを行っているため、非常に高い信頼性が得られることです。短所は、ユニットテストの方が速く、一般にシンプルであるためです。また、一時ファイルを作成し、テスト後にそれらが確実に削除されるように、さらに作業を行う必要があります。
サードパーティのライブラリを使用できます。Mattettiは、必要なインターフェイスを実装し、ファイルシステムに触れないライブラリfilebufferを作成しました。
ここでは特に間違った答えはないと思いますが、サードパーティのライブラリを使用することを選択することで、依存関係の管理について説明する必要があります。そのため、代わりにファイルを使用します。
テストを追加する前に、strings.Readerをos.Fileで置き換えることにより、他のテストをコンパイルする必要があります。
内部にいくつかのデータを含む一時ファイルを作成するヘルパー関数を作成しましょう
1
func createTempFile(t *testing.T, initialData string) (io.ReadWriteSeeker, func()) {
2
    t.Helper()
3
​
4
    tmpfile, err := ioutil.TempFile("", "db")
5
​
6
    if err != nil {
7
        t.Fatalf("could not create temp file %v", err)
8
    }
9
​
10
    tmpfile.Write([]byte(initialData))
11
​
12
    removeFile := func() {
13
        tmpfile.Close()
14
        os.Remove(tmpfile.Name())
15
    }
16
​
17
    return tmpfile, removeFile
18
}
Copied!
​TempFileは、使用する一時ファイルを作成します。渡した "db"値は、作成するランダムなファイル名に付けられるプレフィックスです。これは、誤って他のファイルと衝突しないようにするためです。
ReadWriteSeeker（ファイル）だけでなく、関数も返すことに気づくでしょう。 テストが終了したら、ファイルを確実に削除する必要があります。エラーが発生しやすく、読者の興味をそそる可能性があるため、ファイルの詳細をテストに漏らしたくない。removeFile関数を返すことで、ヘルパーの詳細を処理でき、呼び出し側が実行する必要があるのはdefer cleanDatabase()を実行することだけです。
1
func TestFileSystemStore(t *testing.T) {
2
​
3
    t.Run("league from a reader", func(t *testing.T) {
4
        database, cleanDatabase := createTempFile(t, `[
5
            {"Name": "Cleo", "Wins": 10},
6
            {"Name": "Chris", "Wins": 33}]`)
7
        defer cleanDatabase()
8
​
9
        store := FileSystemPlayerStore{database}
10
​
11
        got := store.GetLeague()
12
​
13
        want := []Player{
14
            {"Cleo", 10},
15
            {"Chris", 33},
16
        }
17
​
18
        assertLeague(t, got, want)
19
​
20
        // read again
21
        got = store.GetLeague()
22
        assertLeague(t, got, want)
23
    })
24
​
25
    t.Run("get player score", func(t *testing.T) {
26
        database, cleanDatabase := createTempFile(t, `[
27
            {"Name": "Cleo", "Wins": 10},
28
            {"Name": "Chris", "Wins": 33}]`)
29
        defer cleanDatabase()
30
​
31
        store := FileSystemPlayerStore{database}
32
​
33
        got := store.GetPlayerScore("Chris")
34
        want := 33
35
        assertScoreEquals(t, got, want)
36
    })
37
}
Copied!
テストを実行すると、テストに成功するはずです。かなりの量の変更がありましたが、インターフェースの定義が完了したように感じられ、これから新しいテストを簡単に追加できるはずです。
既存のプレイヤーの勝利を記録する最初の反復を取得しましょう
1
t.Run("store wins for existing players", func(t *testing.T) {
2
    database, cleanDatabase := createTempFile(t, `[
3
        {"Name": "Cleo", "Wins": 10},
4
        {"Name": "Chris", "Wins": 33}]`)
5
    defer cleanDatabase()
6
​
7
    store := FileSystemPlayerStore{database}
8
​
9
    store.RecordWin("Chris")
10
​
11
    got := store.GetPlayerScore("Chris")
12
    want := 34
13
    assertScoreEquals(t, got, want)
14
})
Copied!
テストを実行してみます
./FileSystemStore_test.go:67:8: store.RecordWin undefined (type FileSystemPlayerStore has no field or method RecordWin)
テストを実行するための最小限のコードを記述し、失敗したテスト出力を確認します
新しいメソッドを追加する
1
func (f *FileSystemPlayerStore) RecordWin(name string) {
2
​
3
}
Copied!
1
=== RUN   TestFileSystemStore/store_wins_for_existing_players
2
    --- FAIL: TestFileSystemStore/store_wins_for_existing_players (0.00s)
3
        FileSystemStore_test.go:71: got 33 want 34
Copied!
私たちの実装は空なので、古いスコアが返されます。
成功させるのに十分なコードを書く
1
func (f *FileSystemPlayerStore) RecordWin(name string) {
2
    league := f.GetLeague()
3
​
4
    for i, player := range league {
5
        if player.Name == name {
6
            league[i].Wins++
7
        }
8
    }
9
​
10
    f.database.Seek(0,0)
11
    json.NewEncoder(f.database).Encode(league)
12
}
Copied!
なぜ私が「player.Wins++」ではなく「league[i].Wins++」をしているのか、疑問に思われるかもしれません。
スライス上で「範囲range」を指定すると、ループの現在のインデックス（この場合はi）とそのインデックスにある要素のcopyが返されます。コピーのWins値を変更しても、繰り返し処理するleagueスライスには影響しません。そのため、league[i]を実行して実際の値への参照を取得し、代わりにその値を変更する必要があります。
テストを実行すると、テストは成功するはずです。
リファクタリング♪
GetPlayerScoreとRecordWinでは、名前でプレーヤーを見つけるために[] Playerを繰り返し処理しています。
FileSystemStoreの内部でこの共通コードをリファクタリングすることもできますが、私には、これが新しいタイプに引き上げることができるおそらく有用なコードであると感じています。これまで「リーグ"League"」での作業は常に[]Playerで行っていましたが、Leagueという新しいタイプを作成できます。これは、他の開発者が理解しやすくなり、そのタイプに便利なメソッドをアタッチして使用できるようになります。
league.go内に以下を追加します
1
type League []Player
2
​
3
func (l League) Find(name string) *Player {
4
    for i, p := range l {
5
        if p.Name==name {
6
            return &l[i]
7
        }
8
    }
9
    return nil
10
}
Copied!
誰かがLeagueを持っている場合、彼らは与えられたプレイヤーを簡単に見つけることができます。
PlayerStoreインターフェイスを変更して、[]PlayerではなくLeagueを返すようにします。テストを再実行してみてください。インターフェイスを変更したためコンパイルの問題が発生しますが、修正は非常に簡単です。戻り値の型を []PlayerからLeagueに変更するだけです。
これにより、FileSystemStoreのメソッドを簡略化できます。
1
func (f *FileSystemPlayerStore) GetPlayerScore(name string) int {
2
​
3
    player := f.GetLeague().Find(name)
4
​
5
    if player != nil {
6
        return player.Wins
7
    }
8
​
9
    return 0
10
}
11
​
12
func (f *FileSystemPlayerStore) RecordWin(name string) {
13
    league := f.GetLeague()
14
    player := league.Find(name)
15
​
16
    if player != nil {
17
        player.Wins++
18
    }
19
​
20
    f.database.Seek(0, 0)
21
    json.NewEncoder(f.database).Encode(league)
22
}
Copied!
これはかなり見栄えがよく、Leagueの他の便利な機能をリファクタリングできる方法を見つけることができます。
新しいプレーヤーの勝利を記録するシナリオを処理する必要があります。
最初にテストを書く
1
t.Run("store wins for new players", func(t *testing.T) {
2
    database, cleanDatabase := createTempFile(t, `[
3
        {"Name": "Cleo", "Wins": 10},
4
        {"Name": "Chris", "Wins": 33}]`)
5
    defer cleanDatabase()
6
​
7
    store := FileSystemPlayerStore{database}
8
​
9
    store.RecordWin("Pepper")
10
​
11
    got := store.GetPlayerScore("Pepper")
12
    want := 1
13
    assertScoreEquals(t, got, want)
14
})
Copied!
テストを実行してみます
1
=== RUN   TestFileSystemStore/store_wins_for_new_players#01
2
    --- FAIL: TestFileSystemStore/store_wins_for_new_players#01 (0.00s)
3
        FileSystemStore_test.go:86: got 0 want 1
Copied!
成功させるのに十分なコードを書く
プレーヤーが見つからなかったためにFindがnilを返すシナリオを処理する必要があるだけです。
1
func (f *FileSystemPlayerStore) RecordWin(name string) {
2
    league := f.GetLeague()
3
    player := league.Find(name)
4
​
5
    if player != nil {
6
        player.Wins++
7
    } else {
8
        league = append(league, Player{name, 1})
9
    }
10
​
11
    f.database.Seek(0, 0)
12
    json.NewEncoder(f.database).Encode(league)
13
}
Copied!
ハッピーパスは問題なく見えたので、統合テストで新しいStoreを使用してみることができます。これにより、ソフトウェアが動作するという確信が高まり、冗長なInMemoryPlayerStoreを削除できます。
TestRecordingWinsAndRetrievingThemで古いストアを置き換えます。
1
database, cleanDatabase := createTempFile(t, "")
2
defer cleanDatabase()
3
store := &FileSystemPlayerStore{database}
Copied!
テストを実行すると、テストに成功し、InMemoryPlayerStoreを削除できます。 main.goにコンパイルの問題が発生し、「実際の」コードで新しいストアを使用するようになります。
1
package main
2
​
3
import (
4
    "log"
5
    "net/http"
6
    "os"
7
)
8
​
9
const dbFileName = "game.db.json"
10
​
11
func main() {
12
    db, err := os.OpenFile(dbFileName, os.O_RDWR|os.O_CREATE, 0666)
13
​
14
    if err != nil {
15
        log.Fatalf("problem opening %s %v", dbFileName, err)
16
    }
17
​
18
    store := &FileSystemPlayerStore{db}
19
    server := NewPlayerServer(store)
20
​
21
    if err := http.ListenAndServe(":5000", server); err != nil {
22
        log.Fatalf("could not listen on port 5000 %v", err)
23
    }
24
}
Copied!
データベース用のファイルを作成します。
os.OpenFileの2番目の引数では、ファイルを開くための権限を定義できます。この場合、O_RDWRは読み取りと書き込みを行うことを意味し、os.O_CREATEはファイルが存在しない場合にファイルを作成することを意味します。
3番目の引数は、ファイルのアクセス権を設定することを意味します。この場合、すべてのユーザーがファイルの読み取りと書き込みを行うことができます。（詳細な説明はsuperuser.comを参照）.
プログラムを実行すると、再起動の間にデータがファイルに永続化されるようになりました。
より多くのリファクタリングとパフォーマンスの懸念
誰かがGetLeague()またはGetPlayerScore()を呼び出すたびに、ファイル全体を読み取ってJSONに解析します。FileSystemStoreはリーグの状態に完全に責任があるため、これを行う必要はありません。プログラムの起動時にファイルを読み取るだけで、データが変更されたときにファイルを更新するだけです。
この初期化の一部を実行できるコンストラクターを作成し、代わりに読み取りで使用するためにリーグをFileSystemStoreの値として保存できます。
1
type FileSystemPlayerStore struct {
2
    database io.ReadWriteSeeker
3
    league League
4
}
5
​
6
func NewFileSystemPlayerStore(database io.ReadWriteSeeker) *FileSystemPlayerStore {
7
    database.Seek(0, 0)
8
    league, _ := NewLeague(database)
9
    return &FileSystemPlayerStore{
10
        database:database,
11
        league:league,
12
    }
13
}
Copied!
この方法では、ディスクから一度だけ読み取る必要があります。ディスクからリーグを取得するための以前の呼び出しをすべて置き換えて、代わりにf.leagueを使用できます。
1
func (f *FileSystemPlayerStore) GetLeague() League {
2
    return f.league
3
}
4
​
5
func (f *FileSystemPlayerStore) GetPlayerScore(name string) int {
6
​
7
    player := f.league.Find(name)
8
​
9
    if player != nil {
10
        return player.Wins
11
    }
12
​
13
    return 0
14
}
15
​
16
func (f *FileSystemPlayerStore) RecordWin(name string) {
17
    player := f.league.Find(name)
18
​
19
    if player != nil {
20
        player.Wins++
21
    } else {
22
        f.league = append(f.league, Player{name, 1})
23
    }
24
​
25
    f.database.Seek(0, 0)
26
    json.NewEncoder(f.database).Encode(f.league)
27
}
Copied!
テストを実行しようとすると、FileSystemPlayerStoreの初期化について不平を言うので、新しいコンストラクターを呼び出して修正するだけです。
別の問題
非常に厄介なバグを発生させる可能性があるファイルを処理する方法には、いくつかの単純な点があります。
RecordWinを実行すると、ファイルの先頭にSeekして新しいデータを書き込みますが、新しいデータが以前のデータよりも小さい場合はどうなるでしょうか。
私たちの現在のケースでは、これは不可能です。スコアを編集または削除することはないため、データが大きくなるだけです。ただし、コードをこのようにしておくのは無責任です。削除シナリオが発生することは考えられないことではありません。
しかし、これをどのようにテストしますか？最初にコードをリファクタリングする必要があるので、書き込むデータの種類の懸念を書き込みから分離します。次に、それを個別にテストして、期待どおりに動作することを確認できます。
新しいタイプを作成して、「最初から書き始める」機能をカプセル化します。これを「テープTape」と呼びます。以下を使用して新しいファイルを作成します。
1
package main
2
​
3
import "io"
4
​
5
type tape struct {
6
    file io.ReadWriteSeeker
7
}
8
​
9
func (t *tape) Write(p []byte) (n int, err error) {
10
    t.file.Seek(0, 0)
11
    return t.file.Write(p)
12
}
Copied!
ここでは、Seek部分をカプセル化しているため、ここではWriteのみを実装していることに注意してください。つまり、FileSystemStoreは、代わりにWriterへの参照のみを持つことができます。
1
type FileSystemPlayerStore struct {
2
    database io.Writer
3
    league   League
4
}
Copied!
Tapeを使用するようにコンストラクタを更新します
1
func NewFileSystemPlayerStore(database io.ReadWriteSeeker) *FileSystemPlayerStore {
2
    database.Seek(0, 0)
3
    league, _ := NewLeague(database)
4
​
5
    return &FileSystemPlayerStore{
6
        database: &tape{database},
7
        league:   league,
8
    }
9
}
Copied!
最後に、RecordWinからSeek呼び出しを削除することで、私たちが望んでいた驚くべき見返りを得ることができます。はい、あまり感じませんが、少なくとも他の種類の書き込みを行う場合、Writeを使用して必要な動作を実行できることを意味します。さらに、潜在的に問題のあるコードを個別にテストして修正できるようになります。
ファイルの内容全体を元の内容よりも小さいもので更新するテストを書いてみましょう。
最初にテストを書く
テストでは、コンテンツを含むファイルを作成し、tapeを使用してファイルに書き込み、もう一度読み取って、ファイルの内容を確認します。
tape_test.go
1
func TestTape_Write(t *testing.T) {
2
    file, clean := createTempFile(t, "12345")
3
    defer clean()
4
​
5
    tape := &tape{file}
6
​
7
    tape.Write([]byte("abc"))
8
​
9
    file.Seek(0, 0)
10
    newFileContents, _ := ioutil.ReadAll(file)
11
​
12
    got := string(newFileContents)
13
    want := "abc"
14
​
15
    if got != want {
16
        t.Errorf("got %q want %q", got, want)
17
    }
18
}
Copied!
テストを実行してみます
1
=== RUN   TestTape_Write
2
--- FAIL: TestTape_Write (0.00s)
3
    tape_test.go:23: got 'abc45' want 'abc'
Copied!
思った通り！
必要なデータを書き込みますが、元のデータの残りを残します。
成功させるのに十分なコードを書く
os.Fileには、ファイルを効率的に空にできる切り捨て関数があります。これを呼び出して、必要なものを取得できます。
tapeを次のように変更します。
1
type tape struct {
2
    file *os.File
3
}
4
​
5
func (t *tape) Write(p []byte) (n int, err error) {
6
    t.file.Truncate(0)
7
    t.file.Seek(0, 0)
8
    return t.file.Write(p)
9
}
Copied!
コンパイラーは、io.ReadWriteSeekerを想定している多くの場所で失敗しますが、*os.Fileで送信しています。これらの問題は自分で修正できるはずですが、行き詰まった場合はソースコードを確認してください。
それを取得したら、TestTape_Writeテストはパスするはずです！
もう1つの小さなリファクタリング
RecordWinには、json.NewEncoder(f.database).Encode(f.league)という行があります。
書き込むたびに新しいエンコーダを作成する必要はありません。コンストラクタでエンコーダを初期化して、代わりに使用できます。
タイプに「エンコーダーEncoder」への参照を保存します。
1
type FileSystemPlayerStore struct {
2
    database *json.Encoder
3
    league   League
4
}
Copied!
コンストラクタで初期化します。
1
func NewFileSystemPlayerStore(file *os.File) *FileSystemPlayerStore {
2
    file.Seek(0, 0)
3
    league, _ := NewLeague(file)
4
​
5
    return &FileSystemPlayerStore{
6
        database: json.NewEncoder(&tape{file}),
7
        league:   league,
8
    }
9
}
Copied!
RecordWinで使用します。
ルールを破っただけではありませんか？プライベートなものをテストしていますか？インターフェイスはありませんか？
プライベート型のテストについて
一般的にプライベートなものをテストしないほうがよい場合があります。テストが実装に密に結合しすぎて、将来のリファクタリングを妨げる可能性があるためです。
ただし、テストによって 確信 が得られることを忘れてはなりません。
なんらかの編集または削除機能を追加した場合、実装が機能するかどうか確信が持てませんでした。特に、最初のアプローチの欠点を認識していない複数の人が作業している場合は、コードをそのままにしたくありませんでした。
最後に、これは1つのテストにすぎません。動作方法を変更することを決定した場合、テストを削除するだけで障害になることはありませんが、少なくとも将来のメンテナの要件を把握しています。
インターフェース
新しいPlayerStoreを単体テストするための最も簡単な方法であったため、io.Readerを使用してコードを開始しました。コードを開発したとき、io.ReadWriterに移動し、次にio.ReadWriteSeekerに移動しました。次に、標準ライブラリには、*os.File以外に実際にそれを実装したものは何もないことがわかりました。独自に作成するか、オープンソースを使用するかを決定することもできましたが、テスト用の一時ファイルを作成するだけで実用的でした。
最後に、*os.FileにもあるTruncateが必要でした。これらの要件を取り込んだ独自のインターフェースを作成することはオプションでした。
1
type ReadWriteSeekTruncate interface {
2
    io.ReadWriteSeeker
3
    Truncate(size int64) error
4
}
Copied!
しかし、これは本当に私たちに何を与えているのでしょうか？私たちはモックではないことを覚えておいてください。ファイルシステムストアが *os.File以外のタイプを取ることは非現実的であるため、インターフェイスが提供するポリモーフィズムは必要ありません。
ここにあるように、タイプを切り刻んで変更し、実験することを恐れないでください。静的に型付けされた言語を使用することの素晴らしい点は、コンパイラーがすべての変更を支援することです。
エラー処理
並べ替えに取り掛かる前に、現在のコードに満足していることを確認し、技術的な負債をすべて取り除く必要があります。ソフトウェアをできるだけ早く（赤の状態から抜け出す）ことは重要な原則ですが、それはエラーのケースを無視する必要があるという意味ではありません。
FileSystemStore.goに戻ると、コンストラクターにleague, _ := NewLeague(f.database)があります。
私たちが提供するio.Readerからリーグを解析できない場合、NewLeagueはエラーを返す可能性があります。
すでに失敗したテストがあったため、その時点でそれを無視するのは実用的でした。同時にそれに取り組んだとしたら、2つのことを同時に処理していたことになります。
コンストラクタがエラーを返すことができるようにしましょう。
1
func NewFileSystemPlayerStore(file *os.File) (*FileSystemPlayerStore, error) {
2
    file.Seek(0, 0)
3
    league, err := NewLeague(file)
4
​
5
    if err != nil {
6
        return nil, fmt.Errorf("problem loading player store from file %s, %v", file.Name(), err)
7
    }
8
​
9
    return &FileSystemPlayerStore{
10
        database: json.NewEncoder(&tape{file}),
11
        league:   league,
12
    }, nil
13
}
Copied!
（テストと同じように）役立つエラーメッセージを表示することが非常に重要であることを忘れないでください。インターネットの人々は冗談めかして、ほとんどのGoコードは次のとおりだと言っています
1
if err != nil {
2
    return err
3
}
Copied!
それは100％慣用的ではありません エラーメッセージにコンテキスト情報（つまり、エラーを発生させるために実行していたこと）を追加すると、ソフトウェアの操作がはるかに簡単になります。
コンパイルしようとすると、いくつかのエラーが発生します。
1
./main.go:18:35: multiple-value NewFileSystemPlayerStore() in single-value context
2
./FileSystemStore_test.go:35:36: multiple-value NewFileSystemPlayerStore() in single-value context
3
./FileSystemStore_test.go:57:36: multiple-value NewFileSystemPlayerStore() in single-value context
4
./FileSystemStore_test.go:70:36: multiple-value NewFileSystemPlayerStore() in single-value context
5
./FileSystemStore_test.go:85:36: multiple-value NewFileSystemPlayerStore() in single-value context
6
./server_integration_test.go:12:35: multiple-value NewFileSystemPlayerStore() in single-value context
Copied!
メインでは、プログラムを終了してエラーを出力します。
1
store, err := NewFileSystemPlayerStore(db)
2
​
3
if err != nil {
4
    log.Fatalf("problem creating file system player store, %v ", err)
5
}
Copied!
テストでは、エラーがないことを確認する必要があります。これを手伝ってくれるヘルパーを作ることができます。
1
func assertNoError(t *testing.T, err error) {
2
    t.Helper()
3
    if err != nil {
4
        t.Fatalf("didn't expect an error but got one, %v", err)
5
    }
6
}
Copied!
このヘルパーを使用して、他のコンパイル問題を処理します。 最後に、失敗するテストがあるはずです。
1
=== RUN   TestRecordingWinsAndRetrievingThem
2
--- FAIL: TestRecordingWinsAndRetrievingThem (0.00s)
3
    server_integration_test.go:14: didn't expect an error but got one, problem loading player store from file /var/folders/nj/r_ccbj5d7flds0sf63yy4vb80000gn/T/db841037437, problem parsing league, EOF
Copied!
ファイルが空であるため、リーグを解析できません。以前は常にエラーを無視していたため、以前はエラーが発生していませんでした。
有効なJSONをいくつか入れて、大きな統合テストを修正しましょう。
1
func TestRecordingWinsAndRetrievingThem(t *testing.T) {
2
    database, cleanDatabase := createTempFile(t, `[]`)
3
    //etc...
Copied!
すべてのテストに合格したので、ファイルが空であるシナリオを処理する必要があります。
最初にテストを書く
1
t.Run("works with an empty file", func(t *testing.T) {
2
    database, cleanDatabase := createTempFile(t, "")
3
    defer cleanDatabase()
4
​
5
    _, err := NewFileSystemPlayerStore(database)
6
​
7
    assertNoError(t, err)
8
})
Copied!
テストを実行してみます
1
=== RUN   TestFileSystemStore/works_with_an_empty_file
2
    --- FAIL: TestFileSystemStore/works_with_an_empty_file (0.00s)
3
        FileSystemStore_test.go:108: didn't expect an error but got one, problem loading player store from file /var/folders/nj/r_ccbj5d7flds0sf63yy4vb80000gn/T/db019548018, problem parsing league, EOF
Copied!
成功させるのに十分なコードを書く
コンストラクタを次のように変更します。
1
func NewFileSystemPlayerStore(file *os.File) (*FileSystemPlayerStore, error) {
2
​
3
    file.Seek(0, 0)
4
​
5
    info, err := file.Stat()
6
​
7
    if err != nil {
8
        return nil, fmt.Errorf("problem getting file info from file %s, %v", file.Name(), err)
9
    }
10
​
11
    if info.Size() == 0 {
12
        file.Write([]byte("[]"))
13
        file.Seek(0, 0)
14
    }
15
​
16
    league, err := NewLeague(file)
17
​
18
    if err != nil {
19
        return nil, fmt.Errorf("problem loading player store from file %s, %v", file.Name(), err)
20
    }
21
​
22
    return &FileSystemPlayerStore{
23
        database: json.NewEncoder(&tape{file}),
24
        league:   league,
25
    }, nil
26
}
Copied!
file.Statはファイルの統計を返し、ファイルのサイズを確認できます。空の場合は、空のJSON配列をWrite、Seekを最初に戻して、残りのコードの準備をします。
リファクタリング♪
コンストラクターは少し面倒なので、初期化コードを関数に抽出してみましょう。
1
func initialisePlayerDBFile(file *os.File) error {
2
    file.Seek(0, 0)
3
​
4
    info, err := file