mirror of
https://github.com/zitadel/zitadel.git
synced 2025-07-18 05:28:36 +00:00
39 lines
1.1 KiB
Go
39 lines
1.1 KiB
Go
|
package postgres
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net"
|
||
|
|
"os"
|
||
|
|
|
||
|
|
embeddedpostgres "github.com/fergusstrange/embedded-postgres"
|
||
|
|
"github.com/zitadel/logging"
|
||
|
|
)
|
||
|
|
|
||
|
|
func StartEmbedded() (embeddedpostgres.Config, func()) {
|
||
|
|
path, err := os.MkdirTemp("", "zitadel-embedded-postgres-*")
|
||
|
|
logging.OnError(err).Fatal("unable to create temp dir")
|
||
|
|
|
||
|
|
port, close := getPort()
|
||
|
|
|
||
|
|
config := embeddedpostgres.DefaultConfig().Version(embeddedpostgres.V16).Port(uint32(port)).RuntimePath(path)
|
||
|
|
embedded := embeddedpostgres.NewDatabase(config)
|
||
|
|
|
||
|
|
close()
|
||
|
|
err = embedded.Start()
|
||
|
|
logging.OnError(err).Fatal("unable to start db")
|
||
|
|
|
||
|
|
return config, func() {
|
||
|
|
logging.OnError(embedded.Stop()).Error("unable to stop db")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// getPort returns a free port and locks it until close is called
|
||
|
|
func getPort() (port uint16, close func()) {
|
||
|
|
l, err := net.Listen("tcp", ":0")
|
||
|
|
logging.OnError(err).Fatal("unable to get port")
|
||
|
|
port = uint16(l.Addr().(*net.TCPAddr).Port)
|
||
|
|
logging.WithFields("port", port).Info("Port is available")
|
||
|
|
return port, func() {
|
||
|
|
logging.OnError(l.Close()).Error("unable to close port listener")
|
||
|
|
}
|
||
|
|
}
|