fix: adjust listen_addresses in the foundation DLE container

agneum · agneum · commit 379302d6dae9 · 2023-08-15T02:38:07.000Z
diff --git a/engine/internal/retrieval/engine/postgres/tools/db/image_content.go b/engine/internal/retrieval/engine/postgres/tools/db/image_content.go
@@ -216,7 +216,11 @@ func createContainer(ctx context.Context, docker *client.Client, image string, p
 	}
 
 	if err := resetHBA(ctx, docker, containerID); err != nil {
-		return "", fmt.Errorf("failed to init Postgres: %w", err)
+		return "", fmt.Errorf("failed to prepare pg_hba.conf: %w", err)
+	}
+
+	if err := setListenAddresses(ctx, docker, containerID); err != nil {
+		return "", fmt.Errorf("failed to set listen_addresses: %w", err)
 	}
 
 	if err := tools.StartPostgres(ctx, docker, containerID, tools.DefaultStopTimeout); err != nil {
@@ -247,10 +251,27 @@ func resetHBA(ctx context.Context, dockerClient *client.Client, containerID stri
 	})
 
 	if err != nil {
+		log.Dbg(out)
 		return fmt.Errorf("failed to reset pg_hba.conf: %w", err)
 	}
 
-	log.Dbg(out)
+	return nil
+}
+
+func setListenAddresses(ctx context.Context, dockerClient *client.Client, containerID string) error {
+	command := []string{"sh", "-c", `su postgres -c "echo listen_addresses = \'*\' >> ${PGDATA}/postgresql.conf"`}
+
+	log.Dbg("Set listen addresses", command)
+
+	out, err := tools.ExecCommandWithOutput(ctx, dockerClient, containerID, types.ExecConfig{
+		Tty: true,
+		Cmd: command,
+	})
+
+	if err != nil {
+		log.Dbg(out)
+		return fmt.Errorf("failed to set listen addresses: %w", err)
+	}
 
 	return nil
 }
