Fix adding more then one callback for same eventcode, grow slice dynamically.

This commit is contained in:
tj 2010-10-09 15:11:28 +02:00
parent abc18957ff
commit fd854c475a

View File

@ -7,29 +7,44 @@ import (
"strconv" "strconv"
) )
func AppendCallback(slice, data []func(*IRCEvent)) []func(*IRCEvent) {
l := len(slice)
if l+len(data) > cap(slice) {
newSlice := make([]func(*IRCEvent), (l+len(data))*2)
copy(newSlice, slice)
slice = newSlice
}
slice = slice[0 : l+len(data)]
for i, c := range data {
slice[l+i] = c
}
return slice
}
func (irc *IRCConnection) AddCallback(eventcode string, callback func(*IRCEvent)) { func (irc *IRCConnection) AddCallback(eventcode string, callback func(*IRCEvent)) {
eventcode = strings.ToUpper(eventcode) eventcode = strings.ToUpper(eventcode)
if event, ok := irc.events[eventcode]; ok { if event, ok := irc.events[eventcode]; ok {
// TODO: Grow this dynamically newevent := make([]func(*IRCEvent), 1)
event = event[0 : len(event)+1] newevent[0] = callback
event[len(event)-1] = callback irc.events[eventcode] = AppendCallback(event, newevent)
} else { } else {
event = make([]func(*IRCEvent), 1, 20) event = make([]func(*IRCEvent), 1)
event[0] = callback event[0] = callback
irc.events[eventcode] = event irc.events[eventcode] = event
} }
} }
func (irc *IRCConnection) ReplaceCallback(eventcode string, i uint8, callback func(*IRCEvent)) { func (irc *IRCConnection) ReplaceCallback(eventcode string, i int, callback func(*IRCEvent)) {
eventcode = strings.ToUpper(eventcode) eventcode = strings.ToUpper(eventcode)
if event, ok := irc.events[eventcode]; ok { if event, ok := irc.events[eventcode]; ok {
event[i] = callback if i < len(event) {
} else { event[i] = callback
event = make([]func(*IRCEvent), 1, 20) return
event[0] = callback }
irc.events[eventcode] = event fmt.Printf("Event found, but no callback found at index %d. Use AddCallback\n", i)
return
} }
fmt.Printf("Event not found. Use AddCallBack\n")
} }
func (irc *IRCConnection) RunCallbacks(event *IRCEvent) { func (irc *IRCConnection) RunCallbacks(event *IRCEvent) {
@ -92,7 +107,7 @@ func (irc *IRCConnection) setupCallbacks() {
irc.AddCallback("433", func(e *IRCEvent) { irc.AddCallback("433", func(e *IRCEvent) {
if len(irc.nick) > 8 { if len(irc.nick) > 8 {
irc.nick = "_" + irc.nick; irc.nick = "_" + irc.nick
} else { } else {
irc.nick = irc.nick + "_" irc.nick = irc.nick + "_"
} }