Go使用Channel实现简单的消息队列

代码示例

package main

import (
	"bufio"
	"fmt"
	"math/rand"
	"os"
	"strings"
	"time"
)

//使用channel简单实现消息队列
//1

func main() {
	ch := make(chan string,1)
	rand.Seed(time.Now().UnixNano())
	consol := bufio.NewScanner(os.Stdin)
	for consol.Scan(){
		action := consol.Text()
		item := strings.Split(action," ")
		switch item[0] {
		case "push":
			if len(item) !=2 {
				fmt.Println("must be set value")
				continue
			}
			select {
			case ch<-item[1]:
				continue
			case <-time.After(10 * time.Second):
				fmt.Println("time out ")
				return
			}
		case "pop":
			fmt.Println("执行pop....")
			select {
			case v := <-ch:
				fmt.Println(v)
			case <-time.After(10 * time.Second):
				fmt.Println("time out ")
				return
			}
		case "quit", "exit":
			return
		default:
			fmt.Println("unknown action")
		}
	}
}