mirror of
https://github.com/golang/go
synced 2024-11-21 15:44:44 -07:00
doc: add round-robin flag to io2010 balance example.
R=r CC=golang-dev https://golang.org/cl/2050042
This commit is contained in:
parent
7a2daa7d3a
commit
724b66fb15
@ -6,6 +6,7 @@ package main
|
||||
|
||||
import (
|
||||
"container/heap"
|
||||
"flag"
|
||||
"fmt"
|
||||
"rand"
|
||||
"time"
|
||||
@ -14,6 +15,8 @@ import (
|
||||
const nRequester = 100
|
||||
const nWorker = 10
|
||||
|
||||
var roundRobin = flag.Bool("r", false, "use round-robin scheduling")
|
||||
|
||||
// Simulation of some work: just sleep for a while and report how long.
|
||||
func op() int {
|
||||
n := rand.Int63n(1e9)
|
||||
@ -125,7 +128,7 @@ func (b *Balancer) print() {
|
||||
}
|
||||
|
||||
func (b *Balancer) dispatch(req Request) {
|
||||
if false {
|
||||
if *roundRobin {
|
||||
w := b.pool[b.i]
|
||||
w.requests <- req
|
||||
w.pending++
|
||||
@ -144,7 +147,7 @@ func (b *Balancer) dispatch(req Request) {
|
||||
}
|
||||
|
||||
func (b *Balancer) completed(w *Worker) {
|
||||
if false {
|
||||
if *roundRobin {
|
||||
w.pending--
|
||||
return
|
||||
}
|
||||
@ -156,6 +159,7 @@ func (b *Balancer) completed(w *Worker) {
|
||||
}
|
||||
|
||||
func main() {
|
||||
flag.Parse()
|
||||
work := make(chan Request)
|
||||
for i := 0; i < nRequester; i++ {
|
||||
go requester(work)
|
||||
|
Loading…
Reference in New Issue
Block a user