private final Map.Entry<DataSchema, Schema> findUnionMemberSchema(
Object value, UnionDataSchema unionDataSchema, Schema avroSchema) {
int index = _genericData.resolveUnion(avroSchema, value);
Schema memberAvroSchema = avroSchema.getTypes().get(index);
String key;
switch (memberAvroSchema.getType()) {
case ENUM:
case FIXED:
case RECORD:
key = memberAvroSchema.getFullName();
break;
default:
key = memberAvroSchema.getType().toString().toLowerCase();
}
DataSchema memberDataSchema = unionDataSchema.getType(key);
if (memberDataSchema == null) {
for (DataSchema dataSchema : unionDataSchema.getTypes()) {
AvroOverride avroOverride = getAvroOverride(dataSchema);
if (avroOverride != null) {
if (avroOverride.getAvroSchemaFullName().equals(key)) {
memberDataSchema = dataSchema;
break;
}
}
}
}
if (memberDataSchema == null) {
appendMessage("cannot find %1$s in union %2$s for value %3$s", key, unionDataSchema, value);
return null;
}
return new AbstractMap.SimpleEntry<DataSchema, Schema>(memberDataSchema, memberAvroSchema);
}
/** Register a new schema with this repository. * */
public AvroSchemaComposer add(Schema schema) {
for (String alias : schema.getAliases()) {
schemas.put(alias, schema);
}
schemas.put(schema.getFullName(), schema);
mostRecent = schema;
return this;
}
public static Schema parseSchema(String schemaString) {
String completeSchema = resolveSchema(schemaString);
Schema schema = Schema.parse(completeSchema);
String name = schema.getFullName();
schemas.put(name, schema);
return schema;
}
protected Map.Entry<String, Schema> findUnionMember(DataSchema dataSchema, Schema avroSchema) {
AvroOverride avroOverride = getAvroOverride(dataSchema);
String key =
(avroOverride == null
? dataSchema.getUnionMemberKey()
: avroOverride.getAvroSchemaFullName());
List<Schema> members = avroSchema.getTypes();
for (Schema member : members) {
String name;
switch (member.getType()) {
case ENUM:
case FIXED:
case RECORD:
name = member.getFullName();
break;
default:
name = member.getType().toString().toLowerCase();
}
if (name.equals(key)) return new AbstractMap.SimpleEntry<String, Schema>(name, member);
}
appendMessage("cannot find %1$s in union %2$s", key, avroSchema);
return null;
}
@Before
public void beforeTest() throws IOException, DeltaCalculatorException, ControlServiceException {
String dataSchema =
OperationsServiceIT.getResourceAsString(OperationsServiceIT.DATA_SCHEMA_LOCATION);
PROFILE_BYTES = avroConverter.encode(ENDPOINT_PROFILE);
PROFILE_JSON = avroConverter.encodeToJson(ENDPOINT_PROFILE);
tenant = new TenantDto();
tenant.setName(CUSTOMER_ID);
tenant = userService.saveTenant(tenant);
assertNotNull(tenant);
assertNotNull(tenant.getId());
ApplicationDto applicationDto = new ApplicationDto();
applicationDto.setTenantId(tenant.getId());
applicationDto.setApplicationToken(APPLICATION_ID);
applicationDto.setName(APPLICATION_NAME);
applicationDto.setSequenceNumber(NEW_APPLICATION_SEQ_NUMBER);
applicationDto = applicationService.saveApp(applicationDto);
APP_TOKEN = applicationDto.getApplicationToken();
assertNotNull(applicationDto);
assertNotNull(applicationDto.getId());
application = applicationService.findAppById(applicationDto.getId());
EndpointGroupDto groupAll =
endpointService.findEndpointGroupsByAppId(application.getId()).get(0);
CTLSchemaDto profileCtlSchema = new CTLSchemaDto();
CtlSchemaMetaInfoDto metaInfo =
new CtlSchemaMetaInfoDto(
BasicEndpointProfile.SCHEMA$.getFullName(),
application.getTenantId(),
application.getId());
profileCtlSchema.setMetaInfo(metaInfo);
profileCtlSchema.setBody(BasicEndpointProfile.SCHEMA$.toString());
profileCtlSchema.setVersion(1);
profileCtlSchema.setDependencySet(new HashSet<CTLSchemaDto>());
profileCtlSchema = ctlService.saveCtlSchema(profileCtlSchema);
Schema schema = new Schema.Parser().parse(dataSchema);
CTLSchemaDto confCtlSchema = new CTLSchemaDto();
CtlSchemaMetaInfoDto confMetaInfo =
new CtlSchemaMetaInfoDto(
schema.getFullName(), application.getTenantId(), application.getId());
confCtlSchema.setMetaInfo(confMetaInfo);
confCtlSchema.setBody(schema.toString());
confCtlSchema.setVersion(CONF_SCHEMA_VERSION);
confCtlSchema.setDependencySet(new HashSet<CTLSchemaDto>());
confCtlSchema = ctlService.saveCtlSchema(confCtlSchema);
EndpointProfileSchemaDto profileSchemaObj = new EndpointProfileSchemaDto();
profileSchemaObj.setVersion(PROFILE_SCHEMA_VERSION);
profileSchemaObj.setCtlSchemaId(profileCtlSchema.getId());
profileSchemaObj.setApplicationId(application.getId());
EndpointProfileSchemaDto profileSchemaDto = profileService.saveProfileSchema(profileSchemaObj);
profileSchema = profileService.findProfileSchemaById(profileSchemaDto.getId());
EndpointGroupDto endpointGroup = new EndpointGroupDto();
endpointGroup.setApplicationId(application.getId());
endpointGroup.setName("Test group");
endpointGroup.setWeight(277);
endpointGroup.setDescription("Test Description");
endpointGroup = endpointService.saveEndpointGroup(endpointGroup);
ProfileFilterDto profileFilterObj = new ProfileFilterDto();
profileFilterObj.setApplicationId(application.getId());
profileFilterObj.setEndpointGroupId(endpointGroup.getId());
profileFilterObj.setBody("profileBody.contains(\"dummy\")");
profileFilterObj.setEndpointProfileSchemaId(profileSchema.getId());
profileFilter = profileService.saveProfileFilter(profileFilterObj);
profileService.activateProfileFilter(profileFilter.getId(), null);
confSchema = new ConfigurationSchemaDto();
confSchema.setApplicationId(application.getId());
confSchema.setVersion(CONF_SCHEMA_VERSION);
confSchema.setCtlSchemaId(confCtlSchema.getId());
try {
confSchema = configurationService.saveConfSchema(confSchema);
} catch (IncorrectParameterException e) {
Assert.fail("Can't generate schemas");
}
Assert.assertNotNull(confSchema);
Assert.assertNotNull(confSchema.getId());
egAllId = groupAll.getId();
pfAllId = profileFilter.getId();
ConfigurationDto confDto =
configurationService.findConfigurationByEndpointGroupIdAndVersion(
egAllId, CONF_SCHEMA_VERSION);
cfAllId = confDto.getId();
endpointConfiguration = new EndpointConfigurationDto();
endpointConfiguration.setConfiguration(confDto.getBody().getBytes(UTF_8));
endpointConfiguration.setConfigurationHash(
EndpointObjectHash.fromSha1(confDto.getBody()).getData());
endpointConfiguration = endpointService.saveEndpointConfiguration(endpointConfiguration);
assertNotNull(endpointConfiguration);
EndpointGroupStateDto egs = new EndpointGroupStateDto();
egs.setConfigurationId(cfAllId);
egs.setEndpointGroupId(egAllId);
egs.setProfileFilterId(pfAllId);
endpointProfile = new EndpointProfileDto();
endpointProfile.setApplicationId(application.getId());
endpointProfile.setEndpointKeyHash(Base64Utils.decodeFromString("EndpointId"));
endpointProfile.setClientProfileBody(PROFILE_JSON);
endpointProfile.setProfileHash(EndpointObjectHash.fromSha1(PROFILE_BYTES).getData());
endpointProfile.setConfigurationHash(endpointConfiguration.getConfigurationHash());
endpointProfile.setConfigurationVersion(CONF_SCHEMA_VERSION);
endpointProfile.setClientProfileVersion(PROFILE_VERSION);
endpointProfile.setGroupState(Collections.singletonList(egs));
endpointProfile = endpointService.saveEndpointProfile(endpointProfile);
assertNotNull(endpointProfile);
assertNotNull(endpointProfile.getId());
}