From 0795f0b2e8b40502c6fedb469f4cc5b3e2e8146e Mon Sep 17 00:00:00 2001
From: david-leifker <114954101+david-leifker@users.noreply.github.com>
Date: Wed, 29 Nov 2023 09:16:48 -0600
Subject: [PATCH] feat(gradle): quickstart postgres gradle task (#9329)

---
 docker/build.gradle                           | 34 +++++++++++++++++++
 ...ompose-without-neo4j.postgres.override.yml |  2 +-
 2 files changed, 35 insertions(+), 1 deletion(-)

diff --git a/docker/build.gradle b/docker/build.gradle
index 56634a5fe0c67..c7f783af6c997 100644
--- a/docker/build.gradle
+++ b/docker/build.gradle
@@ -15,6 +15,7 @@ ext {
             ':metadata-service:war',
             ':datahub-frontend',
     ]
+
     debug_modules = quickstart_modules - [':metadata-jobs:mce-consumer-job',
                                           ':metadata-jobs:mae-consumer-job']
     debug_compose_args = [
@@ -27,6 +28,13 @@ ext {
             'datahub-gms',
             'datahub-frontend-react'
     ]
+
+    // Postgres
+    pg_quickstart_modules = quickstart_modules - [':docker:mysql-setup'] + [':docker:postgres-setup']
+    pg_compose_args = [
+            '-f', 'docker-compose-without-neo4j.yml',
+            '-f', 'docker-compose-without-neo4j.postgres.override.yml'
+    ]
 }
 
 task quickstart(type: Exec, dependsOn: ':metadata-ingestion:install') {
@@ -125,3 +133,29 @@ task debugReload(type: Exec) {
     def cmd = ['docker compose -p datahub'] + debug_compose_args + ['restart'] + debug_reloadable
     commandLine 'bash', '-c', cmd.join(" ")
 }
+
+task quickstartPg(type: Exec, dependsOn: ':metadata-ingestion:install') {
+    dependsOn(pg_quickstart_modules.collect { it + ':dockerTag' })
+    shouldRunAfter ':metadata-ingestion:clean', 'quickstartNuke'
+
+    environment "DATAHUB_TELEMETRY_ENABLED", "false"
+    environment "DOCKER_COMPOSE_BASE", "file://${rootProject.projectDir}"
+    environment "DATAHUB_POSTGRES_VERSION", "15.5"
+
+    // OpenSearch
+    environment "DATAHUB_SEARCH_IMAGE", 'opensearchproject/opensearch'
+    environment "DATAHUB_SEARCH_TAG", '2.9.0'
+    environment "XPACK_SECURITY_ENABLED", 'plugins.security.disabled=true'
+    environment "USE_AWS_ELASTICSEARCH", 'true'
+
+    def cmd = [
+            'source ../metadata-ingestion/venv/bin/activate && ',
+            'datahub docker quickstart',
+            '--no-pull-images',
+            '--standalone_consumers',
+            '--version', "v${version}",
+            '--dump-logs-on-failure'
+    ] + pg_compose_args
+
+    commandLine 'bash', '-c', cmd.join(" ")
+}
diff --git a/docker/docker-compose-without-neo4j.postgres.override.yml b/docker/docker-compose-without-neo4j.postgres.override.yml
index e4c754b30afd7..369b5a155fc36 100644
--- a/docker/docker-compose-without-neo4j.postgres.override.yml
+++ b/docker/docker-compose-without-neo4j.postgres.override.yml
@@ -53,7 +53,7 @@ services:
   postgres:
     container_name: postgres
     hostname: postgres
-    image: postgres:12.3
+    image: postgres:${DATAHUB_POSTGRES_VERSION:-12.3}
     env_file: postgres/env/docker.env
     ports:
       - '5432:5432'