Skip to content

Commit d93d0c9

Browse files
committed
[Fix][Connector-V2] Enhance privilege checks
1 parent ae8436f commit d93d0c9

File tree

1 file changed

+52
-0
lines changed
  • seatunnel-e2e/seatunnel-connector-v2-e2e/connector-paimon-e2e/src/test/java/org/apache/seatunnel/e2e/connector/paimon

1 file changed

+52
-0
lines changed

seatunnel-e2e/seatunnel-connector-v2-e2e/connector-paimon-e2e/src/test/java/org/apache/seatunnel/e2e/connector/paimon/PrivilegeUtil.java

Lines changed: 52 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,58 @@ public static void awaitPrivilegeApplied(
5656
tableIdentifier);
5757
}
5858
break;
59+
case ALTER_TABLE:
60+
for (Identifier tableIdentifier : identifiers) {
61+
privilegeChecker.assertCanAlterTable(
62+
tableIdentifier);
63+
}
64+
break;
65+
case DROP_TABLE:
66+
for (Identifier tableIdentifier : identifiers) {
67+
privilegeChecker.assertCanDropTable(
68+
tableIdentifier);
69+
}
70+
break;
71+
case CREATE_TABLE:
72+
for (Identifier tableIdentifier : identifiers) {
73+
privilegeChecker.assertCanCreateTable(
74+
tableIdentifier.getDatabaseName());
75+
}
76+
break;
77+
case DROP_DATABASE:
78+
for (Identifier tableIdentifier : identifiers) {
79+
privilegeChecker.assertCanDropDatabase(
80+
tableIdentifier.getDatabaseName());
81+
}
82+
break;
83+
case ALTER_DATABASE:
84+
for (Identifier tableIdentifier : identifiers) {
85+
privilegeChecker.assertCanAlterDatabase(
86+
tableIdentifier.getDatabaseName());
87+
}
88+
break;
89+
case CREATE_DATABASE:
90+
privilegeChecker.assertCanCreateDatabase();
91+
break;
92+
case ADMIN:
93+
for (Identifier tableIdentifier : identifiers) {
94+
privilegeChecker.assertCanSelect(
95+
tableIdentifier);
96+
privilegeChecker.assertCanInsert(
97+
tableIdentifier);
98+
privilegeChecker.assertCanAlterTable(
99+
tableIdentifier);
100+
privilegeChecker.assertCanDropTable(
101+
tableIdentifier);
102+
privilegeChecker.assertCanCreateTable(
103+
tableIdentifier.getDatabaseName());
104+
privilegeChecker.assertCanDropDatabase(
105+
tableIdentifier.getDatabaseName());
106+
privilegeChecker.assertCanAlterDatabase(
107+
tableIdentifier.getDatabaseName());
108+
privilegeChecker.assertCanCreateDatabase();
109+
}
110+
break;
59111
default:
60112
throw new UnsupportedOperationException(
61113
"Unsupported privilege type: " + type);

0 commit comments

Comments
 (0)