001/* 002 * (C) Copyright 2017 Nuxeo SA (http://nuxeo.com/) and others. 003 * 004 * Licensed under the Apache License, Version 2.0 (the "License"); 005 * you may not use this file except in compliance with the License. 006 * You may obtain a copy of the License at 007 * 008 * http://www.apache.org/licenses/LICENSE-2.0 009 * 010 * Unless required by applicable law or agreed to in writing, software 011 * distributed under the License is distributed on an "AS IS" BASIS, 012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 013 * See the License for the specific language governing permissions and 014 * limitations under the License. 015 * 016 * Contributors: 017 * Taken from https://github.com/concord/concord-jvm 018 */ 019package org.nuxeo.lib.stream.computation; 020 021import java.util.HashMap; 022import java.util.Map; 023import java.util.Objects; 024import java.util.stream.Collectors; 025 026/** 027 * Extend the metadata to add a mapping for the input and output stream. 028 * 029 * @since 9.3 030 */ 031public class ComputationMetadataMapping extends ComputationMetadata { 032 033 protected final Map<String, String> mapping; 034 035 protected final Map<String, String> reverseMapping; 036 037 public ComputationMetadataMapping(ComputationMetadata metadata, Map<String, String> mapping) { 038 super(mapping.getOrDefault(metadata.name, metadata.name), 039 metadata.inputStreams().stream().map(s -> mapping.getOrDefault(s, s)).collect(Collectors.toSet()), 040 metadata.outputStreams().stream().map(s -> mapping.getOrDefault(s, s)).collect(Collectors.toSet())); 041 this.mapping = mapping; 042 reverseMapping = new HashMap<>(mapping.size()); 043 mapping.forEach((key, value) -> reverseMapping.put(value, key)); 044 } 045 046 public String map(String name) { 047 return mapping.getOrDefault(name, name); 048 } 049 050 public String reverseMap(String name) { 051 return reverseMapping.getOrDefault(name, name); 052 } 053 054 @Override 055 public boolean equals(Object o) { 056 if (this == o) { 057 return true; 058 } 059 if (o == null || getClass() != o.getClass()) { 060 return false; 061 } 062 if (!super.equals(o)) { 063 return false; 064 } 065 ComputationMetadataMapping that = (ComputationMetadataMapping) o; 066 return Objects.equals(mapping, that.mapping) && Objects.equals(reverseMapping, that.reverseMapping); 067 } 068 069 @Override 070 public int hashCode() { 071 return Objects.hash(super.hashCode(), mapping, reverseMapping); 072 } 073}