2015-11-30 21:27:58 -08:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
2015-11-30 21:52:08 -08:00
|
|
|
"io"
|
|
|
|
"log"
|
2015-11-30 21:27:58 -08:00
|
|
|
"math/rand"
|
2015-11-30 21:52:08 -08:00
|
|
|
"net"
|
2015-11-30 21:27:58 -08:00
|
|
|
"sync"
|
|
|
|
"time"
|
|
|
|
)
|
|
|
|
|
|
|
|
var mu sync.Mutex
|
|
|
|
|
|
|
|
type queue struct {
|
|
|
|
size int
|
|
|
|
load []int
|
|
|
|
}
|
|
|
|
|
|
|
|
func remove(s []int, i int) []int {
|
|
|
|
copy(s[i:], s[i+1:])
|
|
|
|
return s[:len(s)-1]
|
|
|
|
}
|
|
|
|
|
|
|
|
func work(q *queue) <-chan string {
|
|
|
|
ch := make(chan string)
|
|
|
|
go func() {
|
|
|
|
for {
|
|
|
|
mu.Lock()
|
|
|
|
curLoad := len(q.load)
|
|
|
|
mu.Unlock()
|
|
|
|
if curLoad > 0 {
|
|
|
|
mu.Lock()
|
|
|
|
curValue := q.load[0]
|
|
|
|
mu.Unlock()
|
|
|
|
time.Sleep(time.Millisecond * time.Duration(curValue))
|
|
|
|
mu.Lock()
|
|
|
|
q.load = remove(q.load, 0)
|
|
|
|
mu.Unlock()
|
|
|
|
ch <- "finished"
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}()
|
|
|
|
return ch
|
|
|
|
}
|
|
|
|
|
|
|
|
func load(q *queue) {
|
|
|
|
go func() {
|
|
|
|
for {
|
|
|
|
time.Sleep(time.Millisecond * time.Duration(rand.Intn(1000)))
|
|
|
|
mu.Lock()
|
|
|
|
q.load = append(q.load, rand.Intn(1000))
|
|
|
|
mu.Unlock()
|
|
|
|
}
|
|
|
|
}()
|
|
|
|
}
|
|
|
|
|
2015-11-30 21:44:43 -08:00
|
|
|
func printer(q *queue) {
|
|
|
|
go func() {
|
|
|
|
for {
|
|
|
|
mu.Lock()
|
|
|
|
fmt.Println(q)
|
|
|
|
mu.Unlock()
|
|
|
|
time.Sleep(time.Millisecond * 1000)
|
|
|
|
}
|
|
|
|
}()
|
|
|
|
}
|
|
|
|
|
2015-11-30 21:27:58 -08:00
|
|
|
func init() {
|
|
|
|
rand.Seed(time.Now().UnixNano())
|
|
|
|
}
|
|
|
|
|
2015-11-30 21:44:43 -08:00
|
|
|
var q = queue{
|
|
|
|
size: 10,
|
|
|
|
}
|
|
|
|
|
2015-11-30 21:27:58 -08:00
|
|
|
func main() {
|
2015-11-30 21:52:08 -08:00
|
|
|
listener, err := net.Listen("tcp", "localhost:8080")
|
|
|
|
if err != nil {
|
|
|
|
log.Fatal(err)
|
|
|
|
}
|
|
|
|
go func() {
|
|
|
|
for {
|
|
|
|
conn, err := listener.Accept()
|
|
|
|
if err != nil {
|
|
|
|
log.Print(err)
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
handleConn(conn)
|
|
|
|
}
|
|
|
|
}()
|
2015-11-30 21:44:43 -08:00
|
|
|
fmt.Println("here")
|
2015-11-30 21:27:58 -08:00
|
|
|
load(&q)
|
2015-11-30 21:44:43 -08:00
|
|
|
printer(&q)
|
2015-11-30 21:27:58 -08:00
|
|
|
ch := work(&q)
|
|
|
|
for {
|
|
|
|
select {
|
|
|
|
case <-ch:
|
|
|
|
fmt.Println("finished something")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2015-11-30 21:44:43 -08:00
|
|
|
|
2015-11-30 21:52:08 -08:00
|
|
|
func handleConn(c net.Conn) {
|
|
|
|
defer c.Close()
|
|
|
|
for {
|
|
|
|
mu.Lock()
|
|
|
|
_, err := io.WriteString(c, fmt.Sprintf("%v\n", q))
|
|
|
|
mu.Unlock()
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
time.Sleep(time.Millisecond * 100)
|
|
|
|
}
|
2015-11-30 21:44:43 -08:00
|
|
|
}
|