Based on patduin comments, modified the code

sivaponting · Aug 3, 2023 · 42ad1ad · 42ad1ad
1 parent 83d9fce
commit 42ad1ad
Show file tree

Hide file tree

Showing 4 changed files with 502 additions and 60 deletions.
diff --git a/...le-dance-core/src/main/java/com/hotels/bdp/waggledance/mapping/model/DatabaseMapping.java b/...le-dance-core/src/main/java/com/hotels/bdp/waggledance/mapping/model/DatabaseMapping.java
@@ -17,16 +17,29 @@
 
 import java.util.List;
 
+import org.apache.hadoop.hive.metastore.api.AddCheckConstraintRequest;
+import org.apache.hadoop.hive.metastore.api.AddDefaultConstraintRequest;
 import org.apache.hadoop.hive.metastore.api.AddDynamicPartitions;
+import org.apache.hadoop.hive.metastore.api.AddForeignKeyRequest;
+import org.apache.hadoop.hive.metastore.api.AddNotNullConstraintRequest;
 import org.apache.hadoop.hive.metastore.api.AddPartitionsRequest;
 import org.apache.hadoop.hive.metastore.api.AddPartitionsResult;
+import org.apache.hadoop.hive.metastore.api.AddUniqueConstraintRequest;
+import org.apache.hadoop.hive.metastore.api.AllocateTableWriteIdsRequest;
+import org.apache.hadoop.hive.metastore.api.AlterISchemaRequest;
 import org.apache.hadoop.hive.metastore.api.CacheFileMetadataRequest;
+import org.apache.hadoop.hive.metastore.api.CheckConstraintsRequest;
+import org.apache.hadoop.hive.metastore.api.CheckConstraintsResponse;
 import org.apache.hadoop.hive.metastore.api.ColumnStatistics;
 import org.apache.hadoop.hive.metastore.api.CompactionRequest;
+import org.apache.hadoop.hive.metastore.api.CreationMetadata;
 import org.apache.hadoop.hive.metastore.api.Database;
+import org.apache.hadoop.hive.metastore.api.DefaultConstraintsRequest;
+import org.apache.hadoop.hive.metastore.api.DefaultConstraintsResponse;
 import org.apache.hadoop.hive.metastore.api.DropConstraintRequest;
 import org.apache.hadoop.hive.metastore.api.DropPartitionsRequest;
 import org.apache.hadoop.hive.metastore.api.DropPartitionsResult;
+import org.apache.hadoop.hive.metastore.api.FindSchemasByColsResp;
 import org.apache.hadoop.hive.metastore.api.FireEventRequest;
 import org.apache.hadoop.hive.metastore.api.ForeignKeysRequest;
 import org.apache.hadoop.hive.metastore.api.ForeignKeysResponse;
@@ -39,7 +52,12 @@
 import org.apache.hadoop.hive.metastore.api.HiveObjectPrivilege;
 import org.apache.hadoop.hive.metastore.api.HiveObjectRef;
 import org.apache.hadoop.hive.metastore.api.ISchema;
+import org.apache.hadoop.hive.metastore.api.ISchemaName;
 import org.apache.hadoop.hive.metastore.api.LockRequest;
+import org.apache.hadoop.hive.metastore.api.MapSchemaVersionToSerdeRequest;
+import org.apache.hadoop.hive.metastore.api.NotNullConstraintsRequest;
+import org.apache.hadoop.hive.metastore.api.NotNullConstraintsResponse;
+import org.apache.hadoop.hive.metastore.api.NotificationEventsCountRequest;
 import org.apache.hadoop.hive.metastore.api.Partition;
 import org.apache.hadoop.hive.metastore.api.PartitionSpec;
 import org.apache.hadoop.hive.metastore.api.PartitionValuesRequest;
@@ -49,11 +67,22 @@
 import org.apache.hadoop.hive.metastore.api.PrimaryKeysRequest;
 import org.apache.hadoop.hive.metastore.api.PrimaryKeysResponse;
 import org.apache.hadoop.hive.metastore.api.PrivilegeBag;
+import org.apache.hadoop.hive.metastore.api.ReplTblWriteIdStateRequest;
+import org.apache.hadoop.hive.metastore.api.SQLCheckConstraint;
+import org.apache.hadoop.hive.metastore.api.SQLDefaultConstraint;
+import org.apache.hadoop.hive.metastore.api.SQLForeignKey;
+import org.apache.hadoop.hive.metastore.api.SQLNotNullConstraint;
+import org.apache.hadoop.hive.metastore.api.SQLPrimaryKey;
+import org.apache.hadoop.hive.metastore.api.SQLUniqueConstraint;
 import org.apache.hadoop.hive.metastore.api.SchemaVersion;
+import org.apache.hadoop.hive.metastore.api.SchemaVersionDescriptor;
 import org.apache.hadoop.hive.metastore.api.SetPartitionsStatsRequest;
