connect/repositories/firebase_messaging_repository.go

package repositories

import (
	"context"

	"gorm.io/gorm"
	"sikey.com/websocket/models"
)

type FirebaseMessageRepository interface {
	GetFirebaseToken(ctx context.Context, uid string) (*models.FirebaseMessageToken, error)
}

var _ FirebaseMessageRepository = (*firebaseMessageRepository)(nil)

type firebaseMessageRepository struct {
	source *gorm.DB
}

// GetFirebaseToken implements FirebaseMessageRepository.
func (repo *firebaseMessageRepository) GetFirebaseToken(ctx context.Context, uid string) (*models.FirebaseMessageToken, error) {
	var err error
	var token models.FirebaseMessageToken
	err = repo.source.WithContext(ctx).Where(&models.FirebaseMessageToken{AccountId: uid}).First(&token).Error
	return &token, err
}

func NewFirebaseMessageRepository(source *gorm.DB) FirebaseMessageRepository {
	return &firebaseMessageRepository{source: source}
}