package org.opendaylight.yangtools.yang.data.codec.gson;
import com.google.common.base.Preconditions;
-
+import com.google.gson.stream.JsonWriter;
+import java.io.IOException;
import java.net.URI;
-
+import javax.annotation.Nonnull;
+import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
import org.opendaylight.yangtools.yang.data.util.AbstractModuleStringInstanceIdentifierCodec;
+import org.opendaylight.yangtools.yang.data.util.DataSchemaContextTree;
+import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
+import org.opendaylight.yangtools.yang.model.api.LeafSchemaNode;
import org.opendaylight.yangtools.yang.model.api.Module;
import org.opendaylight.yangtools.yang.model.api.SchemaContext;
-final class JSONStringInstanceIdentifierCodec extends AbstractModuleStringInstanceIdentifierCodec {
+final class JSONStringInstanceIdentifierCodec extends AbstractModuleStringInstanceIdentifierCodec
+ implements JSONCodec<YangInstanceIdentifier> {
+ private final DataSchemaContextTree dataContextTree;
+ private final JSONCodecFactory codecFactory;
private final SchemaContext context;
- JSONStringInstanceIdentifierCodec(final SchemaContext context) {
+ JSONStringInstanceIdentifierCodec(final SchemaContext context, final JSONCodecFactory jsonCodecFactory) {
this.context = Preconditions.checkNotNull(context);
+ this.dataContextTree = DataSchemaContextTree.from(context);
+ this.codecFactory = Preconditions.checkNotNull(jsonCodecFactory);
}
@Override
- protected Module moduleForPrefix(final String prefix) {
+ protected Module moduleForPrefix(@Nonnull final String prefix) {
return context.findModuleByName(prefix, null);
}
@Override
- protected String prefixForNamespace(final URI namespace) {
+ protected String prefixForNamespace(@Nonnull final URI namespace) {
final Module module = context.findModuleByNamespaceAndRevision(namespace, null);
return module == null ? null : module.getName();
}
+
+ @Nonnull
+ @Override
+ protected DataSchemaContextTree getDataContextTree() {
+ return dataContextTree;
+ }
+
+ @Override
+ protected Object deserializeKeyValue(final DataSchemaNode schemaNode, final String value) {
+ Preconditions.checkNotNull(schemaNode, "schemaNode cannot be null");
+ Preconditions.checkArgument(schemaNode instanceof LeafSchemaNode, "schemaNode must be of type LeafSchemaNode");
+ final JSONCodec<?> objectJSONCodec = codecFactory.codecFor((LeafSchemaNode) schemaNode);
+ return objectJSONCodec.parseValue(null, value);
+ }
+
+ @Override
+ public Class<YangInstanceIdentifier> getDataType() {
+ return YangInstanceIdentifier.class;
+ }
+
+ @Override
+ public YangInstanceIdentifier parseValue(final Object ctx, final String str) {
+ return deserialize(str);
+ }
+
+ @Override
+ public void writeValue(final JsonWriter ctx, final YangInstanceIdentifier value) throws IOException {
+ ctx.value(serialize(value));
+ }
}