34 lines
509 B
Go
34 lines
509 B
Go
|
package polling
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"io"
|
||
|
|
)
|
||
|
|
|
||
|
|
func MakeSendChan() chan bool {
|
||
|
|
return make(chan bool, 1)
|
||
|
|
}
|
||
|
|
|
||
|
|
type Writer struct {
|
||
|
|
io.WriteCloser
|
||
|
|
server *Polling
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewWriter(w io.WriteCloser, server *Polling) *Writer {
|
||
|
|
return &Writer{
|
||
|
|
WriteCloser: w,
|
||
|
|
server: server,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (w *Writer) Close() error {
|
||
|
|
if w.server.getState() != stateNormal {
|
||
|
|
return errors.New("use of closed network connection")
|
||
|
|
}
|
||
|
|
select {
|
||
|
|
case w.server.sendChan <- true:
|
||
|
|
default:
|
||
|
|
}
|
||
|
|
return w.WriteCloser.Close()
|
||
|
|
}
|