iam verify in init supports aws

Signed-off-by: shaoyue.chen <shaoyue.chen@zilliz.com>
zilliztech · May 24, 2024 · 9fd2ad4 · 9fd2ad4
1 parent 52336c5
commit 9fd2ad4
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/pkg/util/yamlparser/file.go b/pkg/util/yamlparser/file.go
@@ -13,6 +13,9 @@ var UserYamlPath = "/milvus/configs/operator/user.yaml"
 
 type UserYaml struct {
 	Minio struct {
+		Address       string `yaml:"address"`
+		Port          int    `yaml:"port"`
+		UseSSL        bool   `yaml:"useSSL"`
 		UseIAM        bool   `yaml:"useIAM"`
 		CloudProvider string `yaml:"cloudProvider"`
 		AccessKeyID   string `yaml:"accessKeyID"`

diff --git a/tool/iam-verify/main.go b/tool/iam-verify/main.go
@@ -2,6 +2,7 @@ package main
 
 import (
 	"context"
+	"fmt"
 	"log"
 	"os"
 	"time"
@@ -37,6 +38,11 @@ func main() {
 				ContainerName:  userYaml.Minio.BucketName,
 			})
 		}
+	case "aws":
+		verifyFunc = func(ctx context.Context) error {
+			address := fmt.Sprintf("%s:%d", userYaml.Minio.Address, userYaml.Minio.Port)
+			return iam.VerifyAWS(ctx, userYaml.Minio.BucketName, address, userYaml.Minio.UseSSL)
+		}
 	default:
 		log.Printf("iam-verify for csp %s not implement, assume success\n", userYaml.Minio.CloudProvider)
 		os.Exit(0)