2016-05-23 11:19:25 +00:00
|
|
|
package main
|
|
|
|
|
|
|
|
import "strings"
|
|
|
|
|
|
|
|
type messageChanWriter struct {
|
|
|
|
msgChan chan string
|
|
|
|
|
|
|
|
buffer []byte
|
|
|
|
}
|
|
|
|
|
2016-05-23 14:04:50 +00:00
|
|
|
func newMessageChanWriter(outputChannel chan string) *messageChanWriter {
|
2016-05-23 11:19:25 +00:00
|
|
|
return &messageChanWriter{
|
|
|
|
msgChan: outputChannel,
|
|
|
|
buffer: []byte{},
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (m *messageChanWriter) Write(p []byte) (n int, err error) {
|
|
|
|
n = len(p)
|
|
|
|
err = nil
|
|
|
|
|
|
|
|
m.buffer = append(m.buffer, p...)
|
|
|
|
if strings.Contains(string(m.buffer), "\n") {
|
|
|
|
lines := strings.Split(string(m.buffer), "\n")
|
|
|
|
for _, l := range lines[:len(lines)-1] {
|
|
|
|
m.msgChan <- l
|
|
|
|
}
|
|
|
|
m.buffer = []byte(lines[len(lines)-1])
|
|
|
|
}
|
|
|
|
|
|
|
|
return
|
|
|
|
}
|