greatape/components/model/repository/documents_repository.go

package repository

import (
	. "github.com/reiver/greatape/components/constants"
	. "github.com/reiver/greatape/components/contracts/model"
	. "github.com/reiver/greatape/components/model/entity"
	. "github.com/xeronith/diamante/contracts/database"
	. "github.com/xeronith/diamante/contracts/logging"
)

type documentsRepository struct {
	baseRepository
}

func newDocumentsRepository(logger ILogger) IDocumentsRepository {
	return &documentsRepository{
		baseRepository: newBaseRepository("document", "documents", DocumentEntityType, logger, false),
	}
}

func (repository *documentsRepository) Add(entity IDocumentEntity, editor int64) error {
	if entity.Id() < 0 {
		return ERROR_INVALID_PARAMETERS
	}

	// language=SQL
	query := `INSERT INTO "documents" ("id", "content", "editor") VALUES ($1, $2, $3);`
	return repository.database.InsertSingle(query, entity.Id(), entity.Content(), editor)
}

func (repository *documentsRepository) AddAtomic(transaction IRepositoryTransaction, entity IDocumentEntity, editor int64) error {
	if entity.Id() < 0 {
		return ERROR_INVALID_PARAMETERS
	}

	// language=SQL
	query := `INSERT INTO "documents" ("id", "content", "editor") VALUES ($1, $2, $3);`
	return repository.database.InsertSingleAtomic(transaction, query, entity.Id(), entity.Content(), editor)
}

func (repository *documentsRepository) FetchById(id int64) (IDocumentEntity, error) {
	if id <= 0 {
		return nil, ERROR_INVALID_PARAMETERS
	}

	// language=SQL
	query := `SELECT "id", "content" FROM "documents" WHERE "id" = $1 AND "status" = 0;`

	var documentEntity IDocumentEntity
	if err := repository.database.QuerySingle(func(cursor ICursor) error {
		var (
			id      int64
			content string
		)

		if err := cursor.Scan(&id, &content); err != nil {
			return err
		}

		documentEntity = NewDocumentEntity(id, content)
		return nil
	}, query, id); err != nil {
		return nil, err
	}

	return documentEntity, nil
}

func (repository *documentsRepository) Update(entity IDocumentEntity, editor int64) error {
	if entity.Id() <= 0 {
		return ERROR_INVALID_PARAMETERS
	}

	// language=SQL
	query := `UPDATE "documents" SET "content" = $1, "editor" = $2 WHERE "id" = $3;`
	return repository.database.UpdateSingle(query, entity.Content(), editor, entity.Id())
}

func (repository *documentsRepository) UpdateAtomic(transaction IRepositoryTransaction, entity IDocumentEntity, editor int64) error {
	if entity.Id() <= 0 {
		return ERROR_INVALID_PARAMETERS
	}

	// language=SQL
	query := `UPDATE "documents" SET "content" = $1, "editor" = $2 WHERE "id" = $3;`
	return repository.database.UpdateSingleAtomic(transaction, query, entity.Content(), editor, entity.Id())
}

func (repository *documentsRepository) Remove(entity IDocumentEntity, editor int64) error {
	if entity.Id() <= 0 {
		return ERROR_INVALID_PARAMETERS
	}

	// language=SQL
	query := `UPDATE "documents" SET "status" = 1, "editor" = $1 WHERE "id" = $2;`
	return repository.database.DeleteSingle(query, editor, entity.Id())
}

func (repository *documentsRepository) RemoveAtomic(transaction IRepositoryTransaction, entity IDocumentEntity, editor int64) error {
	if entity.Id() <= 0 {
		return ERROR_INVALID_PARAMETERS
	}

	// language=SQL
	query := `UPDATE "documents" SET "status" = 1, "editor" = $1 WHERE "id" = $2;`
	return repository.database.DeleteSingleAtomic(transaction, query, editor, entity.Id())
}

func (repository *documentsRepository) FetchAll() (DocumentEntities, error) {
	// language=SQL
	query := `SELECT "id", "content" FROM "documents" WHERE "id" > 0 AND "status" = 0;`

	var documentEntities DocumentEntities
	if err := repository.database.Query(func(cursor ICursor) error {
		var (
			id      int64
			content string
		)

		if err := cursor.Scan(&id, &content); err != nil {
			return err
		}

		documentEntities = append(documentEntities, NewDocumentEntity(id, content))
		return nil
	}, query); err != nil {
		return nil, err
	}

	return documentEntities, nil
}

func (repository *documentsRepository) UpdateContent(id int64, value string, editor int64) error {
	if id <= 0 {
		return ERROR_INVALID_PARAMETERS
	}

	// language=SQL
	query := `UPDATE "documents" SET "content" = $1, "editor" = $2 WHERE "id" = $3;`
	return repository.database.UpdateSingle(query, value, editor, id)
}

func (repository *documentsRepository) UpdateContentAtomic(transaction IRepositoryTransaction, id int64, value string, editor int64) error {
	if id <= 0 {
		return ERROR_INVALID_PARAMETERS
	}

	// language=SQL
	query := `UPDATE "documents" SET "content" = $1, "editor" = $2 WHERE "id" = $3;`
	return repository.database.UpdateSingleAtomic(transaction, query, value, editor, id)
}