+import org.apache.hadoop.hive.metastore.api.SetSchemaVersionStateRequest;
 import org.apache.hadoop.hive.metastore.api.Table;
 import org.apache.hadoop.hive.metastore.api.TableMeta;
 import org.apache.hadoop.hive.metastore.api.TableStatsRequest;
+import org.apache.hadoop.hive.metastore.api.UniqueConstraintsRequest;
+import org.apache.hadoop.hive.metastore.api.UniqueConstraintsResponse;
 
 public interface DatabaseMapping extends MetaStoreMapping {
 
@@ -151,10 +180,72 @@ public interface DatabaseMapping extends MetaStoreMapping {
 
   PartitionValuesRequest transformInboundPartitionValuesRequest(PartitionValuesRequest req);
 
+  List<SQLPrimaryKey> transformInboundSQLPrimaryKeys(List<SQLPrimaryKey> sqlPrimaryKeys);
+
+  List<SQLForeignKey> transformInboundSQLForeignKeys(List<SQLForeignKey> sqlForeignKeys);
+
+  List<SQLUniqueConstraint> transformInboundSQLUniqueConstraints(List<SQLUniqueConstraint> sqlUniqueConstraints);
+
+  List<SQLNotNullConstraint> transformInboundSQLNotNullConstraints(List<SQLNotNullConstraint> sqlNotNullConstraints);
+
+  List<SQLDefaultConstraint> transformInboundSQLDefaultConstraints(List<SQLDefaultConstraint> sqlDefaultConstraints);
+
+  List<SQLCheckConstraint> transformInboundSQLCheckConstraints(List<SQLCheckConstraint> sqlCheckConstraints);
+
+  ReplTblWriteIdStateRequest transformInboundReplTblWriteIdStateRequest(ReplTblWriteIdStateRequest request);
+
+  AllocateTableWriteIdsRequest transformInboundAllocateTableWriteIdsRequest(AllocateTableWriteIdsRequest request);
+
+  AlterISchemaRequest transformInboundAlterISchemaRequest(AlterISchemaRequest request);
+
+  SchemaVersion transformInboundSchemaVersion(SchemaVersion schemaVersion);
+
   SchemaVersion transformOutboundSchemaVersion(SchemaVersion schemaVersion);
 
   List<SchemaVersion> transformOutboundSchemaVersions(List<SchemaVersion> schemaVersions);
 
+  ISchemaName transformInboundISchemaName(ISchemaName iSchemaName);
+
+  ISchemaName transformOutboundISchemaName(ISchemaName iSchemaName);
+
+  AddForeignKeyRequest transformInboundAddForeignKeyRequest(AddForeignKeyRequest request);
+
+  AddUniqueConstraintRequest transformInboundAddUniqueConstraintRequest(AddUniqueConstraintRequest request);
+
+  AddNotNullConstraintRequest transformInboundAddNotNullConstraintRequest(AddNotNullConstraintRequest request);
+
+  AddDefaultConstraintRequest transformInboundAddDefaultConstraintRequest(AddDefaultConstraintRequest request);
+
+  AddCheckConstraintRequest transformInboundAddCheckConstraintRequest(AddCheckConstraintRequest request);
+
+  FindSchemasByColsResp transformOutboundFindSchemasByColsResp(FindSchemasByColsResp response);
+
+  SchemaVersionDescriptor transformInboundSchemaVersionDescriptor(SchemaVersionDescriptor request);
+
+  MapSchemaVersionToSerdeRequest transformInboundMapSchemaVersionToSerdeRequest(MapSchemaVersionToSerdeRequest request);
+
+  SetSchemaVersionStateRequest transformInboundSetSchemaVersionStateRequest(SetSchemaVersionStateRequest request);
+
+  NotificationEventsCountRequest transformInboundNotificationEventsCountRequest(NotificationEventsCountRequest request);
+
+  UniqueConstraintsRequest transformInboundUniqueConstraintsRequest(UniqueConstraintsRequest request);
+
+  UniqueConstraintsResponse transformOutboundUniqueConstraintsResponse(UniqueConstraintsResponse response);
+
+  NotNullConstraintsRequest transformInboundNotNullConstraintsRequest(NotNullConstraintsRequest request);
+
+  NotNullConstraintsResponse transformOutboundNotNullConstraintsResponse(NotNullConstraintsResponse response);
+
+  DefaultConstraintsRequest transformInboundDefaultConstraintsRequest(DefaultConstraintsRequest request);
+
+  DefaultConstraintsResponse transformOutboundDefaultConstraintsResponse(DefaultConstraintsResponse response);
+
+  CheckConstraintsRequest transformInboundCheckConstraintsRequest(CheckConstraintsRequest request);
+
+  CheckConstraintsResponse transformOutboundCheckConstraintsResponse(CheckConstraintsResponse response);
+
+  CreationMetadata transformInboundCreationMetadata(CreationMetadata request);
+
   @Override
   long getLatency();
 }