golang使用redis实现消息队列

golang使用redis实现消息队列（producer-consumer）

废话不多说，直接上代码，仅供参考

package main

import (
	"fmt"
	"math/rand"
	"os"
	"strconv"
	"time"

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

const RMQ string = "mqtest"

func producer() {
	redis_conn, err := redis.Dial("tcp", "127.0.0.1:6379")
	if err != nil {
		fmt.Println(err)
		return
	}

	defer redis_conn.Close()

	rand.Seed(time.Now().UnixNano())

	var i = 1

	for {
		_, err = redis_conn.Do("rpush", RMQ, strconv.Itoa(i))
		if err != nil {
			fmt.Println("produce error")
			continue
		}
		fmt.Println("produce element:%d", i)
		time.Sleep(time.Duration(rand.Intn(10)) * time.Second)
		i++
	}
}

func consumer() {
	redis_conn, err := redis.Dial("tcp", "127.0.0.1:6379")
	if err != nil {
		fmt.Println(err)
		return
	}

	defer redis_conn.Close()

	rand.Seed(time.Now().UnixNano())

	for {
		ele, err := redis.String(redis_conn.Do("lpop", RMQ))
		if err != nil {
			fmt.Println("no msg.sleep now")
			time.Sleep(time.Duration(rand.Intn(10)) * time.Second)
		} else {
			fmt.Println("cosume element:%s", ele)
		}
	}
}

func main() {
	list := os.Args
	fmt.Println(list)
	if list[1] == "pro" {
		go producer()
	} else if list[1] == "con" {
		go consumer()
	}
	for {
		time.Sleep(time.Duration(10000) * time.Second)
	}
}