Application: gvSIG desktop » gvSIG data provider for PostgreSQL

package org.gvsig.postgresql.dal.expressionbuilderformatter;

import java.text.MessageFormat;

import java.util.List;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.expressionevaluator.ExpressionBuilder;

import static org.gvsig.expressionevaluator.ExpressionBuilder.FUNCTION_IFNULL;

import org.gvsig.expressionevaluator.ExpressionBuilder.Function;

import org.gvsig.expressionevaluator.ExpressionBuilder.Value;

import org.gvsig.expressionevaluator.Formatter;

import org.gvsig.fmap.dal.SQLBuilder;

import org.gvsig.tools.dataTypes.DataTypes;

/**

 *

 * @author jjdelcerro

 */

class JsonValue implements Formatter<Value> {

    private final Formatter<Value> formatter;

    private final SQLBuilder builder;

    public JsonValue(SQLBuilder builder, Formatter<Value> formatter) {

        this.builder = builder;

        this.formatter = formatter;

    }

    @Override

    public boolean canApply(Value value) {

        if (value instanceof Function) {

            if (value instanceof ExpressionBuilder.Function) {

                boolean canApply = StringUtils.equalsIgnoreCase(ExpressionBuilder.FUNCTION_JSON_VALUE, ((Function) value).name());

                return canApply;

            }

        }

        return false;

    }

    @Override

    public String format(Value function) {

        List<Value> parameters = ((Function) function).parameters();

        String p1 = parameters.get(0).toString(formatter);

        String p2 = parameters.get(1).toString(formatter);

        String p3 = null;

        if(parameters.size()==3) {

          p3 = parameters.get(2).toString(formatter).replace("'", "");

        }

        String r;

        if (p3 != null) {

            r = MessageFormat.format(

                    "jsonb_path_query_first({0}::jsonb, {1})::{2}",

                    p1,

                    p2,

                    p3

            );

        } else {

            r = MessageFormat.format(

                    "jsonb_path_query_first({0}::jsonb, {1})",

                    p1,

                    p2

            );

        }

        return r;

    }

}