com.healthmarketscience.sqlbuilder.Condition Java Examples

protected void filterLayer(LevelMemberSet layer) throws OlapException {
	OlapOp func =layer.getFunction();
	if(func!=null){
		SetSubquery sub_query = func.query(mapping, layer);
		if(sub_query!=null){
			
			Condition condition = sub_query.condition();
			if(condition!=null)
				addCondition(condition);

			Condition having_condition = sub_query.havingCondition();
			if(having_condition!=null)
				addHavingCondition(having_condition);
			
		}
			

	}
}
 

@Test
public void testNoConditionsAddedIfNotPartitioning() throws Exception {
    PowerMockito.mockStatic(DriverManager.class);
    SelectQuery selectQuery = PowerMockito.mock(SelectQuery.class);
    ComboCondition comboCondition = PowerMockito.mock(ComboCondition.class);

    when(selectQuery.getWhereClause()).thenReturn(comboCondition);
    when(comboCondition.addCondition(any(Condition.class))).thenReturn(comboCondition);
    when(config.getDriverClass()).thenReturn(JDBC_CELL_EXTRACTOR_CLASSNAME_CONSTANT);
    when(config.getConnectionUrl()).thenReturn(WHATEVER_CONSTANT);
    when(config.getUsername()).thenReturn(WHATEVER_CONSTANT);
    when(config.getPassword()).thenReturn(WHATEVER_CONSTANT);
    when(config.getQuery()).thenReturn(selectQuery);
    when(DriverManager.getConnection(anyString(), anyString(), anyString())).thenReturn(conn);
    when(conn.createStatement()).thenReturn(statement);
    when(statement.executeQuery(anyString())).thenReturn(resultSet);
    when(resultSet.next()).thenReturn(true);

    JdbcReader reader = new JdbcReader(config);

    reader.init(partition);

    verify(comboCondition, times(0)).addCondition(any((BinaryCondition.class)));

}
 

@Test
public void testConditionIsAddedForPartitioning() throws Exception {

    PowerMockito.mockStatic(DriverManager.class);
    SelectQuery selectQuery = PowerMockito.mock(SelectQuery.class);
    ComboCondition comboCondition = PowerMockito.mock(ComboCondition.class);

    when(selectQuery.getWhereClause()).thenReturn(comboCondition);
    when(comboCondition.addCondition(any(Condition.class))).thenReturn(comboCondition);
    when(config.getDriverClass()).thenReturn(JDBC_CELL_EXTRACTOR_CLASSNAME_CONSTANT);
    when(config.getConnectionUrl()).thenReturn(WHATEVER_CONSTANT);
    when(config.getUsername()).thenReturn(WHATEVER_CONSTANT);
    when(config.getPassword()).thenReturn(WHATEVER_CONSTANT);
    when(config.getPartitionKey()).thenReturn(PowerMockito.mock(DbColumn.class));
    when(config.getNumPartitions()).thenReturn(NUM_PARTITIONS);
    when(config.getQuery()).thenReturn(selectQuery);
    when(DriverManager.getConnection(anyString(), anyString(), anyString())).thenReturn(conn);
    when(partition.lower()).thenReturn(0L);
    when(partition.upper()).thenReturn(100L);
    when(conn.createStatement()).thenReturn(statement);
    when(statement.executeQuery(anyString())).thenReturn(resultSet);
    when(resultSet.next()).thenReturn(true);

    JdbcReader reader = new JdbcReader(config);

    reader.init(partition);

    verify(comboCondition, times(2)).addCondition(any((BinaryCondition.class)));
}
 

