pce/src/main/java/org/opendaylight/transportpce/pce/gnpy/ServiceDataStoreOperationsImpl.java

   1 /*
   2  * Copyright © 2018 Orange, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8
   9 package org.opendaylight.transportpce.pce.gnpy;
  10
  11 import com.google.common.collect.FluentIterable;
  12 import com.google.gson.Gson;
  13 import com.google.gson.JsonObject;
  14 import com.google.gson.JsonParser;
  15 import com.google.gson.stream.JsonWriter;
  16 import java.io.FileWriter;
  17 import java.io.IOException;
  18 import java.io.StringWriter;
  19 import java.io.Writer;
  20 import java.nio.charset.StandardCharsets;
  21 import org.opendaylight.mdsal.binding.dom.codec.spi.BindingDOMCodecServices;
  22 import org.opendaylight.mdsal.binding.spec.reflect.BindingReflections;
  23 import org.opendaylight.yang.gen.v1.gnpy.gnpy.api.rev190103.GnpyApi;
  24 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  25 import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeStreamWriter;
  26 import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeWriter;
  27 import org.opendaylight.yangtools.yang.data.codec.gson.JSONCodecFactory;
  28 import org.opendaylight.yangtools.yang.data.codec.gson.JSONCodecFactorySupplier;
  29 import org.opendaylight.yangtools.yang.data.codec.gson.JSONNormalizedNodeStreamWriter;
  30 import org.opendaylight.yangtools.yang.data.codec.gson.JsonWriterFactory;
  31 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
  32
  33 public class ServiceDataStoreOperationsImpl implements ServiceDataStoreOperations {
  34
  35     private static final JsonParser PARSER = new JsonParser();
  36     private BindingDOMCodecServices bindingDOMCodecServices;
  37
  38     public ServiceDataStoreOperationsImpl(BindingDOMCodecServices bindingDOMCodecServices) throws GnpyException {
  39         this.bindingDOMCodecServices = bindingDOMCodecServices;
  40     }
  41
  42     @Override
  43     public String createJsonStringFromDataObject(final InstanceIdentifier<GnpyApi> id, GnpyApi object)
  44         throws GnpyException {
  45         final SchemaPath scPath = SchemaPath.create(FluentIterable
  46                 .from(id.getPathArguments())
  47                 .transform(input -> BindingReflections.findQName(input.getType())), true);
  48         /*
  49          * This function needs : - context - scPath.getParent() -
  50          * scPath.getLastComponent().getNamespace(), -
  51          * JsonWriterFactory.createJsonWriter(writer)
  52          */
  53
  54         JSONCodecFactory codecFactory = JSONCodecFactorySupplier.DRAFT_LHOTKA_NETMOD_YANG_JSON_02
  55                 .getShared(bindingDOMCodecServices.getRuntimeContext().getEffectiveModelContext());
  56         try (Writer writer = new StringWriter();
  57                 JsonWriter jsonWriter = JsonWriterFactory.createJsonWriter(writer, 2);) {
  58             NormalizedNodeStreamWriter jsonStreamWriter = JSONNormalizedNodeStreamWriter.createExclusiveWriter(
  59                     codecFactory, scPath.getParent(), scPath.getLastComponent().getNamespace(), jsonWriter);
  60             try (NormalizedNodeWriter nodeWriter = NormalizedNodeWriter.forStreamWriter(jsonStreamWriter)) {
  61                 nodeWriter.write(bindingDOMCodecServices.toNormalizedNode(id, object).getValue());
  62                 nodeWriter.flush();
  63             }
  64             JsonObject asJsonObject = PARSER.parse(writer.toString()).getAsJsonObject();
  65             return new Gson().toJson(asJsonObject);
  66         } catch (IOException e) {
  67             throw new GnpyException("Cannot convert data to Json string", e);
  68         }
  69     }
  70
  71     // Write the json as a string in a file
  72     @Override
  73     public void writeStringFile(String jsonString, String fileName) throws GnpyException {
  74         try (FileWriter file = new FileWriter(fileName,StandardCharsets.UTF_8)) {
  75             file.write(jsonString);
  76         } catch (IOException e) {
  77             throw new GnpyException("In ServiceDataStoreOperationsImpl : exception during file writing",e);
  78         }
  79     }
  80 }