Change the way the pinger works.

Fix syning threads before reconnections
Formatting
This commit is contained in:
Thomas Jager 2012-11-07 21:51:24 +01:00
parent d53de8bdbd
commit c8f78ebf45
3 changed files with 37 additions and 29 deletions

@ -51,6 +51,7 @@ Commands
ircobj.Password = "[server password]" ircobj.Password = "[server password]"
ircobj.Connect("irc.someserver.com:6667") //Connect to server ircobj.Connect("irc.someserver.com:6667") //Connect to server
ircobj.Sendraw("<string>") //sends string to server. Adds \r\n ircobj.Sendraw("<string>") //sends string to server. Adds \r\n
ircobj.Sendraw("<formatstring>", ...) //sends formatted string to server.n
ircobj.Join("#channel [password]") ircobj.Join("#channel [password]")
ircobj.Privmsg("#channel", "msg") ircobj.Privmsg("#channel", "msg")
ircobj.Privmsg("nickname", "msg") ircobj.Privmsg("nickname", "msg")

56
irc.go

@ -6,28 +6,26 @@ package irc
import ( import (
"bufio" "bufio"
"crypto/tls"
"fmt" "fmt"
"log" "log"
"net" "net"
"os" "os"
"strings" "strings"
"time" "time"
"crypto/tls"
) )
const ( const (
VERSION = "cleanirc v1.0" VERSION = "go-ircevent v2.0"
) )
var error_ bool
func (irc *Connection) readLoop() { func (irc *Connection) readLoop() {
br := bufio.NewReader(irc.socket) br := bufio.NewReader(irc.socket)
for !irc.reconnecting { for {
msg, err := br.ReadString('\n') msg, err := br.ReadString('\n')
if err != nil { if err != nil {
irc.Error <-err irc.Error <- err
break break
} }
@ -66,16 +64,15 @@ func (irc *Connection) readLoop() {
irc.RunCallbacks(event) irc.RunCallbacks(event)
} }
irc.syncreader <-true irc.syncreader <- true
} }
func (irc *Connection) writeLoop() { func (irc *Connection) writeLoop() {
b, ok := <-irc.pwrite b, ok := <-irc.pwrite
for !irc.reconnecting && ok { for ok {
if b == "" || irc.socket == nil { if b == "" || irc.socket == nil {
break break
} }
_, err := irc.socket.Write([]byte(b)) _, err := irc.socket.Write([]byte(b))
if err != nil { if err != nil {
irc.log.Printf("%s\n", err) irc.log.Printf("%s\n", err)
@ -85,21 +82,21 @@ func (irc *Connection) writeLoop() {
b, ok = <-irc.pwrite b, ok = <-irc.pwrite
} }
irc.syncwriter <-true irc.syncwriter <- true
} }
//Pings the server if we have not recived any messages for 5 minutes //Pings the server if we have not recived any messages for 5 minutes
func (irc *Connection) pingLoop() { func (irc *Connection) pingLoop() {
irc.ticker = time.Tick(1 * time.Minute) //Tick every minute. irc.ticker = time.NewTicker(1 * time.Minute) //Tick every minute.
irc.ticker2 = time.Tick(15 * time.Minute) //Tick every 15 minutes. irc.ticker2 = time.NewTicker(15 * time.Minute) //Tick every 15 minutes.
for { for {
select { select {
case <-irc.ticker: case <-irc.ticker.C:
//Ping if we haven't recived anything from the server within 4 minutes //Ping if we haven't recived anything from the server within 4 minutes
if time.Since(irc.lastMessage) >= (4 * time.Minute) { if time.Since(irc.lastMessage) >= (4 * time.Minute) {
irc.SendRawf("PING %d", time.Now().UnixNano()) irc.SendRawf("PING %d", time.Now().UnixNano())
} }
case <-irc.ticker2: case <-irc.ticker2.C:
//Ping every 15 minutes. //Ping every 15 minutes.
irc.SendRawf("PING %d", time.Now().UnixNano()) irc.SendRawf("PING %d", time.Now().UnixNano())
//Try to recapture nickname if it's not as configured. //Try to recapture nickname if it's not as configured.
@ -107,8 +104,13 @@ func (irc *Connection) pingLoop() {
irc.nickcurrent = irc.nick irc.nickcurrent = irc.nick
irc.SendRawf("NICK %s", irc.nick) irc.SendRawf("NICK %s", irc.nick)
} }
case <-irc.endping:
irc.ticker.Stop()
irc.ticker2.Stop()
break
} }
} }
irc.syncpinger <- true
} }
func (irc *Connection) Cycle() { func (irc *Connection) Cycle() {
@ -143,18 +145,25 @@ func (irc *Connection) SendRaw(message string) {
} }
func (irc *Connection) SendRawf(format string, a ...interface{}) { func (irc *Connection) SendRawf(format string, a ...interface{}) {
irc.SendRaw(fmt.Sprintf(format, a...)) irc.SendRaw(fmt.Sprintf(format, a...))
} }
func (irc *Connection) GetNick() string { func (irc *Connection) GetNick() string {
return irc.nickcurrent return irc.nickcurrent
} }
func (irc *Connection) Reconnect() error { func (irc *Connection) Reconnect() error {
close(irc.pwrite) close(irc.pwrite)
close(irc.pread) close(irc.pread)
irc.endping <- true
irc.log.Printf("Syncing Threads\n")
irc.log.Printf("Syncing Reader\n")
<-irc.syncreader <-irc.syncreader
irc.log.Printf("Syncing Writer\n")
<-irc.syncwriter <-irc.syncwriter
irc.log.Printf("Syncing Pinger\n")
<-irc.syncpinger
irc.log.Printf("Syncing Threads Done\n")
for { for {
irc.log.Printf("Reconnecting to %s\n", irc.server) irc.log.Printf("Reconnecting to %s\n", irc.server)
var err error var err error
@ -164,7 +173,6 @@ func (irc *Connection) Reconnect() error {
} }
irc.log.Printf("Error: %s\n", err) irc.log.Printf("Error: %s\n", err)
} }
error_ = false
return nil return nil
} }
@ -175,18 +183,17 @@ func (irc *Connection) Loop() {
break break
} }
irc.log.Printf("Error: %s\n", e) irc.log.Printf("Error: %s\n", e)
error_ = true
irc.Reconnect() irc.Reconnect()
} }
close(irc.pwrite) close(irc.pwrite)
close(irc.pread) close(irc.pread)
irc.endping <- true
<-irc.syncreader <-irc.syncreader
<-irc.syncwriter <-irc.syncwriter
<-irc.syncpinger
} }
func (irc *Connection) Connect(server string) error { func (irc *Connection) Connect(server string) error {
irc.server = server irc.server = server
var err error var err error
@ -209,16 +216,17 @@ func (irc *Connection) postConnect() error {
irc.Error = make(chan error, 10) irc.Error = make(chan error, 10)
irc.syncreader = make(chan bool) irc.syncreader = make(chan bool)
irc.syncwriter = make(chan bool) irc.syncwriter = make(chan bool)
irc.syncpinger = make(chan bool)
irc.endping = make(chan bool)
go irc.readLoop() go irc.readLoop()
go irc.writeLoop() go irc.writeLoop()
go irc.pingLoop() go irc.pingLoop()
if len(irc.Password) > 0 { if len(irc.Password) > 0 {
irc.pwrite <-fmt.Sprintf("PASS %s\r\n", irc.Password) irc.pwrite <- fmt.Sprintf("PASS %s\r\n", irc.Password)
} }
irc.pwrite <-fmt.Sprintf("NICK %s\r\n", irc.nick) irc.pwrite <- fmt.Sprintf("NICK %s\r\n", irc.nick)
irc.pwrite <-fmt.Sprintf("USER %s 0.0.0.0 0.0.0.0 :%s\r\n", irc.user, irc.user) irc.pwrite <- fmt.Sprintf("USER %s 0.0.0.0 0.0.0.0 :%s\r\n", irc.user, irc.user)
return nil return nil
} }

@ -20,8 +20,8 @@ type Connection struct {
socket net.Conn socket net.Conn
pread, pwrite chan string pread, pwrite chan string
syncreader, syncwriter chan bool syncreader, syncwriter, syncpinger chan bool
reconnecting bool endping chan bool
nick string //The nickname we want. nick string //The nickname we want.
nickcurrent string //The nickname we currently have. nickcurrent string //The nickname we currently have.
@ -30,9 +30,8 @@ type Connection struct {
server string server string
events map[string][]func(*Event) events map[string][]func(*Event)
lastMessage time.Time lastMessage time.Time
ticker <-chan time.Time ticker, ticker2 *time.Ticker
ticker2 <-chan time.Time
VerboseCallbackHandler bool VerboseCallbackHandler bool
log *log.Logger log *log.Logger