状态协程
package main
import (
"fmt"
"math/rand"
"sync/atomic"
"time"
)
// 读数据结构体
type readOp struct {
key int
resp chan int
}
// 写据结构体
type writeOp struct {
key int
val int
resp chan bool
}
func main() {
var readOps uint64
var writeOps uint64
reads := make(chan readOp) //创建一个用于监听读数据的协程
writes := make(chan writeOp)//创建一个用于监听写数据的协程
/**
** 监听所有的读写操作,并反馈结果
** 这就是拥有 state 的那个协程, 不过这里的 state 是被这个状态协程私有的。
** 这个协程不断地在 reads 和 writes 通道上进行选择,并在请求到达时做出响应。
** 首先,执行请求的操作;然后,执行响应,在响应通道 resp 上发送一个值,
** 表明请求成功(reads 的值则为 state 对应的值)。
*/
go func() {
var state = make(map[int]int)
for {
select {
case read := <-reads:
read.resp <- state[read.key]// 反馈
case write := <-writes:
state[write.key] = write.val//写入state
write.resp <- true// 反馈
}
}
}()
/**
** 启动 100 个协程通过 reads 通道向拥有 state 的协程发起读取请求。
** 每个读取请求需要构造一个 readOp,发送它到 reads 通道中, 并通过给定的 resp 通道接收结果。
*/
for r := 0; r < 100; r++ {
go func() {
for {
read := readOp{
key: rand.Intn(5),
resp: make(chan int)}
reads <- read//向通道发起读
<-read.resp//等待读结束
atomic.AddUint64(&readOps, 1)
time.Sleep(time.Millisecond)
}
}()
}
// 用相同的方法启动10个写操作。
for w := 0; w < 10; w++ {
go func() {
for {
write := writeOp{
key: rand.Intn(5),
val: rand.Intn(100),
resp: make(chan bool)}
writes <- write//向通道发起写
<-write.resp//等待写结束
atomic.AddUint64(&writeOps, 1)
time.Sleep(time.Millisecond)
}
}()
}
//让程序运行1秒
time.Sleep(time.Second)
readOpsFinal := atomic.LoadUint64(&readOps)
fmt.Println("readOps:", readOpsFinal)
writeOpsFinal := atomic.LoadUint64(&writeOps)
fmt.Println("writeOps:", writeOpsFinal)
}