github.com/m3db/m3@v1.5.1-0.20231129193456-75a402aa583b/src/dbnode/storage/bootstrap.go

// Copyright (c) 2016 Uber Technologies, Inc.
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.

package storage

import (
	"errors"
	"fmt"
	"sync"
	"time"

	"github.com/m3db/m3/src/dbnode/storage/bootstrap"
	"github.com/m3db/m3/src/x/clock"
	"github.com/m3db/m3/src/x/context"
	xerrors "github.com/m3db/m3/src/x/errors"
	xtime "github.com/m3db/m3/src/x/time"
)

var (
	// errNamespaceIsBootstrapping raised when trying to bootstrap a namespace that's being bootstrapped.
	errNamespaceIsBootstrapping = errors.New("namespace is bootstrapping")

	// errNamespaceNotBootstrapped raised when trying to flush/snapshot data for a namespace that's not yet bootstrapped.
	errNamespaceNotBootstrapped = errors.New("namespace is not yet bootstrapped")

	// errShardIsBootstrapping raised when trying to bootstrap a shard that's being bootstrapped.
	errShardIsBootstrapping = errors.New("shard is bootstrapping")

	// errShardNotBootstrappedToFlush raised when trying to flush data for a shard that's not yet bootstrapped.
	errShardNotBootstrappedToFlush = errors.New("shard is not yet bootstrapped to flush")

	// errShardNotBootstrappedToSnapshot raised when trying to snapshot data for a shard that's not yet bootstrapped.
	errShardNotBootstrappedToSnapshot = errors.New("shard is not yet bootstrapped to snapshot")

	// errShardNotBootstrappedToRead raised when trying to read data for a shard that's not yet bootstrapped.
	errShardNotBootstrappedToRead = errors.New("shard is not yet bootstrapped to read")

	// errIndexNotBootstrappedToRead raised when trying to read the index before being bootstrapped.
	errIndexNotBootstrappedToRead = errors.New("index is not yet bootstrapped to read")

	// errBootstrapEnqueued raised when trying to bootstrap and bootstrap becomes enqueued.
	errBootstrapEnqueued = errors.New("database bootstrapping enqueued bootstrap")
)

const (
	bootstrapRetryInterval = 10 * time.Second
)

type bootstrapFn func() error

type bootstrapNamespace struct {
	namespace databaseNamespace
	shards    []databaseShard
}

type bootstrapManager struct {
	sync.RWMutex

	database                    database
	mediator                    databaseMediator
	bootstrapFn                 bootstrapFn
	processProvider             bootstrap.ProcessProvider
	state                       BootstrapState
	hasPending                  bool
	pendingOnCompleteFns        []BootstrapCompleteFn
	sleepFn                     sleepFn
	nowFn                       clock.NowFn
	lastBootstrapCompletionTime xtime.UnixNano
	instrumentation             *bootstrapInstrumentation
}

func newBootstrapManager(
	database database,
	mediator databaseMediator,
	opts Options,
) databaseBootstrapManager {
	m := &bootstrapManager{
		database:        database,
		mediator:        mediator,
		processProvider: opts.BootstrapProcessProvider(),
		sleepFn:         time.Sleep,
		nowFn:           opts.ClockOptions().NowFn(),
		instrumentation: newBootstrapInstrumentation(opts),
	}
	m.bootstrapFn = m.bootstrap
	return m
}

func (m *bootstrapManager) IsBootstrapped() bool {
	m.RLock()
	state := m.state
	m.RUnlock()
	return state == Bootstrapped
}

func (m *bootstrapManager) LastBootstrapCompletionTime() (xtime.UnixNano, bool) {
	m.RLock()
	bsTime := m.lastBootstrapCompletionTime
	m.RUnlock()
	return bsTime, bsTime > 0
}

func (m *bootstrapManager) BootstrapEnqueue(
	opts BootstrapEnqueueOptions,
) {
	go func() {
		result, err := m.startBootstrap(opts.OnCompleteFn)
		if err != nil && !result.AlreadyBootstrapping {
			m.instrumentation.emitAndLogInvariantViolation(err, "error bootstrapping")
		}
	}()
}

