blob: 36b18f80ea9faf66423a310d81d8d9bb6f9337a2 [file] [log] [blame] [edit]
// run
// Copyright 2009 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
// Test communication operations including select.
package main
import "os"
import "runtime"
import "sync"
var randx int
func nrand(n int) int {
randx += 10007
if randx >= 1000000 {
randx -= 1000000
}
return randx % n
}
type Chan struct {
sc, rc chan int // send and recv chan
sv, rv int // send and recv seq
}
var (
nproc int
nprocLock sync.Mutex
cval int
end int = 10000
totr, tots int
totLock sync.Mutex
nc *Chan
)
func init() {
nc = new(Chan)
}
func changeNproc(adjust int) int {
nprocLock.Lock()
nproc += adjust
ret := nproc
nprocLock.Unlock()
return ret
}
func mkchan(c, n int) []*Chan {
ca := make([]*Chan, n)
for i := 0; i < n; i++ {
cval = cval + 100
ch := new(Chan)
ch.sc = make(chan int, c)
ch.rc = ch.sc
ch.sv = cval
ch.rv = cval
ca[i] = ch
}
return ca
}
func expect(v, v0 int) (newv int) {
if v == v0 {
if v%100 == 75 {
return end
}
return v + 1
}
print("got ", v, " expected ", v0+1, "\n")
panic("fail")
}
func (c *Chan) send() bool {
// print("send ", c.sv, "\n");
totLock.Lock()
tots++
totLock.Unlock()
c.sv = expect(c.sv, c.sv)
if c.sv == end {
c.sc = nil
return true
}
return false
}
func send(c *Chan) {
for {
for r := nrand(10); r >= 0; r-- {
runtime.Gosched()
}
c.sc <- c.sv
if c.send() {
break
}
}
changeNproc(-1)
}
func (c *Chan) recv(v int) bool {
// print("recv ", v, "\n");
totLock.Lock()
totr++
totLock.Unlock()
c.rv = expect(c.rv, v)
if c.rv == end {
c.rc = nil
return true
}
return false
}
func recv(c *Chan) {
var v int
for {
for r := nrand(10); r >= 0; r-- {
runtime.Gosched()
}
v = <-c.rc
if c.recv(v) {
break
}
}
changeNproc(-1)
}
func sel(r0, r1, r2, r3, s0, s1, s2, s3 *Chan) {
var v int
a := 0 // local chans running
if r0.rc != nil {
a++
}
if r1.rc != nil {
a++
}
if r2.rc != nil {
a++
}
if r3.rc != nil {
a++
}
if s0.sc != nil {
a++
}
if s1.sc != nil {
a++
}
if s2.sc != nil {
a++
}
if s3.sc != nil {
a++
}
for {
for r := nrand(5); r >= 0; r-- {
runtime.Gosched()
}
select {
case v = <-r0.rc:
if r0.recv(v) {
a--
}
case v = <-r1.rc:
if r1.recv(v) {
a--
}
case v = <-r2.rc:
if r2.recv(v) {
a--
}
case v = <-r3.rc:
if r3.recv(v) {
a--
}
case s0.sc <- s0.sv:
if s0.send() {
a--
}
case s1.sc <- s1.sv:
if s1.send() {
a--
}
case s2.sc <- s2.sv:
if s2.send() {
a--
}
case s3.sc <- s3.sv:
if s3.send() {
a--
}
}
if a == 0 {
break
}
}
changeNproc(-1)
}
// direct send to direct recv
func test1(c *Chan) {
changeNproc(2)
go send(c)
go recv(c)
}
// direct send to select recv
func test2(c int) {
ca := mkchan(c, 4)
changeNproc(4)
go send(ca[0])
go send(ca[1])
go send(ca[2])
go send(ca[3])
changeNproc(1)
go sel(ca[0], ca[1], ca[2], ca[3], nc, nc, nc, nc)
}
// select send to direct recv
func test3(c int) {
ca := mkchan(c, 4)
changeNproc(4)
go recv(ca[0])
go recv(ca[1])
go recv(ca[2])
go recv(ca[3])
changeNproc(1)
go sel(nc, nc, nc, nc, ca[0], ca[1], ca[2], ca[3])
}
// select send to select recv
func test4(c int) {
ca := mkchan(c, 4)
changeNproc(2)
go sel(nc, nc, nc, nc, ca[0], ca[1], ca[2], ca[3])
go sel(ca[0], ca[1], ca[2], ca[3], nc, nc, nc, nc)
}
func test5(c int) {
ca := mkchan(c, 8)
changeNproc(2)
go sel(ca[4], ca[5], ca[6], ca[7], ca[0], ca[1], ca[2], ca[3])
go sel(ca[0], ca[1], ca[2], ca[3], ca[4], ca[5], ca[6], ca[7])
}
func test6(c int) {
ca := mkchan(c, 12)
changeNproc(4)
go send(ca[4])
go send(ca[5])
go send(ca[6])
go send(ca[7])
changeNproc(4)
go recv(ca[8])
go recv(ca[9])
go recv(ca[10])
go recv(ca[11])
changeNproc(2)
go sel(ca[4], ca[5], ca[6], ca[7], ca[0], ca[1], ca[2], ca[3])
go sel(ca[0], ca[1], ca[2], ca[3], ca[8], ca[9], ca[10], ca[11])
}
// wait for outstanding tests to finish
func wait() {
runtime.Gosched()
for changeNproc(0) != 0 {
runtime.Gosched()
}
}
// run all tests with specified buffer size
func tests(c int) {
ca := mkchan(c, 4)
test1(ca[0])
test1(ca[1])
test1(ca[2])
test1(ca[3])
wait()
test2(c)
wait()
test3(c)
wait()
test4(c)
wait()
test5(c)
wait()
test6(c)
wait()
}
// run all test with 4 buffser sizes
func main() {
tests(0)
tests(1)
tests(10)
tests(100)
t := 4 * // buffer sizes
(4*4 + // tests 1,2,3,4 channels
8 + // test 5 channels
12) * // test 6 channels
76 // sends/recvs on a channel
if tots != t || totr != t {
print("tots=", tots, " totr=", totr, " sb=", t, "\n")
os.Exit(1)
}
os.Exit(0)
}