diff options
Diffstat (limited to 'users/wpcarro/go')
-rw-r--r-- | users/wpcarro/go/.envrc | 2 | ||||
-rw-r--r-- | users/wpcarro/go/actors.go | 45 | ||||
-rw-r--r-- | users/wpcarro/go/atomic-counters.go | 26 | ||||
-rw-r--r-- | users/wpcarro/go/channels.go | 81 | ||||
-rw-r--r-- | users/wpcarro/go/mutex.go | 53 | ||||
-rw-r--r-- | users/wpcarro/go/shell.nix | 9 | ||||
-rw-r--r-- | users/wpcarro/go/waitgroups.go | 24 |
7 files changed, 240 insertions, 0 deletions
diff --git a/users/wpcarro/go/.envrc b/users/wpcarro/go/.envrc new file mode 100644 index 000000000000..a4a62da526d3 --- /dev/null +++ b/users/wpcarro/go/.envrc @@ -0,0 +1,2 @@ +source_up +use_nix diff --git a/users/wpcarro/go/actors.go b/users/wpcarro/go/actors.go new file mode 100644 index 000000000000..1409db185e84 --- /dev/null +++ b/users/wpcarro/go/actors.go @@ -0,0 +1,45 @@ +package main + +import ( + "bufio" + "fmt" + "os" +) + +// Call function `f` in a go-routine, passing a reference to a newly created +// channel, `c`, as its only argument. Return a reference to `c` to the caller +// of `act`. When `f` halts, close the channel. +func act(f func(chan interface{})) chan interface{} { + c := make(chan interface{}) + + go func() { + defer close(c) + f(c) + }() + + return c +} + +func prompt(msg string) string { + reader := bufio.NewReader(os.Stdin) + fmt.Print(msg) + text, _ := reader.ReadString('\n') + // TODO: Trim trailing newline from the rhs of text. + return text +} + +func main() { + c := act(func(c chan interface{}) { + for { + x := <-c + fmt.Printf("[A] Received value: %v\n", x) + + } + }) + + for { + x := prompt("[B] Enter a value: ") + c <- x + } + os.Exit(0) +} diff --git a/users/wpcarro/go/atomic-counters.go b/users/wpcarro/go/atomic-counters.go new file mode 100644 index 000000000000..6cbcd2ee4eaf --- /dev/null +++ b/users/wpcarro/go/atomic-counters.go @@ -0,0 +1,26 @@ +// Attempting to apply some of the lessons I learned here: +// https://gobyexample.com/atomic-counters +package main + +import ( + "fmt" + "sync" + "sync/atomic" +) + +func main() { + var count uint64 + var wg sync.WaitGroup + + for i := 0; i < 50; i += 1 { + wg.Add(1) + go func() { + defer wg.Done() + for j := 0; j < 1000; j += 1 { + atomic.AddUint64(&count, 1) + } + }() + } + wg.Wait() + fmt.Println("Count: ", count) +} diff --git a/users/wpcarro/go/channels.go b/users/wpcarro/go/channels.go new file mode 100644 index 000000000000..cba8abfc9621 --- /dev/null +++ b/users/wpcarro/go/channels.go @@ -0,0 +1,81 @@ +package main + +import ( + "fmt" + "math/rand" + "sync" + "sync/atomic" +) + +type readMsg struct { + key int + sender chan int +} + +type writeMsg struct { + key int + value int + sender chan bool +} + +func main() { + fmt.Println("Hello, go.") + + var readOps uint64 + var writeOps uint64 + var wg sync.WaitGroup + + reads := make(chan readMsg) + writes := make(chan writeMsg) + + go func() { + state := make(map[int]int) + for { + select { + case msg := <-reads: + msg.sender <- state[msg.key] + case msg := <-writes: + state[msg.key] = msg.value + msg.sender <- true + } + } + }() + + // Reads + for i := 0; i < 100; i += 1 { + go func() { + wg.Add(1) + defer wg.Done() + for j := 0; j < 100; j += 1 { + msg := readMsg{ + key: rand.Intn(5), + sender: make(chan int)} + reads <- msg + val := <-msg.sender + fmt.Printf("Received %d.\n", val) + atomic.AddUint64(&readOps, 1) + } + }() + } + + // Writes + for i := 0; i < 100; i += 1 { + go func() { + wg.Add(1) + defer wg.Done() + for j := 0; j < 100; j += 1 { + msg := writeMsg{ + key: rand.Intn(5), + value: rand.Intn(10), + sender: make(chan bool)} + writes <- msg + <-msg.sender + fmt.Printf("Set %d as %d in state\n", msg.key, msg.value) + atomic.AddUint64(&writeOps, 1) + } + }() + } + + wg.Wait() + fmt.Printf("Read ops: %d\tWrite ops: %d\n", atomic.LoadUint64(&readOps), atomic.LoadUint64(&writeOps)) +} diff --git a/users/wpcarro/go/mutex.go b/users/wpcarro/go/mutex.go new file mode 100644 index 000000000000..5cea20754bed --- /dev/null +++ b/users/wpcarro/go/mutex.go @@ -0,0 +1,53 @@ +package main + +import ( + "fmt" + "math/rand" + "sync" + "sync/atomic" + "time" +) + +func main() { + state := make(map[int]int) + mux := &sync.Mutex{} + + var readOps uint64 + var writeOps uint64 + + // Read from state + for i := 0; i < 1000; i += 1 { + for j := 0; j < 100; j += 1 { + go func() { + key := rand.Intn(5) + mux.Lock() + fmt.Printf("state[%d] = %d\n", key, state[key]) + mux.Unlock() + atomic.AddUint64(&readOps, 1) + time.Sleep(time.Millisecond) + }() + } + } + + // Write to state + for i := 0; i < 10; i += 1 { + for j := 0; j < 100; j += 1 { + go func() { + key := rand.Intn(5) + mux.Lock() + state[key] += 1 + mux.Unlock() + fmt.Printf("Wrote to state[%d].\n", key) + atomic.AddUint64(&writeOps, 1) + time.Sleep(time.Millisecond) + }() + } + } + + time.Sleep(time.Millisecond) + + mux.Lock() + fmt.Printf("State: %v\n", state) + mux.Unlock() + fmt.Printf("Reads: %d\tWrites: %d\n", atomic.LoadUint64(&readOps), atomic.LoadUint64(&writeOps)) +} diff --git a/users/wpcarro/go/shell.nix b/users/wpcarro/go/shell.nix new file mode 100644 index 000000000000..f777c13fefae --- /dev/null +++ b/users/wpcarro/go/shell.nix @@ -0,0 +1,9 @@ +{ pkgs, ... }: + +pkgs.mkShell { + buildInputs = with pkgs; [ + go + goimports + godef + ]; +} diff --git a/users/wpcarro/go/waitgroups.go b/users/wpcarro/go/waitgroups.go new file mode 100644 index 000000000000..816321b8770f --- /dev/null +++ b/users/wpcarro/go/waitgroups.go @@ -0,0 +1,24 @@ +package main + +import ( + "fmt" + "sync" + "time" +) + +func saySomething(x string, wg *sync.WaitGroup) { + defer wg.Done() + fmt.Println(x) + time.Sleep(time.Second) + fmt.Printf("Finished saying \"%s\"\n", x) +} + +func main() { + var wg sync.WaitGroup + var things = [5]string{"chicken", "panini", "cheeseburger", "rice", "bread"} + for i := 0; i < 5; i += 1 { + wg.Add(1) + go saySomething(things[i], &wg) + } + wg.Wait() +} |