Как в Go прочитать несколько строк не разрывая соединения?

Question

[Арсений Соколов]

Есть вот такой код:

func main() {
  ln,  _ := net.Listen("tcp", ":6556")
  defer ln.Close()
  for {
    con, _ := ln.Accept()
    go handleServer(conn)
  }
}

func handleServer(conn net.Conn) {
  buf := make([]byte, 0, 4096)
  n, _ := conn.Read(buf)
  log.Printf("Received line length: %d\n", n)
  conn.Write([]byte("Message received.\n"))
  conn.Close()
}

Вот если в функции handleServer() убрать conn.Close(), тогда соединение не разрывается, но и больше первой строки ничего не читает, однако, как я думал должно было читать каждую строку.

Каким образом можно читать все строки поочерёдно?

Answer 1

[Александр Павлюк]

Вот, пример кода. Здесь мы читаем входящие данные построчно и посылаем "Message received" после прочтения каждой строки. Чтение продолжается пока клиент не закроет соединение.

package main

import (
	"bufio"
	"log"
	"net"
)

func main() {
	ln, _ := net.Listen("tcp", ":6556")
	defer ln.Close()
	for {
		conn, _ := ln.Accept()
		go handleServer(conn)
	}
}

func handleServer(conn net.Conn) {
	reader := bufio.NewReader(conn)
	defer conn.Close()

	for {
		line, err := reader.ReadString('\n')
		if err != nil {
			return
		}
		conn.Write([]byte("Message received.\n"))
		log.Printf("Received line length: %d\n", len(line))
	}
}

Comments

[Арсений Соколов]

Спасибо! То что доктор прописал.

[Oleg Shevelev]

Если соединение одно то зачем делать это:

for {
    conn, _ := ln.Accept()
    go handleServer(conn)
}

Так же не совсем ясно зачем там не контролиуремое количество горутин.

[Александр Павлюк]

Oleg Shevelev: эту часть я не трогал, она не относилась к вопросу. Где условие насчет того, что соединение одно? По-вашему, сервер должен принять одно соединение и закрыться?

[Oleg Shevelev]

Александр Павлюк: тупанул:) Сплю уже:)