blob: f6538a45fd95978eacfcf447861b3bb2c3af7a91 [file] [log] [blame]
package main
import (
"fmt"
"runtime"
"time"
)
func main() {
runtime.GOMAXPROCS(2)
producerSequence := NewSequence()
consumerSequence1 := NewSequence()
// consumerSequence2 := NewSequence()
producerBarrier := NewBarrier(producerSequence)
consumerBarrier := NewBarrier(consumerSequence1) //, consumerSequence2)
sequencer := NewSingleProducerSequencer(producerSequence, RingSize, consumerBarrier)
go consume(producerBarrier, producerSequence, consumerSequence1)
// go consume(producerBarrier, producerSequence, consumerSequence2)
started := time.Now()
for i := int64(0); i < MaxIterations; i++ {
ticket := sequencer.Next(1)
ringBuffer[ticket&RingMask] = ticket
sequencer.Publish(ticket)
if ticket%Mod == 0 && ticket > 0 {
finished := time.Now()
elapsed := finished.Sub(started)
fmt.Println(ticket, elapsed)
started = time.Now()
}
}
}
func consume(barrier Barrier, source, sequence *Sequence) {
worker := NewWorker(barrier, TestHandler{}, source, sequence)
for {
worker.Process()
}
}
const MaxIterations = MaxSequenceValue
const Mod = 1000000 * 1 // 1 million * N
const RingSize = 1024
const RingMask = RingSize - 1
var ringBuffer [RingSize]int64
type TestHandler struct{}
func (this TestHandler) Consume(sequence, remaining int64) {
message := ringBuffer[sequence&RingMask]
if message != sequence {
text := fmt.Sprintf("[Consumer] ERROR Sequence: %d, Message: %d\n", sequence, message)
fmt.Printf(text)
panic(text)
}
if sequence%Mod == 0 && sequence > 0 {
// fmt.Printf("[Consumer] Sequence: %d, Message: %d\n", sequence, message)
}
}