e2m/dispatch
4
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Prevent duplicate users in ChannelStore

Browse Source
This commit is contained in:
Ken-Håvard Lieng 2016-01-22 04:45:41 +01:00
parent 924bd8fdb1
commit a3a934fff2
2 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
storage/channel.go
View File

@ -51,6 +51,13 @@ func (c *ChannelStore) AddUser(user, server, channel string) {
} }
if users, ok := c.users[server][channel]; ok { if users, ok := c.users[server][channel]; ok {
for _, u := range users {
if u == user {
c.userLock.Unlock()
return
}
}
c.users[server][channel] = append(users, user) c.users[server][channel] = append(users, user)
} else { } else {
c.users[server][channel] = []string{user} c.users[server][channel] = []string{user}

2
storage/channel_test.go
View File

@ -16,6 +16,8 @@ func TestGetSetUsers(t *testing.T) {
func TestAddRemoveUser(t *testing.T) { func TestAddRemoveUser(t *testing.T) {
channelStore := NewChannelStore() channelStore := NewChannelStore()
channelStore.AddUser("user", "srv", "#chan") channelStore.AddUser("user", "srv", "#chan")
channelStore.AddUser("user", "srv", "#chan")
assert.Len(t, channelStore.GetUsers("srv", "#chan"), 1)
channelStore.AddUser("user2", "srv", "#chan") channelStore.AddUser("user2", "srv", "#chan")
assert.Equal(t, []string{"user", "user2"}, channelStore.GetUsers("srv", "#chan")) assert.Equal(t, []string{"user", "user2"}, channelStore.GetUsers("srv", "#chan"))
channelStore.RemoveUser("user", "srv", "#chan") channelStore.RemoveUser("user", "srv", "#chan")