public Condition condition() {
	
	if((values==null || values.size()==0) && from_value==null && to_value==null){
		if(except==null)
			return null;
		else
			return new NotCondition(except.condition());
	}
		
	DimensionTable dim_table = join.getDimensionTable();
	
	SelectQuery dim_query = new SelectQuery().
			addFromTable(dim_table.getDbTable()).
			addColumns(dim_table.getDbKey());
	
	if(from_value!=null){
		dim_query = dim_query.addCondition(BinaryCondition.greaterThan(column.getDbColumn(), from_value, true));
	}

	if(to_value!=null){
		dim_query = dim_query.addCondition(BinaryCondition.lessThan(column.getDbColumn(), to_value, true));
	}

	
	if(values!=null && values.size()>0){
		if(values.size()==1){
			String value = values.iterator().next();
			if(LevelMember.NULL_MEMBER.equals(value))
				dim_query = dim_query.addCondition(UnaryCondition.isNull(column.getDbColumn()));
			else		
				dim_query = dim_query.addCondition(BinaryCondition.equalTo(column.getDbColumn(), value));
		}else{ 
			
			if(values.contains(LevelMember.NULL_MEMBER)){
				Set<String> values_without_null = new HashSet<String>(values);
				values_without_null.remove(LevelMember.NULL_MEMBER);
				dim_query = dim_query.addCondition(ComboCondition.or(
						UnaryCondition.isNull(column.getDbColumn()),
						new InCondition(column.getDbColumn(), values_without_null)));
			}else{
				dim_query = dim_query.addCondition(new InCondition(column.getDbColumn(), values));	
			}
			
		}
	}
	
	Condition myCondition = new InCondition(join.getForeign_key(), new Subquery(dim_query) );
	
	if(except!=null)
		myCondition = ComboCondition.and(myCondition, new NotCondition(except.condition()) );

	if(exists!=null)
		myCondition = ComboCondition.and(myCondition, exists.condition());
	
	return myCondition;
	 
	
}
 

public Condition havingCondition(){
	
	if(having_expression==null){
		if(except == null || except.having_expression==null)
			return null;
		else
			return new NotCondition(except.havingCondition());
	}
	
	Condition myCondition = new CustomCondition(having_expression);
	
	if(except!=null && except.having_expression!=null)
		myCondition = ComboCondition.and(myCondition, new NotCondition(except.havingCondition()) );
	
	if(exists!=null && exists.having_expression!=null)
		myCondition = ComboCondition.and(myCondition, exists.havingCondition() );
	
	
	return myCondition;
	
	
}
 

private void addCondition(Condition condition){
	conditions.add(condition);
}
 

private void addHavingCondition(Condition having_condition) {
	having_conditions.add(having_condition);
}
 

public DbCheckConstraint(DbColumn parent, String name, Condition condition) {
  super(parent, name, Type.CHECK);
  _condition = condition;
}
 

public DbCheckConstraint(DbTable parent, String name, Condition condition) {
  super(parent, name, Type.CHECK, (DbColumn[])null);
  _condition = condition;
}
 

@Override
public Condition getCondition() {
  return _condition;
}
 

/**
 * @return the check condition for this constraint
 */
public Condition getCondition();
 

/**
 * Creates and adds check constraint with the given parameters to this
 * table.
 * <p>
 * Note, no effort is made to make sure the given name is unique.
 * @param condition the check condition
 */
public DbCheckConstraint checkCondition(String name, Condition condition) {
  DbCheckConstraint constraint = getSpec().createColumnCheckConstraint(
      this, name, condition);
  return addConstraint(constraint);
}
 

/**
 * Creates and adds check constraint with the given parameters to this
 * table.
 * <p>
 * Note, no effort is made to make sure the given name is unique.
 * @param condition the check condition
 */
public DbCheckConstraint checkCondition(String name, Condition condition) {
  DbCheckConstraint constraint = getSpec().createTableCheckConstraint(
      this, name, condition);
  return addConstraint(constraint);
}
 

/**
 * Creates and returns a new column {@link DbCheckConstraint} with the given
 * parameters.
 * <p>
 * This method can be overriden to utilize custom model subclasses.
 */
public DbCheckConstraint createColumnCheckConstraint(
    DbColumn parent, String name, Condition condition)
{
  return new DbCheckConstraint(parent, name, condition);
}
 

/**
 * Creates and returns a new table {@link DbCheckConstraint} with the given
 * parameters.
 * <p>
 * This method can be overriden to utilize custom model subclasses.
 */
public DbCheckConstraint createTableCheckConstraint(
    DbTable parent, String name, Condition condition)
{
  return new DbCheckConstraint(parent, name, condition);
}