From bb62cf4e6051ec4e0e1054b1dc133b8495d5fd39 Mon Sep 17 00:00:00 2001 From: smellthemoon <64083300+smellthemoon@users.noreply.github.com> Date: Wed, 31 May 2023 19:01:09 +0800 Subject: [PATCH] Support upsert api rbac (#24550) Signed-off-by: lixinguo Co-authored-by: lixinguo --- internal/proxy/privilege_interceptor_test.go | 5 +++++ pkg/util/constant.go | 1 + 2 files changed, 6 insertions(+) diff --git a/internal/proxy/privilege_interceptor_test.go b/internal/proxy/privilege_interceptor_test.go index 262784b525..d6d58455c7 100644 --- a/internal/proxy/privilege_interceptor_test.go +++ b/internal/proxy/privilege_interceptor_test.go @@ -111,6 +111,11 @@ func TestPrivilegeInterceptor(t *testing.T) { CollectionName: "col1", }) assert.NotNil(t, err) + _, err = PrivilegeInterceptor(ctx, &milvuspb.UpsertRequest{ + DbName: "db_test", + CollectionName: "col1", + }) + assert.NotNil(t, err) _, err = PrivilegeInterceptor(fooCtx, &milvuspb.GetLoadingProgressRequest{ CollectionName: "col1", }) diff --git a/pkg/util/constant.go b/pkg/util/constant.go index bf13a6045b..c7de317b7d 100644 --- a/pkg/util/constant.go +++ b/pkg/util/constant.go @@ -74,6 +74,7 @@ var ( MetaStore2API(commonpb.ObjectPrivilege_PrivilegeCompaction.String()), MetaStore2API(commonpb.ObjectPrivilege_PrivilegeInsert.String()), MetaStore2API(commonpb.ObjectPrivilege_PrivilegeDelete.String()), + MetaStore2API(commonpb.ObjectPrivilege_PrivilegeUpsert.String()), MetaStore2API(commonpb.ObjectPrivilege_PrivilegeGetStatistics.String()), MetaStore2API(commonpb.ObjectPrivilege_PrivilegeCreateIndex.String()),