safing-portbase/database/main.go

package database

import (
	"errors"
	"fmt"
	"path"

	"github.com/tevino/abool"
)

var (
	initialized = abool.NewBool(false)

	shuttingDown   = abool.NewBool(false)
	shutdownSignal = make(chan struct{})
)

// SetLocation sets the location of the database. This is separate from the initialization to provide the location to other modules earlier.
func SetLocation(location string) (ok bool) {
	if !initialized.IsSet() && rootDir == "" {
		rootDir = location
		return true
	}
	return false
}

// Initialize initialized the database
func Initialize() error {
	if initialized.SetToIf(false, true) {

		err := ensureDirectory(rootDir)
		if err != nil {
			return fmt.Errorf("could not create/open database directory (%s): %s", rootDir, err)
		}

		err = loadRegistry()
		if err != nil {
			return fmt.Errorf("could not load database registry (%s): %s", path.Join(rootDir, registryFileName), err)
		}

		// start registry writer
		go registryWriter()

		return nil
	}
	return errors.New("database already initialized")
}

// Shutdown shuts down the whole database system.
func Shutdown() (err error) {
	if shuttingDown.SetToIf(false, true) {
		close(shutdownSignal)
	}

	all := duplicateControllers()
	for _, c := range all {
		err = c.Shutdown()
		if err != nil {
			return
		}
	}
	return
}