mirror of
https://github.com/ergochat/ergo.git
synced 2024-11-10 22:19:31 +01:00
use a Scanner instead of ReadString
This commit is contained in:
parent
d696f2313e
commit
2dc69c7e3d
@ -4,7 +4,6 @@ import (
|
||||
"bufio"
|
||||
"io"
|
||||
"net"
|
||||
"strings"
|
||||
)
|
||||
|
||||
const (
|
||||
@ -15,14 +14,12 @@ const (
|
||||
|
||||
type Socket struct {
|
||||
conn net.Conn
|
||||
reader *bufio.Reader
|
||||
writer *bufio.Writer
|
||||
}
|
||||
|
||||
func NewSocket(conn net.Conn, commands chan<- Command) *Socket {
|
||||
socket := &Socket{
|
||||
conn: conn,
|
||||
reader: bufio.NewReader(conn),
|
||||
writer: bufio.NewWriter(conn),
|
||||
}
|
||||
|
||||
@ -41,12 +38,9 @@ func (socket *Socket) Close() {
|
||||
}
|
||||
|
||||
func (socket *Socket) readLines(commands chan<- Command) {
|
||||
for {
|
||||
line, err := socket.reader.ReadString('\n')
|
||||
if socket.isError(err, R) {
|
||||
break
|
||||
}
|
||||
line = strings.TrimRight(line, CRLF)
|
||||
scanner := bufio.NewScanner(socket.conn)
|
||||
for scanner.Scan() {
|
||||
line := scanner.Text()
|
||||
if len(line) == 0 {
|
||||
continue
|
||||
}
|
||||
@ -60,6 +54,10 @@ func (socket *Socket) readLines(commands chan<- Command) {
|
||||
commands <- msg
|
||||
}
|
||||
|
||||
if err := scanner.Err(); err != nil {
|
||||
Log.debug.Printf("%s error: %s", socket, err)
|
||||
}
|
||||
|
||||
close(commands)
|
||||
}
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user