channels.go

package uim

import (
	"sync"

	"github.com/uim/logger"
)

type ChannelMap interface {
	Add(channel Channel)
	Remove(id string)
	Get(id string) (Channel, bool)
	All() []Channel
}

type ChannelsImpl struct {
	channels *sync.Map
}

var _ ChannelMap = (*ChannelsImpl)(nil)

func NewChannels(num int) ChannelMap {
	return &ChannelsImpl{
		channels: new(sync.Map),
	}
}

func (ch *ChannelsImpl) Add(channel Channel) {
	if channel.ID() == "" {
		logger.WithFields(logger.Fields{
			"modules": "ChannelsImpl",
		})
	}

	ch.channels.Store(channel.ID(), channel)
}

func (ch *ChannelsImpl) Remove(id string) {
	ch.channels.Delete(id)
}

func (ch *ChannelsImpl) Get(id string) (Channel, bool) {
	if id == "" {
		logger.WithFields(logger.Fields{
			"module": "ChannelsImpl",
		})
	}
	val, ok := ch.channels.Load(id)
	if !ok {
		return nil, false
	}
	return val.(Channel), true
}

func (ch *ChannelsImpl) All() []Channel {
	arr := make([]Channel, 0)
	ch.channels.Range(func(key, val any) bool {
		arr = append(arr, val.(Channel))
		return true
	})
	return arr
}