func (m *bootstrapManager) Bootstrap() (BootstrapResult, error) {
	return m.startBootstrap(nil)
}

func (m *bootstrapManager) startBootstrap(
	onCompleteFn BootstrapCompleteFn,
) (BootstrapResult, error) {
	m.Lock()
	if onCompleteFn != nil {
		// Append completion fn if specified.
		m.pendingOnCompleteFns = append(m.pendingOnCompleteFns, onCompleteFn)
	}
	switch m.state {
	case Bootstrapping:
		// NB(r): Already bootstrapping, now a consequent bootstrap
		// request comes in - we queue this up to bootstrap again
		// once the current bootstrap has completed.
		// This is an edge case that can occur if during either an
		// initial bootstrap or a resharding bootstrap if a new
		// reshard occurs and we need to bootstrap more shards.
		m.hasPending = true
		m.Unlock()
		result := BootstrapResult{AlreadyBootstrapping: true}
		return result, errBootstrapEnqueued
	default:
		m.state = Bootstrapping
	}
	m.Unlock()
	// NB(xichen): disable filesystem manager before we bootstrap to minimize
	// the impact of file operations on bootstrapping performance
	m.instrumentation.log.Info("disable fileOps and wait")
	m.mediator.DisableFileOpsAndWait()
	defer m.mediator.EnableFileOps()
	m.instrumentation.log.Info("fileOps disabled")

	var result BootstrapResult
	// Keep performing bootstraps until none pending and no error returned.
	for i := 0; true; i++ {
		// NB(r): Decouple implementation of bootstrap so can override in tests.
		bootstrapErr := m.bootstrapFn()
		if bootstrapErr != nil {
			result.ErrorsBootstrap = append(result.ErrorsBootstrap, bootstrapErr)
		}

		m.Lock()
		currPending := m.hasPending
		if currPending {
			// New bootstrap calls should now enqueue another pending bootstrap
			m.hasPending = false
		}
		m.Unlock()

		if currPending {
			// NB(r): Requires another bootstrap.
			continue
		}

		if bootstrapErr != nil {
			// NB(r): Last bootstrap failed, since this could be due to transient
			// failure we retry the bootstrap again. This is to avoid operators
			// needing to manually intervene for cases where failures are transient.
			m.instrumentation.bootstrapFailed(i+1, bootstrapErr)
			m.sleepFn(bootstrapRetryInterval)
			continue
		}

		// No pending bootstraps and last finished successfully.
		break
	}

	// NB(xichen): in order for bootstrapped data to be flushed to disk, a tick
	// needs to happen to drain the in-memory buffers and a consequent flush will
	// flush all the data onto disk. However, this has shown to be too intensive
	// to do immediately after bootstrap due to bootstrapping nodes simultaneously
	// attempting to tick through their series and flushing data, adding significant
	// load to the cluster. It turns out to be better to let ticking happen naturally
	// on its own course so that the load of ticking and flushing is more spread out
	// across the cluster.
	m.Lock()
	m.lastBootstrapCompletionTime = xtime.ToUnixNano(m.nowFn())
	m.state = Bootstrapped
	// NB(r): Clear out the pending completion functions and execute them if
	// needed.
	pendingOnCompleteFns := m.pendingOnCompleteFns
	m.pendingOnCompleteFns = nil
	m.Unlock()

	if len(pendingOnCompleteFns) > 0 {
		// Execute any on complete functions that were queued.
		for _, fn := range pendingOnCompleteFns {
			fn(result)
		}
	}

	return result, nil
}

func (m *bootstrapManager) Report() {
	m.instrumentation.setIsBootstrapped(m.IsBootstrapped())
	m.instrumentation.setIsBootstrappedAndDurable(m.database.IsBootstrappedAndDurable())
}

