mirror of
https://github.com/buger/goreplay.git
synced 2024-04-21 12:32:02 +00:00
Urban Ishimwe
6d812ceb7f
// PluginReader is an interface for input plugins
type PluginReader interface {
PluginRead() (msg *Message, err error)
}
// PluginWriter is an interface for output plugins
type PluginWriter interface {
PluginWrite(msg *Message) (n int, err error)
}
28 lines
591 B
Go
28 lines
591 B
Go
package main
|
|
|
|
type writeCallback func(*Message)
|
|
|
|
// TestOutput used in testing to intercept any output into callback
|
|
type TestOutput struct {
|
|
cb writeCallback
|
|
}
|
|
|
|
// NewTestOutput constructor for TestOutput, accepts callback which get called on each incoming Write
|
|
func NewTestOutput(cb writeCallback) PluginWriter {
|
|
i := new(TestOutput)
|
|
i.cb = cb
|
|
|
|
return i
|
|
}
|
|
|
|
// PluginWrite write message to this plugin
|
|
func (i *TestOutput) PluginWrite(msg *Message) (int, error) {
|
|
i.cb(msg)
|
|
|
|
return len(msg.Data) + len(msg.Meta), nil
|
|
}
|
|
|
|
func (i *TestOutput) String() string {
|
|
return "Test Output"
|
|
}
|