mirror of
https://github.com/versia-pub/versia-go.git
synced 2025-12-06 06:28:18 +01:00
107 lines
2.8 KiB
Go
107 lines
2.8 KiB
Go
package repo_impls
|
|
|
|
import (
|
|
"context"
|
|
"github.com/versia-pub/versia-go/internal/repository"
|
|
|
|
"git.devminer.xyz/devminer/unitel"
|
|
"github.com/go-logr/logr"
|
|
"github.com/versia-pub/versia-go/ent"
|
|
"github.com/versia-pub/versia-go/internal/database"
|
|
)
|
|
|
|
type Factory[T any] func(db *ent.Client, log logr.Logger, telemetry *unitel.Telemetry) T
|
|
|
|
var _ repository.Manager = (*ManagerImpl)(nil)
|
|
|
|
type ManagerImpl struct {
|
|
users repository.UserRepository
|
|
notes repository.NoteRepository
|
|
follows repository.FollowRepository
|
|
instanceMetadata repository.InstanceMetadataRepository
|
|
|
|
uRFactory Factory[repository.UserRepository]
|
|
nRFactory Factory[repository.NoteRepository]
|
|
fRFactory Factory[repository.FollowRepository]
|
|
imRFactory Factory[repository.InstanceMetadataRepository]
|
|
|
|
db *ent.Client
|
|
log logr.Logger
|
|
telemetry *unitel.Telemetry
|
|
}
|
|
|
|
func NewManagerImpl(
|
|
db *ent.Client, telemetry *unitel.Telemetry, log logr.Logger,
|
|
userRepositoryFunc Factory[repository.UserRepository],
|
|
noteRepositoryFunc Factory[repository.NoteRepository],
|
|
followRepositoryFunc Factory[repository.FollowRepository],
|
|
instanceMetadataRepositoryFunc Factory[repository.InstanceMetadataRepository],
|
|
) *ManagerImpl {
|
|
return &ManagerImpl{
|
|
users: userRepositoryFunc(db, log.WithName("users"), telemetry),
|
|
notes: noteRepositoryFunc(db, log.WithName("notes"), telemetry),
|
|
follows: followRepositoryFunc(db, log.WithName("follows"), telemetry),
|
|
instanceMetadata: instanceMetadataRepositoryFunc(db, log.WithName("instanceMetadata"), telemetry),
|
|
|
|
uRFactory: userRepositoryFunc,
|
|
nRFactory: noteRepositoryFunc,
|
|
fRFactory: followRepositoryFunc,
|
|
imRFactory: instanceMetadataRepositoryFunc,
|
|
|
|
db: db,
|
|
log: log,
|
|
telemetry: telemetry,
|
|
}
|
|
}
|
|
|
|
func (i *ManagerImpl) withDB(db *ent.Client) *ManagerImpl {
|
|
return NewManagerImpl(
|
|
db, i.telemetry, i.log,
|
|
i.uRFactory,
|
|
i.nRFactory,
|
|
i.fRFactory,
|
|
i.imRFactory,
|
|
)
|
|
}
|
|
|
|
func (i *ManagerImpl) Atomic(ctx context.Context, fn func(ctx context.Context, tx repository.Manager) error) error {
|
|
s := i.telemetry.StartSpan(ctx, "function", "repo_impls/ManagerImpl.Atomic")
|
|
defer s.End()
|
|
ctx = s.Context()
|
|
|
|
tx, err := database.BeginTx(ctx, i.db, i.telemetry)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer func(tx *database.Tx) {
|
|
err := tx.Finish()
|
|
if err != nil {
|
|
i.log.Error(err, "Failed to finish transaction")
|
|
}
|
|
}(tx)
|
|
|
|
if err := fn(ctx, i.withDB(tx.Client())); err != nil {
|
|
return err
|
|
}
|
|
|
|
tx.MarkForCommit()
|
|
|
|
return tx.Finish()
|
|
}
|
|
|
|
func (i *ManagerImpl) Users() repository.UserRepository {
|
|
return i.users
|
|
}
|
|
|
|
func (i *ManagerImpl) Notes() repository.NoteRepository {
|
|
return i.notes
|
|
}
|
|
|
|
func (i *ManagerImpl) Follows() repository.FollowRepository {
|
|
return i.follows
|
|
}
|
|
|
|
func (i *ManagerImpl) InstanceMetadata() repository.InstanceMetadataRepository {
|
|
return i.instanceMetadata
|
|
}
|