X-Git-Url: http://www.git.cypherpunks.ru/?a=blobdiff_plain;f=client.go;h=c2923d0d4f3404e0a42b35353607ab26fd702ae0;hb=b640107b1fd5f1013d59e26ff0102aeb0b3f5f99;hp=fbaf8a326e8d6e3137f50e9ed2ee202013846497;hpb=66eaa48372228e098b464e8eac64d15a8582e9ef;p=goircd.git diff --git a/client.go b/client.go index fbaf8a3..c2923d0 100644 --- a/client.go +++ b/client.go @@ -46,7 +46,7 @@ type Client struct { away *string recvTimestamp time.Time sendTimestamp time.Time - outBuf chan string + outBuf chan *string alive bool sync.Mutex } @@ -76,20 +76,19 @@ func NewClient(conn net.Conn) *Client { recvTimestamp: time.Now(), sendTimestamp: time.Now(), alive: true, - outBuf: make(chan string, MaxOutBuf), + outBuf: make(chan *string, MaxOutBuf), } go c.MsgSender() return &c } func (c *Client) SetDead() { - close(c.outBuf) + c.outBuf <- nil c.alive = false } func (c *Client) Close() { c.Lock() - c.conn.Close() if c.alive { c.SetDead() } @@ -133,7 +132,11 @@ func (c *Client) Processor(sink chan ClientEvent) { func (c *Client) MsgSender() { for msg := range c.outBuf { - c.conn.Write(append([]byte(msg), CRLF...)) + if msg == nil { + c.conn.Close() + break + } + c.conn.Write(append([]byte(*msg), CRLF...)) } } @@ -146,11 +149,12 @@ func (c *Client) Msg(text string) { } if len(c.outBuf) == MaxOutBuf { log.Println(c, "output buffer size exceeded, kicking him") - go c.Close() - c.SetDead() + if c.alive { + c.SetDead() + } return } - c.outBuf <- text + c.outBuf <- &text } // Send message from server. It has ": servername" prefix.