Merge pull request #14 from aliyun/features/load_single_project

[fix]:only load project with input project name
aliyun · Apr 23, 2024 · fd73095 · fd73095
2 parents 21d703e + b2c3940
commit fd73095
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/api/api_fs_project.go b/api/api_fs_project.go
@@ -22,7 +22,7 @@ FsProjectApiService List Projects
 */
 func (a *FsProjectApiService) ListProjects() (ListProjectsResponse, error) {
 	request := paifeaturestore.ListProjectsRequest{}
-	request.SetName(a.client.cfg.projectName)
+	request.SetName(a.client.cfg.ProjectName)
 
 	response, err := a.client.ListProjects(&a.client.instanceId, &request)
 	var (

diff --git a/api/configuration.go b/api/configuration.go
@@ -6,7 +6,7 @@ type Configuration struct {
 	regionId        string
 	AccessKeyId     string
 	AccessKeySecret string
-	projectName     string
+	ProjectName     string
 	UserAgent       string
 	domain          string
 }
@@ -15,7 +15,7 @@ func NewConfiguration(regionId, accessKeyId, accessKeySecret, projectName string
 	cfg := &Configuration{
 		UserAgent:       "PAI-FeatureStore/1.0.0/go",
 		regionId:        regionId,
-		projectName:     projectName,
+		ProjectName:     projectName,
 		AccessKeyId:     accessKeyId,
 		AccessKeySecret: accessKeySecret,
 	}

diff --git a/featurestore/feature_store_client.go b/featurestore/feature_store_client.go
@@ -169,6 +169,9 @@ func (c *FeatureStoreClient) LoadProjectData() {
 	}
 
 	for _, p := range listProjectsResponse.Projects {
+		if p.ProjectName != c.client.GetConfig().ProjectName {
+			continue
+		}
 		// get datasource
 		getDataSourceResponse, err := c.client.DatasourceApi.DatasourceDatasourceIdGet(p.OnlineDatasourceId)
 		if err != nil {