avrilfanomar / flexiblesearchbuilder Goto Github PK

WHERE TRIM({object:field}) = TRIM(?parameter1)

package org.bitbucket.andriichukandrii.hybris.flexiblesearchbuilder;

import org.spockframework.util.Assert;

public class TrimmedField implements Field {

    private final Field decorated;

    public TrimmedField(Field decorated) {
        Assert.notNull(decorated, "The parameter decorated can not be null.");
        this.decorated = decorated;
    }

    @Override
    public String getFieldName() {
        return decorated.getFieldName();
    }

    @Override
    public String toString() {
        return "TRIM(" + decorated.toString() + ")";
    }
}

package org.bitbucket.andriichukandrii.hybris.flexiblesearchbuilder;

import java.util.Collection;
import java.util.Map;

public class TrimmedParameterFieldCondition extends AbstractFieldCondition {

    private final ParameterConditionType conditionType;
    private final Object conditionParameter;
    private String parameterCode;

    public TrimmedParameterFieldCondition(Field field, ParameterConditionType conditionType, Object conditionParameter) {
        super(field);
        this.conditionType = conditionType;
        this.conditionParameter = conditionParameter;
    }

    protected void appendQuery(StringBuilder sb) {
        super.appendQuery(sb);
        boolean collectionParameter = this.conditionParameter instanceof Collection;
        sb.append(this.conditionType.getOperator());
        if (collectionParameter) {
            sb.append("(");
        }

        sb.append("TRIM(?").append(this.parameterCode).append(")");
        if (collectionParameter) {
            sb.append(")");
        }
    }

    protected void addParameters(Map<String, Object> parameterMap) {
        super.addParameters(parameterMap);
        this.parameterCode = FlexibleSearchBuilderFieldUtils.createUniqueParameterCode(parameterMap, this.field.getFieldName());
        parameterMap.put(this.parameterCode, this.conditionParameter);
    }
}