tests/honeynode/minimal-distribution-core/src/main/java/io/fd/honeycomb/infra/distro/schema/SerializerProvider.java

   1 /*
   2  * Copyright (c) 2016 Cisco and/or its affiliates.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at:
   7  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 package io.fd.honeycomb.infra.distro.schema;
  18
  19 import com.google.inject.Inject;
  20 import io.fd.honeycomb.binding.init.ProviderTrait;
  21 import javassist.ClassPool;
  22 import org.opendaylight.controller.md.sal.binding.impl.BindingToNormalizedNodeCodec;
  23 import org.opendaylight.mdsal.binding.generator.impl.ModuleInfoBackedContext;
  24 import org.opendaylight.mdsal.binding.generator.util.BindingRuntimeContext;
  25 import org.opendaylight.mdsal.binding.generator.util.JavassistUtils;
  26 import org.opendaylight.yangtools.binding.data.codec.gen.impl.DataObjectSerializerGenerator;
  27 import org.opendaylight.yangtools.binding.data.codec.gen.impl.StreamWriterGenerator;
  28 import org.opendaylight.yangtools.binding.data.codec.impl.BindingNormalizedNodeCodecRegistry;
  29
  30 public class SerializerProvider extends ProviderTrait<BindingToNormalizedNodeCodec> {
  31
  32     @Inject
  33     private ModuleInfoBackedContext mibCtx;
  34
  35     @Override
  36     protected BindingToNormalizedNodeCodec create() {
  37         final DataObjectSerializerGenerator serializerGenerator =
  38             StreamWriterGenerator.create(JavassistUtils.forClassPool(ClassPool.getDefault()));
  39         BindingNormalizedNodeCodecRegistry codecRegistry = new BindingNormalizedNodeCodecRegistry(serializerGenerator);
  40         BindingRuntimeContext ctx = BindingRuntimeContext.create(mibCtx, mibCtx.getSchemaContext());
  41         codecRegistry.onBindingRuntimeContextUpdated(ctx);
  42         BindingToNormalizedNodeCodec codec = new BindingToNormalizedNodeCodec(mibCtx, codecRegistry);
  43         codec.onGlobalContextUpdated(mibCtx.getSchemaContext());
  44         return codec;
  45     }
  46 }