mirror of
https://github.com/lightningnetwork/lnd.git
synced 2025-08-04 10:42:25 +02:00
df1e6da94e
In this commit, we update the batch schedular so that it has the ability to do read-only calls. It will do a best effort attempt at keeping a transaction in read-only mode and then if any requests get added to a batch that require a read-write tx, then the entire batch's tx will be upgraded to use a read-write tx.
95 lines
3.0 KiB
Go
95 lines
3.0 KiB
Go
package batch
|
|
|
|
import "context"
|
|
|
|
// Request defines an operation that can be batched into a single bbolt
|
|
// transaction.
|
|
type Request[Q any] struct {
|
|
// Opts holds various configuration options for a scheduled request.
|
|
Opts *SchedulerOptions
|
|
|
|
// Reset is called before each invocation of Update and is used to clear
|
|
// any possible modifications to local state as a result of previous
|
|
// calls to Update that were not committed due to a concurrent batch
|
|
// failure.
|
|
//
|
|
// NOTE: This field is optional.
|
|
Reset func()
|
|
|
|
// Do is applied alongside other operations in the batch.
|
|
//
|
|
// NOTE: This method MUST NOT acquire any mutexes.
|
|
Do func(tx Q) error
|
|
|
|
// OnCommit is called if the batch or a subset of the batch including
|
|
// this request all succeeded without failure. The passed error should
|
|
// contain the result of the transaction commit, as that can still fail
|
|
// even if none of the closures returned an error.
|
|
//
|
|
// NOTE: This field is optional.
|
|
OnCommit func(commitErr error) error
|
|
}
|
|
|
|
// SchedulerOptions holds various configuration options for a scheduled request.
|
|
type SchedulerOptions struct {
|
|
// Lazy should be true if we don't have to immediately execute this
|
|
// request when it comes in. This means that it can be scheduled later,
|
|
// allowing larger batches.
|
|
Lazy bool
|
|
|
|
// ReadOnly should be true if the request is read-only. By default,
|
|
// this is false.
|
|
ReadOnly bool
|
|
}
|
|
|
|
// NewDefaultSchedulerOpts returns a new SchedulerOptions with default values.
|
|
func NewDefaultSchedulerOpts() *SchedulerOptions {
|
|
return &SchedulerOptions{
|
|
Lazy: false,
|
|
ReadOnly: false,
|
|
}
|
|
}
|
|
|
|
// NewSchedulerOptions returns a new SchedulerOptions with the given options
|
|
// applied on top of the default options.
|
|
func NewSchedulerOptions(options ...SchedulerOption) *SchedulerOptions {
|
|
opts := NewDefaultSchedulerOpts()
|
|
for _, o := range options {
|
|
o(opts)
|
|
}
|
|
|
|
return opts
|
|
}
|
|
|
|
// SchedulerOption is a type that can be used to supply options to a scheduled
|
|
// request.
|
|
type SchedulerOption func(*SchedulerOptions)
|
|
|
|
// LazyAdd will make the request be executed lazily, added to the next batch to
|
|
// reduce db contention.
|
|
func LazyAdd() SchedulerOption {
|
|
return func(opts *SchedulerOptions) {
|
|
opts.Lazy = true
|
|
}
|
|
}
|
|
|
|
// ReadOnly will mark the request as read-only. This means that the
|
|
// transaction will be executed in read-only mode, and no changes will be
|
|
// made to the database. If any requests in the same batch are not read-only,
|
|
// then the entire batch will be executed in read-write mode.
|
|
func ReadOnly() SchedulerOption {
|
|
return func(opts *SchedulerOptions) {
|
|
opts.ReadOnly = true
|
|
}
|
|
}
|
|
|
|
// Scheduler abstracts a generic batching engine that accumulates an incoming
|
|
// set of Requests, executes them, and returns the error from the operation.
|
|
type Scheduler[Q any] interface {
|
|
// Execute schedules a Request for execution with the next available
|
|
// batch. This method blocks until the underlying closure has been
|
|
// run against the database. The resulting error is returned to the
|
|
// caller.
|
|
Execute(ctx context.Context, req *Request[Q]) error
|
|
}
|