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
19 package org.apache.giraph.master.input;
20
21 import java.util.List;
22 import java.util.concurrent.ConcurrentLinkedQueue;
23
24 /**
25 * Input splits organizer for vertex and edge input splits on master, which
26 * doesn't use locality information
27 */
28 public class BasicInputSplitsMasterOrganizer
29 implements InputSplitsMasterOrganizer {
30 /** Available splits queue */
31 private final ConcurrentLinkedQueue<byte[]> splits;
32
33 /**
34 * Constructor
35 *
36 * @param serializedSplits Splits
37 */
38 public BasicInputSplitsMasterOrganizer(List<byte[]> serializedSplits) {
39 splits = new ConcurrentLinkedQueue<>(serializedSplits);
40 }
41
42 @Override
43 public byte[] getSerializedSplitFor(int workerTaskId) {
44 return splits.poll();
45 }
46 }