Null condition check

sivaponting · Aug 3, 2023 · 55beeb2 · 55beeb2
1 parent 32fb8a3
commit 55beeb2
Showing 1 changed file with 16 additions and 5 deletions.
diff --git a/...ance-core/src/main/java/com/hotels/bdp/waggledance/mapping/model/DatabaseMappingImpl.java b/...ance-core/src/main/java/com/hotels/bdp/waggledance/mapping/model/DatabaseMappingImpl.java
@@ -604,22 +604,31 @@ public AllocateTableWriteIdsRequest transformInboundAllocateTableWriteIdsRequest
 
   @Override
   public AlterISchemaRequest transformInboundAlterISchemaRequest(AlterISchemaRequest request) {
-    request.setName(transformInboundISchemaName(request.getName()));
-    request.setNewSchema(transformInboundISchema(request.getNewSchema()));
+    if(request.getName() !=null) {
+      request.setName(transformInboundISchemaName(request.getName()));
+    }
+    if(request.getNewSchema() != null) {
+      request.setNewSchema(transformInboundISchema(request.getNewSchema()));
+    }
     return request;
   }
 
 
   @Override
   public SchemaVersion transformInboundSchemaVersion(SchemaVersion schemaVersion) {
-    schemaVersion.getSchema().setDbName(transformInboundDatabaseName(schemaVersion.getSchema().getDbName()));
+    if(schemaVersion.getSchema() !=null ) {
+      schemaVersion.getSchema().setDbName(transformInboundDatabaseName(schemaVersion.getSchema().getDbName()));
+    }
     return schemaVersion;
   }
 
 
   @Override
   public SchemaVersion transformOutboundSchemaVersion(SchemaVersion schemaVersion) {
-    schemaVersion.getSchema().setDbName(metaStoreMapping.transformOutboundDatabaseName(schemaVersion.getSchema().getDbName()));
+    if(schemaVersion.getSchema() !=null ) {
+      schemaVersion.getSchema().setDbName(metaStoreMapping.transformOutboundDatabaseName(
+              schemaVersion.getSchema().getDbName()));
+    }
     return schemaVersion;
   }
 
@@ -692,7 +701,9 @@ public AddCheckConstraintRequest transformInboundAddCheckConstraintRequest(AddCh
   @Override
   public FindSchemasByColsResp transformOutboundFindSchemasByColsResp(FindSchemasByColsResp response) {
     for(SchemaVersionDescriptor schemaVersionDescriptor: response.getSchemaVersions()) {
-      schemaVersionDescriptor.setSchema(transformOutboundISchemaName(schemaVersionDescriptor.getSchema()));
+      if(schemaVersionDescriptor.getSchema() != null) {
+        schemaVersionDescriptor.setSchema(transformOutboundISchemaName(schemaVersionDescriptor.getSchema()));
+      }
     }
     return response;
   }