infra/minimal-distribution/src/main/java/io/fd/honeycomb/infra/distro/data/PersistingDataTreeProvider.groovy

   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 groovy.transform.ToString
  22 import groovy.util.logging.Slf4j
  23 import io.fd.honeycomb.data.impl.PersistingDataTreeAdapter
  24 import io.fd.honeycomb.infra.distro.ProviderTrait
  25 import io.fd.honeycomb.infra.distro.cfgattrs.HoneycombConfiguration
  26 import io.fd.honeycomb.infra.distro.data.context.ContextPipelineModule
  27 import org.opendaylight.controller.sal.core.api.model.SchemaService
  28 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTree
  29 import org.opendaylight.yangtools.yang.data.api.schema.tree.TreeType
  30
  31 import java.nio.file.Paths
  32
  33 @Slf4j
  34 @ToString
  35 abstract class PersistingDataTreeProvider extends ProviderTrait<DataTree> {
  36
  37     @Inject
  38     SchemaService schemaService
  39     @Inject
  40     HoneycombConfiguration config
  41
  42     def create() {
  43         new PersistingDataTreeAdapter(delegate, schemaService, Paths.get(path))
  44     }
  45
  46     abstract String getPath()
  47     abstract TreeType getType()
  48     abstract DataTree getDelegate()
  49
  50     static class ConfigPersistingDataTreeProvider extends PersistingDataTreeProvider {
  51
  52         @Inject
  53         @Named(ConfigAndOperationalPipelineModule.HONEYCOMB_CONFIG_NONPERSIST)
  54         DataTree delegate
  55
  56         String getPath() { config.peristConfigPath }
  57         TreeType getType() { TreeType.CONFIGURATION }
  58         DataTree getDelegate() { return delegate }
  59     }
  60
  61     static class ContextPersistingDataTreeProvider extends PersistingDataTreeProvider {
  62
  63         @Inject
  64         @Named(ContextPipelineModule.HONEYCOMB_CONTEXT_NOPERSIST)
  65         DataTree delegate
  66
  67         String getPath() { config.peristContextPath }
  68         TreeType getType() { TreeType.OPERATIONAL }
  69         DataTree getDelegate() { return delegate }
  70     }
  71 }