package disruptor

import "sync/atomic"

type SharedWriter struct {
	capacity  int64
	gate      int64 // TODO: determine if this should be a *Cursor
	mask      int64
	shift     uint8
	committed []int32
	read      Barrier
	written   *Cursor
}

func NewSharedWriter(write *SharedWriterBarrier, read Barrier) *SharedWriter {
	return &SharedWriter{
		capacity:  write.capacity,
		gate:      InitialSequenceValue,
		mask:      write.mask,
		shift:     write.shift,
		committed: write.committed,
		read:      read,
		written:   write.written,
	}
}

func (this *SharedWriter) Reserve(count int64) (int64, int64) {
	for {
		previous := this.written.Load()
		upper := previous + count
		wrap := upper - this.capacity

		if wrap > this.gate {
			min := this.read.LoadBarrier(0)
			if wrap > min {
				return InitialSequenceValue, Gating
			}

			this.gate = min // doesn't matter which write wins, BUT will most likely need to be a Cursor
		}

		if atomic.CompareAndSwapInt64(&this.written.sequence, previous, upper) {
			return previous + 1, upper
		}
	}
}

func (this *SharedWriter) Commit(lower, upper int64) {
	for shift, mask := this.shift, this.mask; lower <= upper; lower++ {
		this.committed[lower&mask] = int32(lower >> shift)
	}
}