restconf/restconf-client-impl/src/main/java/org/opendaylight/yangtools/restconf/client/AbstractDataStore.java

   1 package org.opendaylight.yangtools.restconf.client;
   2
   3 import com.google.common.base.Function;
   4 import com.google.common.base.Optional;
   5 import com.google.common.util.concurrent.ListenableFuture;
   6 import com.sun.jersey.api.client.ClientResponse;
   7 import java.io.InputStream;
   8 import java.util.Map.Entry;
   9 import org.opendaylight.yangtools.binding.data.codec.impl.BindingNormalizedNodeCodecRegistry;
  10 import org.opendaylight.yangtools.restconf.client.api.data.Datastore;
  11 import org.opendaylight.yangtools.restconf.client.api.data.DefaultRetrievalStrategy;
  12 import org.opendaylight.yangtools.restconf.client.api.data.RetrievalStrategy;
  13 import org.opendaylight.yangtools.restconf.utils.RestconfUtils;
  14 import org.opendaylight.yangtools.yang.binding.DataObject;
  15 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  16 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  17 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
  18 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
  19
  20 public abstract class AbstractDataStore implements Datastore {
  21
  22     private final RestconfClientImpl client;
  23
  24     public AbstractDataStore(RestconfClientImpl client) {
  25         super();
  26         this.client = client;
  27     }
  28
  29     protected RestconfClientImpl getClient() {
  30         return client;
  31     }
  32
  33     @Override
  34     public <T extends DataObject> ListenableFuture<Optional<T>> readData(InstanceIdentifier<T> path) {
  35         return readData(path, DefaultRetrievalStrategy.getInstance());
  36     }
  37
  38     @Override
  39     public <T extends DataObject> ListenableFuture<Optional<T>> readData(final InstanceIdentifier<T> path,
  40             RetrievalStrategy strategy) {
  41
  42         final SchemaContext schemaContext = client.getSchemaContext();
  43         final BindingNormalizedNodeCodecRegistry mappingService = client.getMappingService();
  44         final YangInstanceIdentifier domPath = mappingService.toYangInstanceIdentifier(path);
  45         final Entry<String, DataSchemaNode> pathWithSchema = RestconfUtils.toRestconfIdentifier(domPath,schemaContext);
  46         String restconfPath = getStorePrefix() + pathWithSchema.getKey();
  47
  48         return client.get(restconfPath, "application/xml",new Function<ClientResponse, Optional<T>>() {
  49
  50             @SuppressWarnings("unchecked")
  51             @Override
  52             public com.google.common.base.Optional<T> apply(ClientResponse response) {
  53                 switch (response.getStatus()) {
  54                 case 200: // Status OK
  55                     DataObject dataObject = deserialize(domPath,response.getEntityInputStream());
  56                     return (Optional<T>) Optional.of(dataObject);
  57                 case 404: // Status Not Found
  58                     return Optional.<T> absent();
  59                 default:
  60                     throw new IllegalStateException("Failed : HTTP error code : " + response.getStatus());
  61                 }
  62             }
  63
  64
  65         });
  66     }
  67
  68     protected final DataObject deserialize(YangInstanceIdentifier domPath, InputStream entityInputStream) {
  69         throw new UnsupportedOperationException("Not implemented yet.");
  70     }
  71
  72     protected abstract String getStorePrefix();
  73
  74 }