@Override
 public void setEntityAudit(int customerId, String userId, RECORD_TYPE recordType) {
   System.out.println("Setting logged in user info for " + recordType);
   if (entityAudit == null) {
     entityAudit = new EntityAudit();
   }
   if (recordType == RECORD_TYPE.ADD) {
     this.entityAudit.setCreatedBy(userId);
   } else {
     this.entityAudit.setUpdatedBy(userId);
   }
   setSystemInformation(recordType);
   if (this.communicationData == null) {
     this.communicationData = new java.util.ArrayList<CommunicationData>();
   }
   for (CommunicationData _communicationData : communicationData) {
     _communicationData.setEntityAudit(customerId, userId, recordType);
     _communicationData.setSystemInformation(recordType);
   }
   if (this.address == null) {
     this.address = new java.util.ArrayList<Address>();
   }
   for (Address _address : address) {
     _address.setEntityAudit(customerId, userId, recordType);
     _address.setSystemInformation(recordType);
   }
 }
 @Override
 public void setEntityAudit(int customerId, String userId) {
   if (entityAudit == null) {
     entityAudit = new EntityAudit();
   }
   if (getPrimaryKey() == null) {
     this.entityAudit.setCreatedBy(userId);
     this.systemInfo.setActiveStatus(1);
   } else {
     this.entityAudit.setUpdatedBy(userId);
   }
   if (this.communicationData == null) {
     this.communicationData = new java.util.ArrayList<CommunicationData>();
   }
   for (CommunicationData _communicationData : communicationData) {
     _communicationData.setEntityAudit(customerId, userId);
   }
   if (this.address == null) {
     this.address = new java.util.ArrayList<Address>();
   }
   for (Address _address : address) {
     _address.setEntityAudit(customerId, userId);
   }
 }