1 /* 2 * Licensed to the Apache Software Foundation (ASF) under one 3 * or more contributor license agreements. See the NOTICE file 4 * distributed with this work for additional information 5 * regarding copyright ownership. The ASF licenses this file 6 * to you under the Apache License, Version 2.0 (the 7 * "License"); you may not use this file except in compliance 8 * with the License. You may obtain a copy of the License at 9 * 10 * http://www.apache.org/licenses/LICENSE-2.0 11 * 12 * Unless required by applicable law or agreed to in writing, software 13 * distributed under the License is distributed on an "AS IS" BASIS, 14 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 15 * See the License for the specific language governing permissions and 16 * limitations under the License. 17 */ 18 package org.apache.giraph.block_app.framework.piece.global_comm; 19 20 import org.apache.giraph.block_app.framework.api.BlockMasterApi; 21 import org.apache.giraph.master.MasterGlobalCommUsage; 22 import org.apache.giraph.worker.WorkerBroadcastUsage; 23 24 /** 25 * Handle that wraps both reducerHandle and broadcastHandle, so callers 26 * don't need to have two fields. 27 * 28 * @param <S> Single value type 29 * @param <R> Reduced value type 30 */ 31 public class ReducerAndBroadcastWrapperHandle<S, R> { 32 private ReducerHandle<S, R> reducerHandle; 33 private BroadcastHandle<R> broadcastHandle; 34 35 /** Set reducer handle to just registered handle */ 36 public void registeredReducer(ReducerHandle<S, R> reducerHandle) { 37 this.reducerHandle = reducerHandle; 38 } 39 40 /** Reduce single value */ 41 public void reduce(S valueToReduce) { 42 reducerHandle.reduce(valueToReduce); 43 } 44 45 /** Get reduced value */ 46 public R getReducedValue(MasterGlobalCommUsage master) { 47 return reducerHandle.getReducedValue(master); 48 } 49 50 /** 51 * Broadcast reduced value from master 52 */ 53 public void broadcastValue(BlockMasterApi master) { 54 broadcastHandle = reducerHandle.broadcastValue(master); 55 } 56 57 /** Get broadcasted value */ 58 public R getBroadcast(WorkerBroadcastUsage worker) { 59 return broadcastHandle.getBroadcast(worker); 60 } 61 }