| /* |
| * |
| * * Licensed to the Apache Software Foundation (ASF) under one or more |
| * * contributor license agreements. 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. For additional information regarding |
| * * copyright in this work, please see the NOTICE file in the top level |
| * * directory of this distribution. |
| * |
| */ |
| package org.apache.usergrid.persistence.graph.serialization.impl; |
| |
| |
| import com.google.inject.Inject; |
| import org.apache.usergrid.persistence.graph.GraphManager; |
| import org.apache.usergrid.persistence.graph.serialization.EdgesObservable; |
| import org.apache.usergrid.persistence.graph.serialization.TargetIdObservable; |
| import org.apache.usergrid.persistence.model.entity.Id; |
| import org.slf4j.Logger; |
| import org.slf4j.LoggerFactory; |
| import rx.Observable; |
| |
| /** |
| * Emits the id of all nodes that are target nodes from the given source node |
| */ |
| public class TargetIdObservableImpl implements TargetIdObservable { |
| |
| private static final Logger logger = LoggerFactory.getLogger(TargetIdObservableImpl.class); |
| private final EdgesObservable edgesFromSourceObservable; |
| |
| @Inject |
| public TargetIdObservableImpl(final EdgesObservable edgesFromSourceObservable){ |
| this.edgesFromSourceObservable = edgesFromSourceObservable; |
| } |
| |
| /** |
| * Get all nodes that are target nodes from the sourceNode |
| * @param gm |
| * @param sourceNode |
| * |
| * @return |
| */ |
| @Override |
| public Observable<Id> getTargetNodes(final GraphManager gm, final Id sourceNode) { |
| |
| //only search edge types that start with collections |
| return edgesFromSourceObservable.edgesFromSourceDescending( gm, sourceNode, true).map(edge -> { |
| final Id targetNode = edge.getTargetNode(); |
| |
| if (logger.isDebugEnabled()) { |
| logger.debug("Emitting targetId of {}", edge); |
| } |
| |
| |
| return targetNode; |
| } ); |
| } |
| } |