diff options
author | Jan Srnicek <jsrnicek@cisco.com> | 2017-08-30 14:55:04 +0200 |
---|---|---|
committer | Marek Gradzki <mgradzki@cisco.com> | 2017-09-05 15:39:31 +0000 |
commit | bb9141c63012d0305ac65da763af3fb0c427076f (patch) | |
tree | 85b75f316f4c7705d34c6c4c4a7b324d0fe8fc02 /common/yang-whitelist/yang-whitelist-impl | |
parent | 0d4a04e1452e19ce96c308cd45a430a3f5b72f04 (diff) |
HONEYCOMB-389 - Static yang binding api and impl
Also provides ignore option for default module list generation
Change-Id: I9e10215491c964e6e88f90d6a3ad4335c94ec7d2
Signed-off-by: Jan Srnicek <jsrnicek@cisco.com>
Diffstat (limited to 'common/yang-whitelist/yang-whitelist-impl')
8 files changed, 378 insertions, 0 deletions
diff --git a/common/yang-whitelist/yang-whitelist-impl/asciidoc/Readme.adoc b/common/yang-whitelist/yang-whitelist-impl/asciidoc/Readme.adoc new file mode 100644 index 000000000..c5794045c --- /dev/null +++ b/common/yang-whitelist/yang-whitelist-impl/asciidoc/Readme.adoc @@ -0,0 +1,3 @@ += whitelist-impl + +Overview of whitelist-impl
\ No newline at end of file diff --git a/common/yang-whitelist/yang-whitelist-impl/pom.xml b/common/yang-whitelist/yang-whitelist-impl/pom.xml new file mode 100644 index 000000000..0e1f02511 --- /dev/null +++ b/common/yang-whitelist/yang-whitelist-impl/pom.xml @@ -0,0 +1,79 @@ +<?xml version="1.0" encoding="UTF-8"?> +<!-- + ~ Copyright (c) 2017 Cisco and/or its affiliates. + ~ + ~ Licensed under the Apache License, Version 2.0 (the "License"); + ~ you may not use this file except in compliance with the License. + ~ You may obtain a copy of the License at: + ~ + ~ http://www.apache.org/licenses/LICENSE-2.0 + ~ + ~ Unless required by applicable law or agreed to in writing, software + ~ distributed under the License is distributed on an "AS IS" BASIS, + ~ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + ~ See the License for the specific language governing permissions and + ~ limitations under the License. + --> + +<project xmlns="http://maven.apache.org/POM/4.0.0" + xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" + xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd"> + <!-- Cannot have parent , to break cycle, its used in build --> + <modelVersion>4.0.0</modelVersion> + + <groupId>io.fd.honeycomb.yang</groupId> + <artifactId>yang-whitelist-impl</artifactId> + <version>1.17.10-SNAPSHOT</version> + + <properties> + <jaxb-runtime.version>2.2.11</jaxb-runtime.version> + <slf4j-simple.version>1.7.25</slf4j-simple.version> + </properties> + + <dependencies> + <dependency> + <groupId>io.fd.honeycomb.yang</groupId> + <artifactId>yang-whitelist-api</artifactId> + <version>${project.version}</version> + </dependency> + <dependency> + <groupId>org.glassfish.jaxb</groupId> + <artifactId>jaxb-runtime</artifactId> + <version>${jaxb-runtime.version}</version> + </dependency> + <dependency> + <groupId>com.google.code.findbugs</groupId> + <artifactId>jsr305</artifactId> + <version>3.0.2</version> + </dependency> + <dependency> + <groupId>org.slf4j</groupId> + <artifactId>slf4j-api</artifactId> + <version>${slf4j-simple.version}</version> + </dependency> + <dependency> + <groupId>junit</groupId> + <artifactId>junit</artifactId> + <version>4.12</version> + <scope>test</scope> + </dependency> + <dependency> + <groupId>com.google.guava</groupId> + <artifactId>guava</artifactId> + <version>19.0</version> + <scope>test</scope> + </dependency> + <dependency> + <groupId>org.hamcrest</groupId> + <artifactId>hamcrest-core</artifactId> + <version>1.3</version> + <scope>test</scope> + </dependency> + <dependency> + <groupId>org.slf4j</groupId> + <artifactId>slf4j-simple</artifactId> + <version>${slf4j-simple.version}</version> + <scope>test</scope> + </dependency> + </dependencies> +</project>
\ No newline at end of file diff --git a/common/yang-whitelist/yang-whitelist-impl/src/main/java/io/fd/honeycomb/yang/JAXBContextHolder.java b/common/yang-whitelist/yang-whitelist-impl/src/main/java/io/fd/honeycomb/yang/JAXBContextHolder.java new file mode 100644 index 000000000..ed1aee645 --- /dev/null +++ b/common/yang-whitelist/yang-whitelist-impl/src/main/java/io/fd/honeycomb/yang/JAXBContextHolder.java @@ -0,0 +1,37 @@ +/* + * Copyright (c) 2017 Cisco and/or its affiliates. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at: + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package io.fd.honeycomb.yang; + +import javax.annotation.Nonnull; +import javax.xml.bind.JAXBContext; +import javax.xml.bind.JAXBException; + +abstract class JAXBContextHolder { + private final JAXBContext ctx; + + protected JAXBContextHolder(@Nonnull final Class<?>... ctxClasses) { + try { + ctx = JAXBContext.newInstance(ctxClasses); + } catch (JAXBException e) { + throw new IllegalStateException("Unable to create JAXB context", e); + } + } + + protected JAXBContext getCtx() { + return ctx; + } +} diff --git a/common/yang-whitelist/yang-whitelist-impl/src/main/java/io/fd/honeycomb/yang/YangModuleWhitelistReader.java b/common/yang-whitelist/yang-whitelist-impl/src/main/java/io/fd/honeycomb/yang/YangModuleWhitelistReader.java new file mode 100644 index 000000000..f4897c275 --- /dev/null +++ b/common/yang-whitelist/yang-whitelist-impl/src/main/java/io/fd/honeycomb/yang/YangModuleWhitelistReader.java @@ -0,0 +1,53 @@ +/* + * Copyright (c) 2017 Cisco and/or its affiliates. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at: + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package io.fd.honeycomb.yang; + +import static java.lang.String.format; + +import java.nio.file.Path; +import javax.annotation.Nonnull; +import javax.xml.bind.JAXBException; +import javax.xml.bind.Unmarshaller; + +public class YangModuleWhitelistReader extends JAXBContextHolder { + + public YangModuleWhitelistReader() { + super(YangModuleWhitelist.class); + } + + @Nonnull + public YangModuleWhitelist read(@Nonnull final Path path) { + final Unmarshaller unmarshaller = createUnmarshaller(); + return YangModuleWhitelist.class.cast(readWhitelist(path, unmarshaller)); + } + + private static Object readWhitelist(final Path path, final Unmarshaller unmarshaller) { + try { + return unmarshaller.unmarshal(path.toFile()); + } catch (JAXBException e) { + throw new IllegalStateException(format("Unable to read whitelist from %s", path), e); + } + } + + private Unmarshaller createUnmarshaller() { + try { + return getCtx().createUnmarshaller(); + } catch (JAXBException e) { + throw new IllegalStateException("Unable to create unmarshaller", e); + } + } +} diff --git a/common/yang-whitelist/yang-whitelist-impl/src/main/java/io/fd/honeycomb/yang/YangModuleWhitelistWriter.java b/common/yang-whitelist/yang-whitelist-impl/src/main/java/io/fd/honeycomb/yang/YangModuleWhitelistWriter.java new file mode 100644 index 000000000..d14b2428f --- /dev/null +++ b/common/yang-whitelist/yang-whitelist-impl/src/main/java/io/fd/honeycomb/yang/YangModuleWhitelistWriter.java @@ -0,0 +1,80 @@ +/* + * Copyright (c) 2017 Cisco and/or its affiliates. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at: + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package io.fd.honeycomb.yang; + +import java.nio.file.Path; +import java.util.Objects; +import javax.annotation.Nonnull; +import javax.xml.bind.JAXBContext; +import javax.xml.bind.JAXBException; +import javax.xml.bind.Marshaller; +import javax.xml.bind.PropertyException; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; + +public class YangModuleWhitelistWriter extends JAXBContextHolder { + + private static final Logger LOG = LoggerFactory.getLogger(YangModuleWhitelistWriter.class); + + public YangModuleWhitelistWriter() { + super(YangModuleWhitelist.class); + } + + /** + * Output serialized whitelist on specified path + * + * @param whitelist whitelist configuration + * @param outPath output path + */ + public void write(@Nonnull final YangModuleWhitelist whitelist, + @Nonnull final Path outPath, + final boolean formatOutput) { + Objects.requireNonNull(whitelist, "Cannot white null whitelist"); + // mashaller is not synchronized and lightweight, best practice is to create it per request(as opose to ctx, + // that should be created just once) + final Marshaller marshaller = createMarshaller(getCtx()); + setupPrettyPrint(marshaller, formatOutput); + whiteWhitelist(whitelist, outPath, marshaller); + } + + private static void whiteWhitelist(final YangModuleWhitelist whitelist, final Path outPath, + final Marshaller marshaller) { + try { + LOG.debug("Writing whitelist {} to file {}", whitelist, outPath); + marshaller.marshal(whitelist, outPath.toFile()); + LOG.debug("Whitelist successfully written to file {}", outPath); + } catch (JAXBException e) { + throw new IllegalStateException("Unable to write whitelist", e); + } + } + + private static void setupPrettyPrint(final Marshaller marshaller, final boolean value) { + try { + marshaller.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, value); + } catch (PropertyException e) { + throw new IllegalStateException("Unable to setup pretty print"); + } + } + + private static Marshaller createMarshaller(final JAXBContext ctx) { + try { + return ctx.createMarshaller(); + } catch (JAXBException e) { + throw new IllegalStateException("Unable to create marshaller", e); + } + } +} diff --git a/common/yang-whitelist/yang-whitelist-impl/src/test/java/io/fd/honeycomb/yang/YangModuleWhitelistReaderTest.java b/common/yang-whitelist/yang-whitelist-impl/src/test/java/io/fd/honeycomb/yang/YangModuleWhitelistReaderTest.java new file mode 100644 index 000000000..6ab917a09 --- /dev/null +++ b/common/yang-whitelist/yang-whitelist-impl/src/test/java/io/fd/honeycomb/yang/YangModuleWhitelistReaderTest.java @@ -0,0 +1,56 @@ +/* + * Copyright (c) 2017 Cisco and/or its affiliates. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at: + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package io.fd.honeycomb.yang; + +import static org.junit.Assert.assertEquals; +import static org.junit.Assert.assertNotNull; +import static org.junit.Assert.assertThat; + +import java.nio.file.Paths; +import java.util.List; +import org.hamcrest.CoreMatchers; +import org.junit.Before; +import org.junit.Test; + +public class YangModuleWhitelistReaderTest { + + private YangModuleWhitelistReader reader; + + @Before + public void init() throws Exception { + reader = new YangModuleWhitelistReader(); + } + + @Test + public void read() throws Exception { + final YangModuleWhitelist whitelist = reader.read( + Paths.get(this.getClass().getClassLoader().getResource("expected-whitelist.xml").getPath())); + + assertNotNull(whitelist); + final List<Module> modules = whitelist.getModules(); + assertEquals(2, modules.size()); + + final Module moduleA = new Module(); + final Module moduleB = new Module(); + moduleA.setPckg("module.a.package"); + moduleA.setDescription("desc"); + + moduleB.setPckg("module.b.package"); + + assertThat(modules, CoreMatchers.hasItems(moduleA, moduleB)); + } +}
\ No newline at end of file diff --git a/common/yang-whitelist/yang-whitelist-impl/src/test/java/io/fd/honeycomb/yang/YangModuleWhitelistWriterTest.java b/common/yang-whitelist/yang-whitelist-impl/src/test/java/io/fd/honeycomb/yang/YangModuleWhitelistWriterTest.java new file mode 100644 index 000000000..c0ca170b8 --- /dev/null +++ b/common/yang-whitelist/yang-whitelist-impl/src/test/java/io/fd/honeycomb/yang/YangModuleWhitelistWriterTest.java @@ -0,0 +1,69 @@ +/* + * Copyright (c) 2017 Cisco and/or its affiliates. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at: + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package io.fd.honeycomb.yang; + + +import static org.junit.Assert.assertEquals; + +import com.google.common.collect.ImmutableList; +import com.google.common.io.Resources; +import java.io.IOException; +import java.nio.charset.StandardCharsets; +import java.nio.file.Files; +import java.nio.file.Path; +import java.util.stream.Collectors; +import org.junit.After; +import org.junit.Before; +import org.junit.Test; + +public class YangModuleWhitelistWriterTest { + + private YangModuleWhitelistWriter writer; + private Path path; + + @Before + public void init() throws IOException { + writer = new YangModuleWhitelistWriter(); + path = Files.createTempFile("tmp", "whitelist"); + } + + @Test + public void write() throws Exception { + final Module moduleA = new Module(); + final Module moduleB = new Module(); + moduleA.setPckg("module.a.package"); + moduleA.setDescription("desc"); + + moduleB.setPckg("module.b.package"); + + final YangModuleWhitelist whitelist = new YangModuleWhitelist(); + whitelist.setModules(ImmutableList.of(moduleA, moduleB)); + + writer.write(whitelist, path, false); + final String output = Files.readAllLines(path).stream().collect(Collectors.joining()); + final String expectedOutput = Resources + .toString(this.getClass().getClassLoader().getResource("expected-whitelist.xml"), + StandardCharsets.UTF_8); + assertEquals(expectedOutput, output); + } + + + @After + public void clean() { + path.toFile().delete(); + } +}
\ No newline at end of file diff --git a/common/yang-whitelist/yang-whitelist-impl/src/test/resources/expected-whitelist.xml b/common/yang-whitelist/yang-whitelist-impl/src/test/resources/expected-whitelist.xml new file mode 100644 index 000000000..6ec5c22ae --- /dev/null +++ b/common/yang-whitelist/yang-whitelist-impl/src/test/resources/expected-whitelist.xml @@ -0,0 +1 @@ +<?xml version="1.0" encoding="UTF-8" standalone="yes"?><yangModuleWhitelist><modules><module><description>desc</description><package>module.a.package</package></module><module><package>module.b.package</package></module></modules></yangModuleWhitelist>
\ No newline at end of file |