godfreyhe commented on a change in pull request #8736: [FLINK-12846][table] Carry primary key and unique key information in TableSchema URL: https://github.com/apache/flink/pull/8736#discussion_r294118265
########## File path: flink-table/flink-table-common/src/main/java/org/apache/flink/table/api/TableSchema.java ########## @@ -343,13 +412,66 @@ public Builder field(String name, TypeInformation<?> typeInfo) { return field(name, fromLegacyInfoToDataType(typeInfo)); } + /** + * Add a primary key with the given field names. + * There can only be one PRIMARY KEY for a given table + * See the {@link TableSchema} class javadoc for more definition about primary key. + */ + public Builder primaryKey(String... fields) { + Preconditions.checkArgument( + fields != null && fields.length > 0, + "The primary key fields shouldn't be null or empty."); + Preconditions.checkArgument( + primaryKey == null, + "A primary key " + primaryKey + + " have been defined, can not define another primary key " + + Arrays.toString(fields)); + for (String field : fields) { + if (!fieldNames.contains(field)) { + throw new IllegalArgumentException("The field '" + field + + "' is not existed in the schema."); + } + } + primaryKey = Arrays.asList(fields); + return this; + } + + /** + * Add an unique key with the given field names. + * There can be more than one UNIQUE KEY for a given table. + * See the {@link TableSchema} class javadoc for more definition about unique key. + */ + public Builder uniqueKey(String... fields) { + Preconditions.checkArgument( + fields != null && fields.length > 0, + "The unique key fields shouldn't be null or empty."); + for (String field : fields) { + if (!fieldNames.contains(field)) { + throw new IllegalArgumentException("The field '" + field + + "' is not existed in the schema."); + } + } + if (uniqueKeys == null) { + uniqueKeys = new ArrayList<>(); + } + uniqueKeys.add(Arrays.asList(fields)); Review comment: should distinct uniqueKey ? ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services