RenewCMS/adapters/secondary/gateways/articleRepository.go

package gateways

import (
	"GohCMS2/db"
	. "GohCMS2/domain/article"
	. "GohCMS2/domain/gateways"
	"context"
)

type ArticleRepository struct {
	client *db.PrismaClient
}

func NewArticleRepository() *ArticleRepository {
	a := ArticleRepository{
		client: db.NewClient(),
	}
	return &a
}

func (a *ArticleRepository) connectClient() {
	err := a.client.Connect()
	if err != nil {
		panic(err)
	}
}

func (a *ArticleRepository) disconnectClient() {
	err := a.client.Disconnect()
	if err != nil {
		panic(err)
	}
	a.client = db.NewClient()
}

func (a *ArticleRepository) Get(id int) Article {
	a.connectClient()
	defer a.disconnectClient()

	article, err := a.client.Article.FindUnique(db.Article.ID.Equals(id)).Exec(context.Background())
	if err != nil {
		panic(err)
	}

	return FromDb(article.ID, article.Title, article.Body, article.CreatedAt.String(), article.UpdatedAt.String())
}

func (a *ArticleRepository) Create(article Article) Article {
	a.connectClient()
	defer a.disconnectClient()

	articleDb, err := a.client.Article.CreateOne(
		db.Article.Title.Set(article.Title),
		db.Article.Body.Set(article.Body),
	).Exec(context.Background())
	if err != nil {
		panic(err)
	}

	return FromDb(articleDb.ID, articleDb.Title, articleDb.Body, articleDb.CreatedAt.String(), articleDb.UpdatedAt.String())
}

func (a *ArticleRepository) GetAll() []Article {
	a.connectClient()
	defer a.disconnectClient()

	articlesDb, err := a.client.Article.FindMany().Exec(context.Background())
	if err != nil {
		panic(err)
	}

	var articles []Article
	for _, articleDb := range articlesDb {
		articles = append(articles, FromDb(articleDb.ID, articleDb.Title, articleDb.Body, articleDb.CreatedAt.String(), articleDb.UpdatedAt.String()))
	}

	return articles
}

var _ IArticleRepository = &ArticleRepository{}