Исходники и презентации

lessons/sync_primitives/sync_stack/main.go

@@ -0,0 +1,76 @@
+package main
+
+import (
+	"sync"
+)
+
+// Need to show solution
+
+type Stack struct {
+	mutex sync.Mutex
+	data  []string
+}
+
+func NewStack() Stack {
+	return Stack{}
+}
+
+func (b Stack) Push(value string) {
+	b.mutex.Lock()
+	defer b.mutex.Unlock()
+
+	b.data = append(b.data, value)
+}
+
+func (b Stack) Pop() {
+	if len(b.data) < 0 {
+		panic("pop: stack is empty")
+	}
+
+	b.mutex.Lock()
+	defer b.mutex.Unlock()
+
+	b.data = b.data[:len(b.data)-1]
+}
+
+func (b Stack) Top() string {
+	if len(b.data) < 0 {
+		panic("top: stack is empty")
+	}
+
+	b.mutex.Lock()
+	defer b.mutex.Unlock()
+
+	return b.data[len(b.data)-1]
+}
+
+var stack Stack
+
+func producer() {
+	for i := 0; i < 1000; i++ {
+		stack.Push("message")
+	}
+}
+
+func consumer() {
+	for i := 0; i < 10; i++ {
+		_ = stack.Top()
+		stack.Pop()
+	}
+}
+
+func main() {
+	producer()
+
+	wg := sync.WaitGroup{}
+	wg.Add(100)
+
+	for i := 0; i < 100; i++ {
+		go func() {
+			defer wg.Done()
+			consumer()
+		}()
+	}
+
+	wg.Wait()
+}