api/database/postgres/channel.go

package postgres

import (
	"context"
	"database/sql"
	"git.aiterp.net/rpdata/api/database/postgres/psqlcore"
	"git.aiterp.net/rpdata/api/models"
)

type channelRepository struct {
	insertWithIDs bool
	db            *sql.DB
}

func (r *channelRepository) Find(ctx context.Context, name string) (*models.Channel, error) {
	row, err := psqlcore.New(r.db).SelectChannelByName(ctx, name)
	if err != nil {
		return nil, err
	}

	return r.channel(row), nil
}

func (r *channelRepository) List(ctx context.Context, filter models.ChannelFilter) ([]*models.Channel, error) {
	params := psqlcore.SelectChannelsParams{
		LimitSize: 1000,
	}

	if filter.Names != nil {
		params.FilterName = true
		params.Names = filter.Names
	}
	if filter.LocationName != nil {
		params.FilterLocationName = true
		params.LocationName = *filter.LocationName
	}
	if filter.EventName != nil {
		params.FilterEventName = true
		params.EventName = *filter.EventName
	}
	if filter.Limit > 0 {
		params.LimitSize = int32(filter.Limit)
	}

	rows, err := psqlcore.New(r.db).SelectChannels(ctx, params)
	if err != nil {
		return nil, err
	}

	return r.channels(rows), nil
}

func (r *channelRepository) Insert(ctx context.Context, channel models.Channel) (*models.Channel, error) {
	err := psqlcore.New(r.db).InsertChannel(ctx, psqlcore.InsertChannelParams(channel))
	if err != nil {
		return nil, err
	}

	return &channel, nil
}

func (r *channelRepository) Update(ctx context.Context, channel models.Channel, update models.ChannelUpdate) (*models.Channel, error) {
	channel.ApplyUpdate(update)

	err := psqlcore.New(r.db).UpdateChannel(ctx, psqlcore.UpdateChannelParams{
		Name:         channel.Name,
		Logged:       channel.Logged,
		Hub:          channel.Hub,
		EventName:    channel.EventName,
		LocationName: channel.LocationName,
	})
	if err != nil {
		return nil, err
	}

	return &channel, nil
}

func (r *channelRepository) Remove(ctx context.Context, channel models.Channel) error {
	return psqlcore.New(r.db).DeleteChannel(ctx, channel.Name)
}

func (r *channelRepository) channel(row psqlcore.DataChannel) *models.Channel {
	return &models.Channel{
		Name:         row.Name,
		Logged:       row.Logged,
		Hub:          row.Hub,
		EventName:    row.EventName,
		LocationName: row.LocationName,
	}
}

func (r *channelRepository) channels(rows []psqlcore.DataChannel) []*models.Channel {
	results := make([]*models.Channel, 0, len(rows))
	for _, row := range rows {
		results = append(results, r.channel(row))
	}

	return results
}