123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117 |
- package org.apache.archiva.configuration.util;
- /*
- * Licensed to the Apache Software Foundation (ASF) under one
- * or more contributor license agreements. See the NOTICE file
- * distributed with this work for additional information
- * regarding copyright ownership. The ASF licenses this file
- * to you 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.
- */
-
- import java.util.ArrayList;
- import java.util.Arrays;
- import java.util.HashMap;
- import java.util.List;
- import java.util.Map;
- import java.util.function.BiFunction;
- import java.util.function.Function;
- import java.util.stream.Collectors;
-
- /**
- * Helper class that can be used for mapping configuration keys (e.g. user configuration keys) to
- * archiva configuration objects.
- *
- * @param <T> The class used to retrieve the attribute data
- * @param <K> The class used to retrieve the data that is for prefix matching
- * @author Martin Stockhammer <martin_s@apache.org>
- * @since 3.0
- */
- public class ConfigMapper<T, K>
- {
- private final Map<String, Function<T, String>> stringFunctionMap = new HashMap<>( );
- private final Map<String, Function<T, Integer>> intFunctionMap = new HashMap<>( );
- private final Map<String, Function<T, Boolean>> booleanFunctionMap = new HashMap<>( );
- private final Map<String, BiFunction<String, K, String>> prefixStringFunctionMap = new HashMap<>( );
-
- public void addStringMapping( String attributeName, Function<T, String> mapping) {
- this.stringFunctionMap.put( attributeName, mapping );
- }
-
- public void addPrefixStringMapping(String prefix, BiFunction<String, K, String> mapping) {
- prefixStringFunctionMap.put( prefix, mapping );
- }
-
- public String getString( String attributeName, T instance) {
- return stringFunctionMap.get( attributeName ).apply( instance );
- }
-
- public String getPrefixString(String attributeName, K instance) {
- BiFunction<String, K, String> function = prefixStringFunctionMap.entrySet( ).stream( ).filter( entry -> attributeName.startsWith( entry.getKey( ) ) ).findFirst( )
- .map( entry -> entry.getValue( ) )
- .get( );
- return function.apply( attributeName, instance );
- }
-
- public boolean isStringMapping(String attributeName) {
- return stringFunctionMap.containsKey( attributeName );
- }
-
- public boolean isIntMapping(String attributeName) {
- return intFunctionMap.containsKey( attributeName );
- }
-
- public boolean isBooleanMapping(String attributeName) {
- return booleanFunctionMap.containsKey( attributeName );
- }
-
- public boolean isPrefixMapping(String attributeName) {
- return prefixStringFunctionMap.keySet( ).stream( ).anyMatch( prefix -> attributeName.startsWith( prefix ) );
- }
-
- public boolean isMapping(String attributeName) {
- return isStringMapping( attributeName ) || isIntMapping( attributeName ) || isBooleanMapping( attributeName );
- }
-
- public void addIntMapping( String attributeName, Function<T, Integer> mapping) {
- this.intFunctionMap.put( attributeName, mapping );
- }
-
- public int getInt( String attributeName, T instance) {
- return this.intFunctionMap.get( attributeName ).apply( instance );
- }
-
- public void addBooleanMapping( String attributeName, Function<T, Boolean> mapping) {
- this.booleanFunctionMap.put( attributeName, mapping );
- }
-
- public boolean getBoolean( String attributeName, T instance) {
- return this.booleanFunctionMap.get( attributeName ).apply( instance );
- }
-
- public List<String> getStringAttributes() {
- return new ArrayList<>( stringFunctionMap.keySet( ) );
- }
-
- public List<String> getIntAttributes() {
- return new ArrayList<>( intFunctionMap.keySet( ) );
- }
-
- public List<String> getBooleanAttributes() {
- return new ArrayList<>( booleanFunctionMap.keySet( ) );
- }
-
- public List<String> getAllAttributes() {
- return Arrays.asList( stringFunctionMap,intFunctionMap, booleanFunctionMap).stream()
- .flatMap( map->map.keySet().stream() ).collect( Collectors.toList());
- }
-
- }
|