- if (cap.getCapabilitySchema().isPresent()) {
- final SchemaBuilder builder = new SchemaBuilder();
-
- Preconditions.checkState(isValidModuleCapability(cap));
-
- builder.setNamespace(new Uri(cap.getModuleNamespace().get()));
-
- final String version = cap.getRevision().get();
- builder.setVersion(version);
-
- final String identifier = cap.getModuleName().get();
- builder.setIdentifier(identifier);
-
- builder.setFormat(Yang.class);
-
- builder.setLocation(transformLocations(cap.getLocation()));
-
- builder.withKey(new SchemaKey(Yang.class, identifier, version));
-
- schemas.add(builder.build());
+ if (isValidModuleCapability(cap)) {
+ final SchemaKey key = new SchemaKey(Yang.class, cap.getModuleName().get(),
+ cap.getRevision().orElse(""));
+ schemas.put(key, new SchemaBuilder()
+ .withKey(key)
+ .setNamespace(new Uri(cap.getModuleNamespace().get()))
+ .setLocation(transformLocations(cap.getLocation()))
+ .build());