fix: Use docker-container as the buildkitd socket (#72)

Signed-off-by: Ce Gao <cegao@tensorchord.ai>
tensorchord · Apr 27, 2022 · 2ad4d94 · 2ad4d94
1 parent 758bc0f
commit 2ad4d94
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/cmd/midi/build.go b/cmd/midi/build.go
@@ -62,8 +62,7 @@ func build(clicontext *cli.Context) error {
 
 	tag := clicontext.String("tag")
 
-	builder := builder.New(
-		fmt.Sprintf("docker-container://%s", viper.GetString(flag.FlagBuildkitdContainer)),
-		config, path, tag)
+	builder := builder.New(fmt.Sprintf("docker-container://%s",
+		viper.GetString(flag.FlagBuildkitdContainer)), config, path, tag)
 	return builder.Build(clicontext.Context)
 }
diff --git a/cmd/midi/up.go b/cmd/midi/up.go
@@ -15,14 +15,17 @@
 package main
 
 import (
+	"fmt"
 	"path/filepath"
 
 	"github.com/cockroachdb/errors"
 	"github.com/sirupsen/logrus"
+	"github.com/spf13/viper"
 	cli "github.com/urfave/cli/v2"
 
 	"github.com/tensorchord/MIDI/pkg/builder"
 	"github.com/tensorchord/MIDI/pkg/docker"
+	"github.com/tensorchord/MIDI/pkg/flag"
 	"github.com/tensorchord/MIDI/pkg/home"
 	"github.com/tensorchord/MIDI/pkg/ssh"
 )
@@ -68,7 +71,8 @@ func up(clicontext *cli.Context) error {
 
 	tag := clicontext.String("tag")
 
-	builder := builder.New("unix:///run/buildkit/buildkitd.sock", config, path, tag)
+	builder := builder.New(fmt.Sprintf("docker-container://%s",
+		viper.GetString(flag.FlagBuildkitdContainer)), config, path, tag)
 	if err := builder.Build(clicontext.Context); err != nil {
 		return err
 	}