Update Dockerfile.driver

R3hankhan123 · Dec 6, 2024 · 9031d14 · 9031d14
1 parent 6900ff9
commit 9031d14
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/backend/Dockerfile.driver b/backend/Dockerfile.driver
@@ -14,15 +14,13 @@
 # limitations under the License.
 
 FROM golang:1.21.7-alpine3.19 as builder
+ARG TARGETOS
+ARG TARGETARCH
 
 WORKDIR /go/src/github.com/kubeflow/pipelines
 COPY . .
 
-RUN ARCH=$(uname -m) && \
-    if [ "$ARCH" = "x86_64" ]; then \
-        ARCH="amd64"; \
-    fi && \
-    GO111MODULE=on CGO_ENABLED=0 GOOS=linux GOARCH=$ARCH go build -tags netgo -ldflags '-extldflags "-static"' -o /bin/driver ./backend/src/v2/cmd/driver/*.go
+RUN GO111MODULE=on CGO_ENABLED=0 GOOS=${TARGETOS:-linux} GOARCH=${TARGETARCH} go build -tags netgo -ldflags '-extldflags "-static"' -o /bin/driver ./backend/src/v2/cmd/driver/*.go
 
 
 # Check licenses and comply with license terms.