Implement Go bindings for pipe io client
This commit is contained in:
parent
61eb57c918
commit
deafc516bd
101
clients/go/pipeio/pipeio.go
Normal file
101
clients/go/pipeio/pipeio.go
Normal file
@ -0,0 +1,101 @@
|
||||
package pipeio
|
||||
|
||||
import (
|
||||
"os"
|
||||
"bufio"
|
||||
)
|
||||
|
||||
|
||||
type pipeReader struct {
|
||||
pipe *os.File
|
||||
scanner *bufio.Scanner
|
||||
}
|
||||
|
||||
func NewPipeReader(pipePath string) (*pipeReader, error) {
|
||||
pipe, err := os.Open(pipePath)
|
||||
scanner := bufio.NewScanner(pipe)
|
||||
scanner.Split(bufio.ScanLines)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &pipeReader{
|
||||
pipe: pipe,
|
||||
scanner: scanner,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (preader *pipeReader) RecvMessage() []byte {
|
||||
if preader.scanner.Scan() {
|
||||
return preader.scanner.Bytes()
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (preader *pipeReader) Close() {
|
||||
preader.pipe.Close()
|
||||
}
|
||||
|
||||
|
||||
type pipeWriter struct {
|
||||
pipe *os.File
|
||||
}
|
||||
|
||||
func NewPipeWriter(pipePath string) (*pipeWriter, error) {
|
||||
pipe, err := os.OpenFile(pipePath, os.O_WRONLY, 0644)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &pipeWriter{pipe: pipe}, nil
|
||||
}
|
||||
|
||||
func (pwriter *pipeWriter) Close() {
|
||||
pwriter.pipe.Close()
|
||||
}
|
||||
|
||||
func (pwriter *pipeWriter) SendMessage(msg []byte) {
|
||||
pwriter.pipe.Write(append(msg, '\n'))
|
||||
}
|
||||
|
||||
|
||||
type pipeIO struct {
|
||||
Reader pipeReader
|
||||
Writer pipeWriter
|
||||
handleMessage handleMessageFunc
|
||||
}
|
||||
|
||||
type handleMessageFunc func([]byte)
|
||||
|
||||
func NewPipeIO(inPipePath, outPipePath string) (*pipeIO, error) {
|
||||
reader, err := NewPipeReader(inPipePath)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
writer, err := NewPipeWriter(outPipePath)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &pipeIO{
|
||||
Reader: *reader,
|
||||
Writer: *writer,
|
||||
handleMessage: func([]byte){},
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (pipeio *pipeIO) Close() {
|
||||
pipeio.Reader.Close()
|
||||
pipeio.Writer.Close()
|
||||
}
|
||||
|
||||
func (pipeio *pipeIO) Run() {
|
||||
for {
|
||||
msg := pipeio.Reader.RecvMessage()
|
||||
if msg == nil {
|
||||
break
|
||||
}
|
||||
pipeio.handleMessage(msg)
|
||||
}
|
||||
}
|
||||
|
||||
func (pipeio *pipeIO) HandleMessage(fun handleMessageFunc) {
|
||||
pipeio.handleMessage = fun
|
||||
}
|
20
clients/go/test.go
Normal file
20
clients/go/test.go
Normal file
@ -0,0 +1,20 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"log"
|
||||
"./pipeio"
|
||||
)
|
||||
|
||||
|
||||
func main() {
|
||||
pipeio, err := pipeio.NewPipeIO("in", "out")
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
defer pipeio.Close()
|
||||
|
||||
pipeio.HandleMessage(func(msg []byte) {
|
||||
pipeio.Writer.SendMessage(msg)
|
||||
})
|
||||
pipeio.Run()
|
||||
}
|
Loading…
Reference in New Issue
Block a user