Compare commits
3 Commits
08ae30b305
...
e582fa42bb
| Author | SHA1 | Date | |
|---|---|---|---|
| e582fa42bb | |||
| 62b05dcf0f | |||
| 1b02a8181a |
4
.gitignore
vendored
4
.gitignore
vendored
|
|
@ -3,7 +3,7 @@ cmd/ek/ek
|
||||||
.idea
|
.idea
|
||||||
.vscode
|
.vscode
|
||||||
|
|
||||||
ek
|
/ek
|
||||||
eksterd
|
/eksterd
|
||||||
|
|
||||||
backend.json
|
backend.json
|
||||||
|
|
|
||||||
|
|
@ -215,10 +215,15 @@ func (b *memoryBackend) ChannelsGetList() ([]microsub.Channel, error) {
|
||||||
|
|
||||||
// ChannelsCreate creates a channels
|
// ChannelsCreate creates a channels
|
||||||
func (b *memoryBackend) ChannelsCreate(name string) (microsub.Channel, error) {
|
func (b *memoryBackend) ChannelsCreate(name string) (microsub.Channel, error) {
|
||||||
defer b.save()
|
// Try to fetch the channel, if it exists, we don't create it
|
||||||
|
if channel, e := b.fetchChannel(name); e {
|
||||||
|
return channel, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Otherwise create the channel
|
||||||
channel := b.createChannel(name)
|
channel := b.createChannel(name)
|
||||||
b.setChannel(channel)
|
b.setChannel(channel)
|
||||||
|
b.save()
|
||||||
|
|
||||||
conn := b.pool.Get()
|
conn := b.pool.Get()
|
||||||
defer conn.Close()
|
defer conn.Close()
|
||||||
|
|
@ -834,9 +839,23 @@ func (b *memoryBackend) createChannel(name string) microsub.Channel {
|
||||||
return channel
|
return channel
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (b *memoryBackend) fetchChannel(name string) (microsub.Channel, bool) {
|
||||||
|
b.lock.RLock()
|
||||||
|
defer b.lock.RUnlock()
|
||||||
|
|
||||||
|
for _, c := range b.Channels {
|
||||||
|
if c.Name == name {
|
||||||
|
return c, true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return microsub.Channel{}, false
|
||||||
|
}
|
||||||
|
|
||||||
func (b *memoryBackend) setChannel(channel microsub.Channel) {
|
func (b *memoryBackend) setChannel(channel microsub.Channel) {
|
||||||
b.lock.Lock()
|
b.lock.Lock()
|
||||||
defer b.lock.Unlock()
|
defer b.lock.Unlock()
|
||||||
|
|
||||||
b.Channels[channel.UID] = channel
|
b.Channels[channel.UID] = channel
|
||||||
b.Feeds[channel.UID] = []microsub.Feed{}
|
b.Feeds[channel.UID] = []microsub.Feed{}
|
||||||
b.NextUID++
|
b.NextUID++
|
||||||
|
|
|
||||||
114
cmd/eksterd/memory_test.go
Normal file
114
cmd/eksterd/memory_test.go
Normal file
|
|
@ -0,0 +1,114 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"reflect"
|
||||||
|
"sync"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/gomodule/redigo/redis"
|
||||||
|
"p83.nl/go/ekster/pkg/microsub"
|
||||||
|
"p83.nl/go/ekster/pkg/sse"
|
||||||
|
)
|
||||||
|
|
||||||
|
func Test_memoryBackend_ChannelsCreate(t *testing.T) {
|
||||||
|
type fields struct {
|
||||||
|
hubIncomingBackend hubIncomingBackend
|
||||||
|
lock sync.RWMutex
|
||||||
|
Channels map[string]microsub.Channel
|
||||||
|
Feeds map[string][]microsub.Feed
|
||||||
|
Settings map[string]channelSetting
|
||||||
|
NextUID int
|
||||||
|
Me string
|
||||||
|
TokenEndpoint string
|
||||||
|
AuthEnabled bool
|
||||||
|
ticker *time.Ticker
|
||||||
|
quit chan struct{}
|
||||||
|
broker *sse.Broker
|
||||||
|
pool *redis.Pool
|
||||||
|
}
|
||||||
|
type args struct {
|
||||||
|
name string
|
||||||
|
}
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
fields fields
|
||||||
|
args args
|
||||||
|
want microsub.Channel
|
||||||
|
wantErr bool
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "Duplicate channel",
|
||||||
|
fields: fields{
|
||||||
|
hubIncomingBackend: hubIncomingBackend{},
|
||||||
|
lock: sync.RWMutex{},
|
||||||
|
Channels: func() map[string]microsub.Channel {
|
||||||
|
channels := make(map[string]microsub.Channel)
|
||||||
|
channels["1234"] = microsub.Channel{
|
||||||
|
UID: "1234",
|
||||||
|
Name: "Test",
|
||||||
|
Unread: microsub.Unread{
|
||||||
|
Type: microsub.UnreadCount,
|
||||||
|
Unread: false,
|
||||||
|
UnreadCount: 0,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
return channels
|
||||||
|
}(),
|
||||||
|
Feeds: func() map[string][]microsub.Feed {
|
||||||
|
feeds := make(map[string][]microsub.Feed)
|
||||||
|
return feeds
|
||||||
|
}(),
|
||||||
|
Settings: nil,
|
||||||
|
NextUID: 1,
|
||||||
|
Me: "",
|
||||||
|
TokenEndpoint: "",
|
||||||
|
AuthEnabled: false,
|
||||||
|
ticker: nil,
|
||||||
|
quit: nil,
|
||||||
|
broker: nil,
|
||||||
|
pool: nil,
|
||||||
|
},
|
||||||
|
args: args{
|
||||||
|
name: "Test",
|
||||||
|
},
|
||||||
|
want: microsub.Channel{
|
||||||
|
UID: "1234",
|
||||||
|
Name: "Test",
|
||||||
|
Unread: microsub.Unread{
|
||||||
|
Type: microsub.UnreadCount,
|
||||||
|
Unread: false,
|
||||||
|
UnreadCount: 0,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
wantErr: false,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
b := &memoryBackend{
|
||||||
|
hubIncomingBackend: tt.fields.hubIncomingBackend,
|
||||||
|
lock: tt.fields.lock,
|
||||||
|
Channels: tt.fields.Channels,
|
||||||
|
Feeds: tt.fields.Feeds,
|
||||||
|
Settings: tt.fields.Settings,
|
||||||
|
NextUID: tt.fields.NextUID,
|
||||||
|
Me: tt.fields.Me,
|
||||||
|
TokenEndpoint: tt.fields.TokenEndpoint,
|
||||||
|
AuthEnabled: tt.fields.AuthEnabled,
|
||||||
|
ticker: tt.fields.ticker,
|
||||||
|
quit: tt.fields.quit,
|
||||||
|
broker: tt.fields.broker,
|
||||||
|
pool: tt.fields.pool,
|
||||||
|
}
|
||||||
|
got, err := b.ChannelsCreate(tt.args.name)
|
||||||
|
if (err != nil) != tt.wantErr {
|
||||||
|
t.Errorf("ChannelsCreate() error = %v, wantErr %v", err, tt.wantErr)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if !reflect.DeepEqual(got, tt.want) {
|
||||||
|
t.Errorf("ChannelsCreate() got = %v, want %v", got, tt.want)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue
Block a user