GitHub/ergo
3
0
Fork 0
mirror of https://github.com/ergochat/ergo.git synced 2024-11-22 11:59:40 +01:00
Code Issues Projects Releases Wiki Activity

synchronize on last writes to client sockets

Browse Source
This commit is contained in:
Jeremy Latt 2014-02-19 18:46:46 -08:00
parent af6a8ee6f0
commit 9c385ededd
1 changed files with 33 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
irc/client.go
View File

@ -17,6 +17,7 @@ type Client struct {
awayMessage string
channels ChannelSet
ctime time.Time
doneWriting chan bool
flags map[UserMode]bool
hasQuit bool
hops uint
@ -27,7 +28,7 @@ type Client struct {
phase Phase
quitTimer *time.Timer
realname string
replies chan Reply
replies chan string
server *Server
socket *Socket
username string
@ -39,12 +40,13 @@ func NewClient(server *Server, conn net.Conn) *Client {
atime: now,
channels: make(ChannelSet),
ctime: now,
doneWriting: make(chan bool),
flags: make(map[UserMode]bool),
hostname: AddrLookupHostname(conn.RemoteAddr()),
phase: server.InitPhase(),
server: server,
socket: NewSocket(conn),
replies: make(chan Reply),
replies: make(chan string),
}
client.loginTimer = time.AfterFunc(LOGIN_TIMEOUT, client.connectionTimeout)
@ -94,9 +96,11 @@ func (client *Client) connectionClosed() {
//
func (client *Client) writeReplies() {
for reply := range client.replies {
client.socket.Write(reply.Format(client)...)
for line := range client.replies {
client.socket.Write(line)
}
client.socket.Close()
client.doneWriting <- true
}
//
@ -155,12 +159,13 @@ func (client *Client) Register() {
client.Touch()
}
func (client *Client) Destroy() {
func (client *Client) destroy() {
// clean up self
client.socket.Close()
close(client.replies)
<-client.doneWriting
client.loginTimer.Stop()
if client.idleTimer != nil {
client.idleTimer.Stop()
}
@ -184,7 +189,15 @@ func (client *Client) Destroy() {
}
func (client *Client) Reply(reply Reply) {
client.replies <- reply
if client.hasQuit {
if DEBUG_CLIENT {
log.Printf("%s dropping %s", client, reply)
}
return
}
for _, line := range reply.Format(client) {
client.replies <- line
}
}
func (client *Client) IdleTime() time.Duration {
@ -273,11 +286,13 @@ func (client *Client) Quit(message string) {
if client.hasQuit {
return
}
client.hasQuit = true
client.Reply(RplError(client.server, client.Nick()))
client.hasQuit = true
friends := client.Friends()
friends.Remove(client)
client.Destroy()
client.destroy()
if len(friends) > 0 {
reply := RplQuit(client, message)