73 lines
1.5 KiB
Go
73 lines
1.5 KiB
Go
|
package bridge
|
||
|
|
|
||
|
|
/* Todo: When go generics ships, refactor this and event.infiniteChannel into one */
|
||
|
|
|
||
|
|
// InfiniteChannel implements the Channel interface with an infinite buffer between the input and the output.
|
||
|
|
type InfiniteChannel struct {
|
||
|
|
input, output chan interface{}
|
||
|
|
length chan int
|
||
|
|
buffer *InfiniteQueue
|
||
|
|
}
|
||
|
|
|
||
|
|
func newInfiniteChannel() *InfiniteChannel {
|
||
|
|
ch := &InfiniteChannel{
|
||
|
|
input: make(chan interface{}),
|
||
|
|
output: make(chan interface{}),
|
||
|
|
length: make(chan int),
|
||
|
|
buffer: newInfiniteQueue(),
|
||
|
|
}
|
||
|
|
go ch.infiniteBuffer()
|
||
|
|
return ch
|
||
|
|
}
|
||
|
|
|
||
|
|
// In returns the input channel
|
||
|
|
func (ch *InfiniteChannel) In() chan<- interface{} {
|
||
|
|
return ch.input
|
||
|
|
}
|
||
|
|
|
||
|
|
// Out returns the output channel
|
||
|
|
func (ch *InfiniteChannel) Out() <-chan interface{} {
|
||
|
|
return ch.output
|
||
|
|
}
|
||
|
|
|
||
|
|
// Len returns the length of items in queue
|
||
|
|
func (ch *InfiniteChannel) Len() int {
|
||
|
|
return <-ch.length
|
||
|
|
}
|
||
|
|
|
||
|
|
// Close closes the InfiniteChanel
|
||
|
|
func (ch *InfiniteChannel) Close() {
|
||
|
|
close(ch.input)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (ch *InfiniteChannel) infiniteBuffer() {
|
||
|
|
var input, output chan interface{}
|
||
|
|
var next interface{}
|
||
|
|
input = ch.input
|
||
|
|
|
||
|
|
for input != nil || output != nil {
|
||
|
|
select {
|
||
|
|
case elem, open := <-input:
|
||
|
|
if open {
|
||
|
|
ch.buffer.Add(elem)
|
||
|
|
} else {
|
||
|
|
input = nil
|
||
|
|
}
|
||
|
|
case output <- next:
|
||
|
|
ch.buffer.Remove()
|
||
|
|
case ch.length <- ch.buffer.Length():
|
||
|
|
}
|
||
|
|
|
||
|
|
if ch.buffer.Length() > 0 {
|
||
|
|
output = ch.output
|
||
|
|
next = ch.buffer.Peek()
|
||
|
|
} else {
|
||
|
|
output = nil
|
||
|
|
next = nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
close(ch.output)
|
||
|
|
close(ch.length)
|
||
|
|
}
|