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.edge;
20
21 import org.apache.giraph.conf.GiraphConstants;
22 import org.apache.giraph.conf.ImmutableClassesGiraphConfiguration;
23 import org.apache.hadoop.io.Writable;
24
25 /**
26 * Default implementation of vertex creation decision maker.
27 * By default you can either create all vertices or not create
28 * implicit vertices at all.
29 *
30 * @param <I> Vertex id
31 */
32 public class DefaultCreateSourceVertexCallback<I extends Writable>
33 implements CreateSourceVertexCallback<I> {
34 /**
35 * True if giraph has to create even vertices that only exist
36 * in edge input
37 */
38 private boolean shouldCreateVertices;
39
40 @Override
41 public boolean shouldCreateSourceVertex(I vertexId) {
42 return shouldCreateVertices;
43 }
44
45 @Override
46 public void setConf(ImmutableClassesGiraphConfiguration configuration) {
47 shouldCreateVertices =
48 GiraphConstants.CREATE_EDGE_SOURCE_VERTICES.get(configuration);
49 }
50 }