Improve ModuleInfoSnapshotBuilder API
[mdsal.git] / binding / mdsal-binding-runtime-spi / src / main / java / org / opendaylight / mdsal / binding / runtime / spi / ModuleInfoSnapshotBuilder.java
1 /*
2  * Copyright (c) 2020 PANTHEON.tech, s.r.o. 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 package org.opendaylight.mdsal.binding.runtime.spi;
9
10 import com.google.common.annotations.Beta;
11 import java.util.Arrays;
12 import java.util.List;
13 import java.util.NoSuchElementException;
14 import org.eclipse.jdt.annotation.NonNull;
15 import org.opendaylight.mdsal.binding.runtime.api.ModuleInfoSnapshot;
16 import org.opendaylight.yangtools.concepts.CheckedBuilder;
17 import org.opendaylight.yangtools.yang.binding.YangModuleInfo;
18 import org.opendaylight.yangtools.yang.model.parser.api.YangParserFactory;
19
20 @Beta
21 public final class ModuleInfoSnapshotBuilder implements CheckedBuilder<ModuleInfoSnapshot, NoSuchElementException> {
22     private final ModuleInfoSnapshotResolver registry;
23
24     public ModuleInfoSnapshotBuilder(final String name, final YangParserFactory parserFactory) {
25         registry = new ModuleInfoSnapshotResolver(name, parserFactory);
26     }
27
28     public @NonNull ModuleInfoSnapshotBuilder add(final YangModuleInfo info) {
29         return add(List.of(info));
30     }
31
32     public @NonNull ModuleInfoSnapshotBuilder add(final YangModuleInfo... infos) {
33         return add(Arrays.asList(infos));
34     }
35
36     public @NonNull ModuleInfoSnapshotBuilder add(final Iterable<? extends YangModuleInfo> infos) {
37         registry.registerModuleInfos(infos);
38         return this;
39     }
40
41     @Override
42     public ModuleInfoSnapshot build() {
43         return registry.takeSnapshot();
44     }
45 }