Function setupSampleInstance

backend/resources/postgres/sample_instance.go:99–129

setupSampleInstance starts a single postgres sample instance.

(ctx context.Context, databaseName string, port int, sampleData string)

// setupSampleInstance starts a single postgres sample instance.
func setupSampleInstance(ctx context.Context, databaseName string, port int, sampleData string) error {
	defaultDB, err := sql.Open("pgx", fmt.Sprintf("user=%s host=%s port=%d database=postgres", SampleUser, common.GetPostgresSocketDir(), port))
	if err != nil {
		return err
	}
	defer defaultDB.Close()

	var ok bool
	if err := defaultDB.QueryRowContext(ctx, "SELECT EXISTS (SELECT 1 FROM pg_database WHERE datname = $1);", databaseName).Scan(&ok); err != nil {
		return err
	}
	if ok {
		slog.Info(fmt.Sprintf("Sample database %s already exists, skip setup", databaseName))
		return nil
	}

	if _, err := defaultDB.ExecContext(ctx, fmt.Sprintf("CREATE DATABASE %s", databaseName)); err != nil { // NOSONAR(go:S2077) databaseName is a hardcoded constant from envDB map, not user input
		return errors.Wrapf(err, "failed to create sample database")
	}

	db, err := sql.Open("pgx", fmt.Sprintf("user=%s host=%s port=%d database=%s", SampleUser, common.GetPostgresSocketDir(), port, databaseName))
	if err != nil {
		return err
	}
	defer db.Close()

	if _, err := db.ExecContext(ctx, sampleData); err != nil {
		return err
	}
	return nil
}