add bulk with go routines

This commit is contained in:
alexisvisco 2019-03-27 15:22:27 +01:00
parent 56a7e26248
commit 52d3ad065b
9 changed files with 324 additions and 127 deletions

84
sonic/connection.go Normal file
View file

@ -0,0 +1,84 @@
package sonic
import (
"bufio"
"bytes"
"errors"
"fmt"
"io"
"net"
"strings"
)
type connection struct {
reader *bufio.Reader
conn net.Conn
closed bool
}
func newConnection(d *driver) (*connection, error) {
c := &connection{}
c.close()
conn, err := net.Dial("tcp", fmt.Sprintf("%s:%d", d.Host, d.Port))
if err != nil {
return nil, err
}
c.closed = false
c.conn = conn
c.reader = bufio.NewReader(c.conn)
err = c.write(fmt.Sprintf("START %s %s", d.channel, d.Password))
if err != nil {
return nil, err
}
_, err = c.read()
_, err = c.read()
if err != nil {
return nil, err
}
return c, nil
}
func (c *connection) read() (string, error) {
if c.closed {
return "", ErrClosed
}
buffer := bytes.Buffer{}
for {
line, isPrefix, err := c.reader.ReadLine()
buffer.Write(line)
if err != nil {
if err == io.EOF {
c.close()
}
return "", err
}
if !isPrefix {
break
}
}
str := buffer.String()
if strings.HasPrefix(str, "ERR ") {
return "", errors.New(str[4:])
}
return str, nil
}
func (c connection) write(str string) error {
if c.closed {
return ErrClosed
}
_, err := c.conn.Write([]byte(str + "\r\n"))
return err
}
func (c *connection) close() {
if c.conn != nil {
_ = c.conn.Close()
c.conn = nil
}
c.closed = true
c.reader = nil
}