Go语言实现Redis的发布订阅功能

之前学习了使用redigo连接操作redis,今天测试一下redigo的发布订阅功能.

本文github: https://github.com/zboyco/go-test/blob/master/redigo-sub/main.go

---

代码相对简单,直接贴代码了

package main

import (
	"fmt"
	"log"

	"github.com/gomodule/redigo/redis"
)

func main() {
	log.Println("测试redis发布订阅")

	conn, err := redis.Dial("tcp", "192.168.2.99:6379")
	checkErr(err)
	defer conn.Close()

	// 开启三个goroutine订阅redis
	go subScribe()
	go subScribe()
	go subScribe()

	// 从控制台获取内容发布到redis
	for {
		var s string
		fmt.Scanln(&s)
		_, err := conn.Do("PUBLISH", "testRoom", s)
		if err != nil {
			fmt.Println("pub err: ", err)
			return
		}
	}
}

// 订阅
func subScribe() {
	conn, err := redis.Dial("tcp", "192.168.2.99:6379")
	checkErr(err)
	defer conn.Close()

	psc := redis.PubSubConn{
		Conn: conn,
	}
	psc.Subscribe("testRoom")
	for {
		switch v := psc.Receive().(type) {
		case redis.Message:
			log.Printf("%s: message: %s\n", v.Channel, v.Data)
		case redis.Subscription:
			log.Printf("%s: %s %d\n", v.Channel, v.Kind, v.Count)
		case error:
			log.Fatalln(v)
		}
	}
}

// checkErr 错误检查
func checkErr(err error) {
	if err != nil {
		log.Fatalln(err)
	}
}

效果如下: