golang redigo发布订阅使用的方法

目录

• redigo 对 发布订阅的使用
• 订阅的主题
• 发布

redigo 对 发布订阅的使用

redigo 对redis 的发布订阅机制放在pubsub.go 中
订阅主题后 通过receive() 函数接受发布订阅主题的消息

// receive returns a pushed message as a subscription, message, pong or error.
// the return value is intended to be used directly in a type switch as
// illustrated in the pubsubconn example.
func (c pubsubconn) receive() interface{} {
	return c.receiveinternal(c.conn.receive())
}

返回的是一个空类型的interface{} , 由于空接口没有方法， 因此所有的类型都实现了空接口， 也就是说可以返回任意类型。

具体返回的类型 在receiveinternal（） 方法里面可以看到

func (c pubsubconn) receiveinternal(replyarg interface{}, errarg error) interface{} {
	reply, err := values(replyarg, errarg)
	if err != nil {
		return err
	}

	var kind string
	reply, err = scan(reply, &kind)
	if err != nil {
		return err
	}

	switch kind {
	case "message":
		var m message
		if _, err := scan(reply, &m.channel, &m.data); err != nil {
			return err
		}
		return m
	case "pmessage":
		var m message
		if _, err := scan(reply, &m.pattern, &m.channel, &m.data); err != nil {
			return err
		}
		return m
	case "subscribe", "psubscribe", "unsubscribe", "punsubscribe":
		s := subscription{kind: kind}
		if _, err := scan(reply, &s.channel, &s.count); err != nil {
			return err
		}
		return s
	case "pong":
		var p pong
		if _, err := scan(reply, &p.data); err != nil {
			return err
		}
		return p
	}
	return errors.new("redigo: unknown pubsub notification")
}

目前返回 message subscription pone

订阅的主题

收到消息之后通过注册的回调函数处理的方式， 所以代码中多了长map存放回调函数

package main

import (
	//"github.com/go-redis/redis"
	"fmt"
	"github.com/labstack/gommon/log"
	"github.com/gomodule/redigo/redis"
	"time"

	//"reflect"
	"unsafe"
)

type subscribecallback func (channel, message string)

type subscriber struct {
	client redis.pubsubconn
	cbmap map[string]subscribecallback
}

func (c *subscriber) connect(ip string, port uint16) {
	conn, err := redis.dial("tcp", "127.0.0.1:6379")
	if err != nil {
		log.error("redis dial failed.")
	}

	c.client = redis.pubsubconn{conn}
	c.cbmap = make(map[string]subscribecallback)

	go func() {
		for {
			log.info("wait...")
			switch res := c.client.receive().(type) {
			case redis.message:
				channel := (*string)(unsafe.pointer(&res.channel))
				message := (*string)(unsafe.pointer(&res.data))
				c.cbmap[*channel](*channel, *message)
			case redis.subscription:
				fmt.printf("%s: %s %d\n", res.channel, res.kind, res.count)
			case error:
				log.error("error handle...")
				continue
			}
		}
	}()

}

func (c *subscriber) close() {
	err := c.client.close()
	if err != nil{
		log.error("redis close error.")
	}
}

func (c *subscriber) subscribe(channel interface{}, cb subscribecallback) {
	err := c.client.subscribe(channel)
	if err != nil{
		log.error("redis subscribe error.")
	}

	c.cbmap[channel.(string)] = cb
}

func testcallback1(chann, msg string){
	log.info("testcallback1 channel : ", chann, " message : ", msg)
}

func testcallback2(chann, msg string){
	log.info("testcallback2 channel : ", chann, " message : ", msg)
}

func testcallback3(chann, msg string){
	log.info("testcallback3 channel : ", chann, " message : ", msg)
}

func main() {

	log.info("===========main start============")

	var sub subscriber
	sub.connect("127.0.0.1", 6397)
	sub.subscribe("test_chan1", testcallback1)
	sub.subscribe("test_chan2", testcallback2)
	sub.subscribe("test_chan3", testcallback3)

	for{ // 这段代码的作用就是 阻止线程结束
		time.sleep(1 * time.second)
	}
}

运行main 文件 然后 看到

在redis 客户端 执行 发布信息

在控制台 看到监控结果 回调函数 执行的结果

发布

上面是订阅的代码和 代码要处理 的回调函数
发布直接使用默认的 conn来send publish 就可以
redigo 的管道使用方法设计到三个函数 do 函数也是下面这个函数的合并

• c.send()
• c.flush()
• c.receive()

解释：
send() 方法吧命令写到缓冲区， flush() 把缓冲区的命令刷新到redis 服务器 receive() 函数接受redis 给予的 回应， 三个操作共同完成一套命令流程。

代码

package main

import(
  //"github.com/go-redis/redis"
  "github.com/gomodule/redigo/redis"
  log "github.com/astaxie/beego/logs"
)

func main() {

  client, err := redis.dial("tcp", "127.0.0.1:6379")
  if err != nil {
      log.critical("redis dial failed.")
  }
  defer client.close()

  _, err = client.do("publish", "test_chan1", "hello")
  if err != nil {
    log.critical("redis publish failed.")
  }

  _, err = client.do("publish", "test_chan2", "hello")
  if err != nil {
    log.critical("redis publish failed.")
  }

  _, err = client.do("publish", "test_chan3", "hello")
  if err != nil {
    log.critical("redis publish failed.")
  }
}

到此这篇关于golang redigo发布订阅使用的方法的文章就介绍到这了,更多相关golang redigo发布订阅内容请搜索www.887551.com以前的文章或继续浏览下面的相关文章希望大家以后多多支持www.887551.com！