diff --git a/boot.go b/boot.go index 18cbbd8..54977b7 100644 --- a/boot.go +++ b/boot.go @@ -20,8 +20,8 @@ import ( const ( localMinioInsecure = true - localMinioHost = "localhost" - localMinioPort = 9000 + defaultMinioHost = "localhost" + defaultMinioPort = "9000" ) var ( @@ -106,9 +106,9 @@ func readSecrets() (string, string) { return strings.TrimSpace(string(keyID)), strings.TrimSpace(string(accessKey)) } -func newMinioClient(host string, port int, accessKey, accessSecret string, insecure bool) (minio.CloudStorageClient, error) { +func newMinioClient(host, port, accessKey, accessSecret string, insecure bool) (minio.CloudStorageClient, error) { return minio.New( - fmt.Sprintf("%s:%d", host, port), + fmt.Sprintf("%s:%s", host, port), accessKey, accessSecret, insecure, @@ -120,7 +120,15 @@ func main() { checkError(err) key, access := readSecrets() - minioClient, err := newMinioClient(localMinioHost, localMinioPort, key, access, localMinioInsecure) + minioHost := os.Getenv("MINIO_HOST") + if minioHost == "" { + minioHost = defaultMinioHost + } + minioPort := os.Getenv("MINIO_PORT") + if minioPort == "" { + minioPort = defaultMinioPort + } + minioClient, err := newMinioClient(minioHost, minioPort, key, access, localMinioInsecure) if err != nil { log.Printf("Error creating minio client (%s)", err) os.Exit(1) diff --git a/boot_test.go b/boot_test.go index 6e2f053..c69c0e3 100644 --- a/boot_test.go +++ b/boot_test.go @@ -5,7 +5,7 @@ import ( ) func TestNewMinioClient(t *testing.T) { - client, err := newMinioClient("localhost", 8095, "access_key", "access_secret_key", false) + client, err := newMinioClient("localhost", "8095", "access_key", "access_secret_key", false) if err != nil { t.Fatalf("unexpected error creating minio client (%s)", err) }