tests/honeynode/2.2.1/minimal-distribution-core/src/main/java/io/fd/honeycomb/infra/distro/data/PersistingDataTreeProvider.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.data;
  18
  19 import com.google.inject.Inject;
  20 import com.google.inject.name.Named;
  21 import io.fd.honeycomb.binding.init.ProviderTrait;
  22 import io.fd.honeycomb.data.impl.PersistingDataTreeAdapter;
  23 import io.fd.honeycomb.infra.distro.cfgattrs.HoneycombConfiguration;
  24 import io.fd.honeycomb.infra.distro.data.context.ContextPipelineModule;
  25 import java.nio.file.Paths;
  26 import org.opendaylight.mdsal.dom.api.DOMSchemaService;
  27 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTree;
  28 import org.opendaylight.yangtools.yang.data.api.schema.tree.TreeType;
  29
  30 public abstract class PersistingDataTreeProvider extends ProviderTrait<DataTree> {
  31
  32     @Inject
  33     private DOMSchemaService schemaService;
  34     @Inject
  35     protected HoneycombConfiguration config;
  36
  37     @Override
  38     public DataTree create() {
  39         return isEnabled()
  40                 ? new PersistingDataTreeAdapter(getDelegate(), schemaService, Paths.get(getPath()))
  41                 : getDelegate();
  42     }
  43
  44     public abstract String getPath();
  45
  46     public abstract TreeType getType();
  47
  48     public abstract DataTree getDelegate();
  49
  50     protected abstract boolean isEnabled();
  51
  52     public static final class ConfigPersistingDataTreeProvider extends PersistingDataTreeProvider {
  53
  54         @Inject
  55         @Named(ConfigAndOperationalPipelineModule.HONEYCOMB_CONFIG_NONPERSIST)
  56         private DataTree delegate;
  57
  58         @Override
  59         public String getPath() {
  60             return config.peristConfigPath;
  61         }
  62
  63         @Override
  64         public TreeType getType() {
  65             return TreeType.CONFIGURATION;
  66         }
  67
  68         @Override
  69         public DataTree getDelegate() {
  70             return delegate;
  71         }
  72
  73         @Override
  74         protected boolean isEnabled() {
  75             return config.isConfigPersistenceEnabled();
  76         }
  77     }
  78
  79     public static final class ContextPersistingDataTreeProvider extends PersistingDataTreeProvider {
  80
  81         @Inject
  82         @Named(ContextPipelineModule.HONEYCOMB_CONTEXT_NOPERSIST)
  83         private DataTree delegate;
  84
  85         @Override
  86         public String getPath() {
  87             return config.peristContextPath;
  88         }
  89
  90         @Override
  91         public TreeType getType() {
  92             return TreeType.OPERATIONAL;
  93         }
  94
  95         @Override
  96         public DataTree getDelegate() {
  97             return delegate;
  98         }
  99
 100         @Override
 101         protected boolean isEnabled() {
 102             return config.isContextPersistenceEnabled();
 103         }
 104
 105     }
 106 }