effect_batch_insert_builder.go

package history

import (
	"context"

	"github.com/TosinShada/stellar-core/support/db"
	"github.com/guregu/null"
)

// EffectBatchInsertBuilder is used to insert effects into the
// history_effects table
type EffectBatchInsertBuilder interface {
	Add(
		ctx context.Context,
		accountID int64,
		muxedAccount null.String,
		operationID int64,
		order uint32,
		effectType EffectType,
		details []byte,
	) error
	Exec(ctx context.Context) error
}

// effectBatchInsertBuilder is a simple wrapper around db.BatchInsertBuilder
type effectBatchInsertBuilder struct {
	builder db.BatchInsertBuilder
}

// NewEffectBatchInsertBuilder constructs a new EffectBatchInsertBuilder instance
func (q *Q) NewEffectBatchInsertBuilder(maxBatchSize int) EffectBatchInsertBuilder {
	return &effectBatchInsertBuilder{
		builder: db.BatchInsertBuilder{
			Table:        q.GetTable("history_effects"),
			MaxBatchSize: maxBatchSize,
		},
	}
}

// Add adds a effect to the batch
func (i *effectBatchInsertBuilder) Add(
	ctx context.Context,
	accountID int64,
	muxedAccount null.String,
	operationID int64,
	order uint32,
	effectType EffectType,
	details []byte,
) error {
	return i.builder.Row(ctx, map[string]interface{}{
		"history_account_id":   accountID,
		"address_muxed":        muxedAccount,
		"history_operation_id": operationID,
		"\"order\"":            order,
		"type":                 effectType,
		"details":              details,
	})
}

func (i *effectBatchInsertBuilder) Exec(ctx context.Context) error {
	return i.builder.Exec(ctx)
}