func (m *bootstrapManager) bootstrap() error {
	ctx := context.NewBackground()
	defer ctx.Close()

	// NB(r): construct new instance of the bootstrap process to avoid
	// state being kept around by bootstrappers.
	process, err := m.processProvider.Provide()
	if err != nil {
		return err
	}

	namespaces, err := m.database.OwnedNamespaces()
	if err != nil {
		return err
	}

	instrCtx := m.instrumentation.bootstrapPreparing()

	accmulators := make([]bootstrap.NamespaceDataAccumulator, 0, len(namespaces))
	defer func() {
		// Close all accumulators at bootstrap completion, only error
		// it returns is if already closed, so this is a code bug if ever
		// an error returned.
		for _, accumulator := range accmulators {
			if err := accumulator.Close(); err != nil {
				instrCtx.emitAndLogInvariantViolation(err, "could not close bootstrap data accumulator")
			}
		}
	}()

	var (
		bootstrapNamespaces = make([]bootstrapNamespace, len(namespaces))
		prepareWg           sync.WaitGroup
		prepareLock         sync.Mutex
		prepareMultiErr     xerrors.MultiError
	)
	for i, namespace := range namespaces {
		i, namespace := i, namespace
		prepareWg.Add(1)
		go func() {
			shards, err := namespace.PrepareBootstrap(ctx)

			prepareLock.Lock()
			defer func() {
				prepareLock.Unlock()
				prepareWg.Done()
			}()

			if err != nil {
				prepareMultiErr = prepareMultiErr.Add(err)
				return
			}

			bootstrapNamespaces[i] = bootstrapNamespace{
				namespace: namespace,
				shards:    shards,
			}
		}()
	}

	prepareWg.Wait()

	if err := prepareMultiErr.FinalError(); err != nil {
		m.instrumentation.bootstrapPrepareFailed(err)
		return err
	}

	var uniqueShards map[uint32]struct{}
	targets := make([]bootstrap.ProcessNamespace, 0, len(namespaces))
	for _, ns := range bootstrapNamespaces {
		bootstrapShards := make([]uint32, 0, len(ns.shards))
		if uniqueShards == nil {
			uniqueShards = make(map[uint32]struct{}, len(ns.shards))
		}

		for _, shard := range ns.shards {
			if shard.IsBootstrapped() {
				continue
			}

			uniqueShards[shard.ID()] = struct{}{}
			bootstrapShards = append(bootstrapShards, shard.ID())
		}

		// Add hooks so that each bootstrapper when it interacts
		// with the namespace and shards during data accumulation
		// gets an up to date view of all the file volumes that
		// actually exist on disk (since bootstrappers can write
		// new blocks to disk).
		hooks := bootstrap.NewNamespaceHooks(bootstrap.NamespaceHooksOptions{
			BootstrapSourceEnd: newBootstrapSourceEndHook(ns.shards),
		})

		accumulator := NewDatabaseNamespaceDataAccumulator(ns.namespace)
		accmulators = append(accmulators, accumulator)

		targets = append(targets, bootstrap.ProcessNamespace{
			Metadata:        ns.namespace.Metadata(),
			Shards:          bootstrapShards,
			Hooks:           hooks,
			DataAccumulator: accumulator,
		})
	}

	instrCtx.bootstrapStarted(len(uniqueShards))
	// Run the bootstrap.
	bootstrapResult, err := process.Run(ctx, instrCtx.start, targets)
	if err != nil {
		instrCtx.bootstrapFailed(err)
		return err
	}

	instrCtx.bootstrapSucceeded()

	instrCtx.bootstrapNamespacesStarted()
	// Use a multi-error here because we want to at least bootstrap
	// as many of the namespaces as possible.
	multiErr := xerrors.NewMultiError()
	for _, namespace := range namespaces {
		id := namespace.ID()
		result, ok := bootstrapResult.Results.Get(id)
		if !ok {
			err := fmt.Errorf("missing namespace from bootstrap result: %v",
				id.String())
			instrCtx.emitAndLogInvariantViolation(err, "bootstrap failed")
			return err
		}

		if err := namespace.Bootstrap(ctx, result); err != nil {
			instrCtx.bootstrapNamespaceFailed(err, id)
			multiErr = multiErr.Add(err)
		}
	}

	if err := multiErr.FinalError(); err != nil {
		instrCtx.bootstrapNamespacesFailed(err)
		return err
	}

	instrCtx.bootstrapNamespacesSucceeded()
	return nil
}