batch.go

package pgxadapter

import (
	"context"

	"github.com/jackc/pgx/v5"
)

// AddPolicies adds policy rules to the storage.
func (a *Adapter) AddPolicies(sec string, ptype string, rules [][]string) error {
	return a.AddPoliciesCtx(context.Background(), sec, ptype, rules)
}

// AddPoliciesCtx adds policy rules to the storage with context.
func (a *Adapter) AddPoliciesCtx(ctx context.Context, sec string, ptype string, rules [][]string) error {
	tx, err := a.pool.Begin(ctx)
	if err != nil {
		return err
	}
	defer tx.Rollback(ctx)

	var rows [][]interface{}
	for _, rule := range rules {
		rows = append(rows, a.genRow(ptype, rule))
	}

	_, err = tx.CopyFrom(
		ctx,
		pgx.Identifier{a.tableName},
		[]string{"ptype", "v0", "v1", "v2", "v3", "v4", "v5"},
		pgx.CopyFromRows(rows),
	)
	if err != nil {
		return err
	}

	return tx.Commit(ctx)
}

// RemovePolicies removes policy rules from the storage.
func (a *Adapter) RemovePolicies(sec string, ptype string, rules [][]string) error {
	return a.RemovePoliciesCtx(context.Background(), sec, ptype, rules)
}

// RemovePoliciesCtx removes policy rules from the storage with context.
func (a *Adapter) RemovePoliciesCtx(ctx context.Context, sec string, ptype string, rules [][]string) error {
	tx, err := a.pool.Begin(ctx)
	if err != nil {
		return err
	}
	defer tx.Rollback(ctx)

	for _, rule := range rules {
		if err := a.removePolicy(ctx, tx, ptype, rule); err != nil {
			return err
		}
	}

	return tx.Commit(ctx)
}