batch.go

// Copyright (C) 2019-2023, Lux Partners Limited. All rights reserved.
// See the file LICENSE for licensing terms.

package archivedb

import "github.com/luxdefi/node/database"

var _ database.Batch = (*batch)(nil)

// batch is how a user performs modifications to the database.
//
// It consumes puts and deletes at a specified height. When committing, an
// atomic operation is created which registers the modifications at the
// specified height and updates the last tracked height to be equal to this
// batch's height.
type batch struct {
	db     *Database
	height uint64
	database.BatchOps
}

func (c *batch) Write() error {
	batch := c.db.db.NewBatch()
	for _, op := range c.Ops {
		key, _ := newDBKeyFromUser(op.Key, c.height)
		var value []byte
		if !op.Delete {
			value = newDBValue(op.Value)
		}
		if err := batch.Put(key, value); err != nil {
			return err
		}
	}

	if err := database.PutUInt64(batch, heightKey, c.height); err != nil {
		return err
	}

	return batch.Write()
}

func (c *batch) Inner() database.Batch {
	return c
}