Как я могу заставить net.Read ждать ввода в golang? - PullRequest
Новая
       19

Как я могу заставить net.Read ждать ввода в golang?

9 голосов
/ 13 марта 2012

Итак, я делаю сервер для своего лифта в Go и запускаю функцию «обработчик» в виде программы с TCP-соединением. Я хочу, чтобы он считывал данные из соединения, и если в течение определенного промежутка времени сигнал не обнаружен, я хочу, чтобы он возвратил ошибку. 

func handler(conn net.Conn){
    conn.SetReadTimeout(5e9)
    for{
        data := make([]byte, 512)
        _,err := conn.Read(data)
    }
}

Пока у меня есть клиент, отправляющий данные по соединению, кажется, что он работает нормально, но как только клиент перестает отправлять функцию net.Read, возвращает ошибку EOF и начинает цикл без каких-либо задержек.

Возможно, именно так должен работать Read, но кто-то может предложить другой способ решения проблемы без необходимости закрывать и открывать соединение каждый раз, когда я хочу что-то прочитать?

Ответы [ 2 ]

20 голосов
/ 19 марта 2012

Чтение работает, как и ожидалось, я думаю.Похоже, вы хотите, чтобы net.Read работал как канал в Go.Это довольно просто, просто оберните сеть. Прочтите текущую программу и используйте select для чтения из каналов. Программа действительно дешевая, как и канал

Пример: 

ch := make(chan []byte)
eCh := make(chan error)

// Start a goroutine to read from our net connection
go func(ch chan []byte, eCh chan error) {
  for {
    // try to read the data
    data := make([]byte, 512)
    _,err := conn.Read(data)
    if err != nil {
      // send an error if it's encountered
      eCh<- err
      return
    }
    // send data if we read some.
    ch<- data
  }
}(ch, eCh)

ticker := time.Tick(time.Second)
// continuously read from the connection
for {
  select {
     // This case means we recieved data on the connection
     case data := <-ch:
       // Do something with the data
     // This case means we got an error and the goroutine has finished
     case err := <-eCh:
       // handle our error then exit for loop
       break;
     // This will timeout on the read.
     case <-ticker:
       // do nothing? this is just so we can time out if we need to.
       // you probably don't even need to have this here unless you want
       // do something specifically on the timeout.
  }
}
0 голосов
/ 13 марта 2012

Возможно, что сборщик мусора (GC) закрывает один конец соединения, потому что net.Conn становится недоступным объектом с точки зрения GC. Чтобы предотвратить это: после того, как клиент прекратит отправку, убедитесь, что клиентское соединение хранится в переменной, доступной для вашего кода.

Другой вариант заключается в том, что где-то в вашей программе вы закрываете один конец соединения, не зная об этом.

...