001/** 002 * Licensed to the Apache Software Foundation (ASF) under one 003 * or more contributor license agreements. See the NOTICE file 004 * distributed with this work for additional information 005 * regarding copyright ownership. The ASF licenses this file 006 * to you under the Apache License, Version 2.0 (the 007 * "License"); you may not use this file except in compliance 008 * with the License. You may obtain a copy of the License at 009 * 010 * http://www.apache.org/licenses/LICENSE-2.0 011 * 012 * Unless required by applicable law or agreed to in writing, software 013 * distributed under the License is distributed on an "AS IS" BASIS, 014 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 015 * See the License for the specific language governing permissions and 016 * limitations under the License. 017 */ 018 019package org.apache.hadoop.mapreduce.lib.db; 020 021import java.sql.ResultSet; 022import java.sql.SQLException; 023import java.util.ArrayList; 024import java.util.List; 025 026import org.apache.hadoop.classification.InterfaceAudience; 027import org.apache.hadoop.classification.InterfaceStability; 028import org.apache.hadoop.conf.Configuration; 029import org.apache.hadoop.mapreduce.InputSplit; 030 031/** 032 * Implement DBSplitter over boolean values. 033 */ 034@InterfaceAudience.Public 035@InterfaceStability.Evolving 036public class BooleanSplitter implements DBSplitter { 037 public List<InputSplit> split(Configuration conf, ResultSet results, String colName) 038 throws SQLException { 039 040 List<InputSplit> splits = new ArrayList<InputSplit>(); 041 042 if (results.getString(1) == null && results.getString(2) == null) { 043 // Range is null to null. Return a null split accordingly. 044 splits.add(new DataDrivenDBInputFormat.DataDrivenDBInputSplit( 045 colName + " IS NULL", colName + " IS NULL")); 046 return splits; 047 } 048 049 boolean minVal = results.getBoolean(1); 050 boolean maxVal = results.getBoolean(2); 051 052 // Use one or two splits. 053 if (!minVal) { 054 splits.add(new DataDrivenDBInputFormat.DataDrivenDBInputSplit( 055 colName + " = FALSE", colName + " = FALSE")); 056 } 057 058 if (maxVal) { 059 splits.add(new DataDrivenDBInputFormat.DataDrivenDBInputSplit( 060 colName + " = TRUE", colName + " = TRUE")); 061 } 062 063 if (results.getString(1) == null || results.getString(2) == null) { 064 // Include a null value. 065 splits.add(new DataDrivenDBInputFormat.DataDrivenDBInputSplit( 066 colName + " IS NULL", colName + " IS NULL")); 067 } 068 069 return splits; 070 } 071}