はじめに
RedisのGo言語向けクライアントライブラリRedigoの使い方を見ます。
あまり知られていないかもしれませんが、RedisにはPub/Sub（パブリッシュ／サブスクライブ）機能があります。この機能を使うと、例えばサーバ間のメッセージ通知のような1対多の通信を行うことができます。この記事ではパブリッシュ／サブスクライブの行い方を見ます。
環境
- OS: Windows 10
- Redis: win-3.2.100
- Go言語: 1.11
サブスクライブ
Redisでは、いわゆる「トピックベース」の通信ができます。受信側（サブスクライバー）は始めにトピック（チャネルと呼びます）を指定して購読を開始し、チャネルからの通信を待機します。
subscriber.go
package main
import (
"fmt"
"github.com/gomodule/redigo/redis"
)
func main() {
// 接続
conn, err := redis.Dial("tcp", "localhost:6379")
if err != nil {
panic(err)
}
defer conn.Close()
psc := redis.PubSubConn{Conn: conn}
psc.Subscribe("channel_1", "channel_2", "channel_3")
for {
switch v := psc.Receive().(type) {
case redis.Message:
fmt.Printf("%s: message: %s\n", v.Channel, v.Data)
case redis.Subscription:
fmt.Printf("%s: %s %d\n", v.Channel, v.Kind, v.Count)
case error:
return
}
}
}
上記はサンプルをほぼそのまま持ってきたコードです。イディオム的にそのまま使えばよいと思います（実用的には切断された時に再接続を試す処理が必要と思います）。Subscribeで購読を開始するチャネルを並べて記し、Receiveでブロッキングでメッセージを待機します。
サブスクライブに成功するとチャネルはredis.Subscriptionを受け取ります（case節の２つ目）。このとき成功したチャネル名と通知の種類（"subscribe"）、購読チャネル数が得られます。
パブリッシュ
次は送信側（パブリッシャー）のコードです。
publisher.go
package main
import (
"fmt"
"github.com/gomodule/redigo/redis"
)
func main() {
// 接続
conn, err := redis.Dial("tcp", "localhost:6379")
if err != nil {
panic(err)
}
defer conn.Close()
// パブリッシュ
r, err := redis.Int(conn.Do("PUBLISH", "channel_1", "hello"))
if err != nil {
panic(err)
}
fmt.Println(r)
}
サブスクライバー側のコードを実行した状態で上のコードを実行すると、サブスクライバー側のターミナルにメッセージが表示されます。複数を起動していれば全てに同じ内容が表示されます。
パブリッシャー側のコードは比較的単純です。PUBLISHコマンドでチャネルとメッセージを指定して送信します。戻り値はメッセージが配信されたサブスクライバーの数になります。
おわりに
Redidoでのパブリッシュ／サブスクライブの行い方を見ていきました。