jokes-bapak2/api/app/core/joke/setter.go

package joke

import (
	"context"
	"jokes-bapak2-api/app/core/schema"

	"github.com/Masterminds/squirrel"
	"github.com/allegro/bigcache/v3"
	"github.com/jackc/pgx/v4/pgxpool"
	"github.com/pquerna/ffjson/ffjson"
)

// SetAllJSONJoke fetches jokes data from GetAllJSONJokes then set it to memory cache.
func SetAllJSONJoke(db *pgxpool.Pool, ctx context.Context, memory *bigcache.BigCache) error {
	jokes, err := GetAllJSONJokes(db, ctx)
	if err != nil {
		return err
	}
	err = memory.Set("jokes", jokes)
	if err != nil {
		return err
	}
	return nil
}

func SetTotalJoke(db *pgxpool.Pool, ctx context.Context, memory *bigcache.BigCache) error {
	check, err := CheckJokesCache(memory)
	if err != nil {
		return err
	}

	if !check {
		err = SetAllJSONJoke(db, ctx, memory)
		if err != nil {
			return err
		}
	}

	jokes, err := memory.Get("jokes")
	if err != nil {
		return err
	}

	var data []schema.Joke
	err = ffjson.Unmarshal(jokes, &data)
	if err != nil {
		return err
	}

	var total = []byte{byte(len(data))}
	err = memory.Set("total", total)
	if err != nil {
		return err
	}

	return nil
}

func InsertJokeIntoDB(db *pgxpool.Pool, ctx context.Context, joke schema.Joke) error {
	conn, err := db.Acquire(ctx)
	if err != nil {
		return err
	}
	defer conn.Release()

	var query = squirrel.StatementBuilder.PlaceholderFormat(squirrel.Dollar)
	sql, args, err := query.
		Insert("jokesbapak2").
		Columns("link", "creator").
		Values(joke.Link, joke.Creator).
		ToSql()
	if err != nil {
		return err
	}

	r, err := conn.Query(context.Background(), sql, args...)
	if err != nil {
		return err
	}
	defer r.Close()
	return nil
}

func DeleteSingleJoke(db *pgxpool.Pool, ctx context.Context, id int) error {
	conn, err := db.Acquire(ctx)
	if err != nil {
		return err
	}
	defer conn.Release()

	var query = squirrel.StatementBuilder.PlaceholderFormat(squirrel.Dollar)
	sql, args, err := query.
		Delete("jokesbapak2").
		Where(squirrel.Eq{"id": id}).
		ToSql()
	if err != nil {
		return err
	}

	r, err := conn.Query(context.Background(), sql, args...)
	if err != nil {
		return err
	}
	defer r.Close()

	return nil
}

func UpdateJoke(db *pgxpool.Pool, ctx context.Context, link, creator string) error {
	conn, err := db.Acquire(ctx)
	if err != nil {
		return err
	}
	defer conn.Release()

	var query = squirrel.StatementBuilder.PlaceholderFormat(squirrel.Dollar)
	sql, args, err := query.
		Update("jokesbapak2").
		Set("link", link).
		Set("creator", creator).
		ToSql()
	if err != nil {
		return err
	}

	r, err := conn.Query(context.Background(), sql, args...)
	if err != nil {
		return err
	}
	defer r.Close()

	return nil
}