HL-1205 Postgres support

CHANGELOG.md

@@ -24,7 +24,13 @@ Adding a requirement of a major version of a dependency is breaking a contract.
 Dropping a requirement of a major version of a dependency is a new contract.
 
 ## [Unreleased]
-[Unreleased]: https://github.com/atlassian/infrastructure/compare/release-4.14.4...master
+[Unreleased]: https://github.com/atlassian/infrastructure/compare/release-4.14.5...master
+
+## [4.14.5] - 2020-02-03
+[4.14.5]: https://github.com/atlassian/infrastructure/compare/release-4.14.4...release-4.14.5
+
+### Added
+- Add support for postgres database.
 
 ## [4.14.4] - 2019-12-09
 [4.14.4]: https://github.com/atlassian/infrastructure/compare/release-4.14.3...release-4.14.4

src/main/kotlin/com/atlassian/performance/tools/infrastructure/api/database/PostgresDatabase.kt

@@ -0,0 +1,48 @@
+package com.atlassian.performance.tools.infrastructure.api.database
+
+import com.atlassian.performance.tools.infrastructure.DockerImage
+import com.atlassian.performance.tools.infrastructure.api.dataset.DatasetPackage
+import com.atlassian.performance.tools.ssh.api.SshConnection
+import org.apache.logging.log4j.LogManager
+import org.apache.logging.log4j.Logger
+import java.net.URI
+import java.time.Duration
+
+class PostgresDatabase(
+    private val source: DatasetPackage,
+    private val maxConnections: Int,
+    private val dataBaseVersion: String
+) : Database {
+    private val logger: Logger = LogManager.getLogger(this::class.java)
+
+    private val image: DockerImage = DockerImage(
+        name = "postgres:$dataBaseVersion",
+        pullTimeout = Duration.ofMinutes(5)
+    )
+
+    constructor(
+        source: DatasetPackage,
+        dataBaseVersion: String = "9.6"
+    ) : this(
+        source = source,
+        maxConnections = 200,
+        dataBaseVersion = dataBaseVersion
+    )
+
+    override fun setup(ssh: SshConnection): String {
+        val pgData = source.download(ssh)
+        image.run(
+            ssh = ssh,
+            parameters = "-p 3306:5432 -v `realpath $pgData`:/var/lib/postgresql/data",
+            arguments = "-c 'listen_addresses='*'' -c 'max_connections=$maxConnections'"
+        )
+        return pgData
+    }
+
+    override fun start(jira: URI, ssh: SshConnection) {
+        // TODO Check logs for the following entry
+        // LOG:  database system is ready to accept connections
+        Thread.sleep(Duration.ofSeconds(15).toMillis())
+    }
+
+}

src/main/kotlin/com/atlassian/performance/tools/infrastructure/api/dataset/FileArchiver.kt

@@ -18,7 +18,11 @@ class FileArchiver {
     ) {
         ubuntu.install(connection, listOf("lbzip2"))
         time("unzip") {
-            connection.execute("tar -I lbzip2 -xf $archive", timeout)
+            val cmd = if (archive.contains("SCALED_ISSUES_UNIMODAL_PG"))
+                "tar -xzvf $archive"
+            else
+                "tar -I lbzip2 -xf $archive"
+            connection.execute(cmd, timeout)
         }
     }
 
@@ -35,7 +39,11 @@ class FileArchiver {
     ) {
         ubuntu.install(connection, listOf("lbzip2"))
         time("unzip") {
-            connection.execute("tar -I lbzip2 -xf $archive -C $destination", timeout).output.splitToSequence("\n").asIterable()
+            val cmd = if (archive.contains("SCALED_ISSUES_UNIMODAL_PG"))
+                "tar -xzvf $archive -C $destination"
+            else
+                "tar -I lbzip2 -xf $archive -C $destination"
+            connection.execute(cmd, timeout).output.splitToSequence("\n").asIterable()
         }
     }