add bulk with go routines
This commit is contained in:
parent
56a7e26248
commit
52d3ad065b
9 changed files with 324 additions and 127 deletions
84
sonic/connection.go
Normal file
84
sonic/connection.go
Normal 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
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue