From 4d6999d130ddc52b2be4359e110728116e95dd96 Mon Sep 17 00:00:00 2001
From: nicolasgere <nicolas.gerelm@gmail.com>
Date: Fri, 26 Apr 2024 13:44:21 -0700
Subject: [PATCH] [ENH]: simplify migration (#2067)

## Description of changes

*Summarize the changes made by this PR.*
 - Improvements & Bug fixes
	 - simplify log migration
---
 go/Dockerfile.migration                             | 6 +-----
 k8s/distributed-chroma/templates/log-migration.yaml | 5 +----
 2 files changed, 2 insertions(+), 9 deletions(-)

diff --git a/go/Dockerfile.migration b/go/Dockerfile.migration
index 72260eea780..96a9ddb2cbc 100644
--- a/go/Dockerfile.migration
+++ b/go/Dockerfile.migration
@@ -8,10 +8,6 @@ RUN curl -sSf https://atlasgo.sh | sh -s -- --community
 COPY ./go/migrations migrations
 COPY ./go/atlas.hcl atlas.hcl
 
-FROM debian:bookworm-slim as logservice-migration
-RUN apt update
-RUN apt upgrade -y
-RUN apt install -y curl jq
-RUN curl -sSf https://atlasgo.sh | sh -s -- --community
+FROM arigaio/atlas:latest as logservice-migration
 COPY ./go/database/log/migrations migrations
 COPY ./go/database/log/atlas.hcl atlas.hcl
diff --git a/k8s/distributed-chroma/templates/log-migration.yaml b/k8s/distributed-chroma/templates/log-migration.yaml
index ffe2a95c23d..e6664d3f819 100644
--- a/k8s/distributed-chroma/templates/log-migration.yaml
+++ b/k8s/distributed-chroma/templates/log-migration.yaml
@@ -11,10 +11,7 @@ spec:
     spec:
       restartPolicy: OnFailure
       containers:
-        - command:
-            - "/bin/sh"
-            - "-c"
-            - "atlas migrate apply --env prod"
+        - args: ['migrate', 'apply', '--env','prod']
           image: "{{ .Values.logServiceMigration.image.repository }}:{{ .Values.logServiceMigration.image.tag }}"
           imagePullPolicy: IfNotPresent
           name: migration