77 lines
930 B
Go
77 lines
930 B
Go
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()
|
|
}
|