SQLQuery.java

/*
 *    Constellation - An open source and standard compliant SDI
 *    http://www.constellation-sdi.org
 *
 * Copyright 2014 Geomatys.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.constellation.filter;

import org.apache.lucene.search.Filter;

import java.util.ArrayList;
import java.util.List;

/**
 *
 * @author Guilhem Legal (Geomatys)
 */
public class SQLQuery {
    
    private String query;
    
    public int nbField;
    
    private final Filter spatialFilter;
    
    private List<SQLQuery> subQueries;
    
    public SQLQuery(String query) {
        this.query         = query;
        this.spatialFilter = null;
        nbField            = 1;
    }
    
    public SQLQuery(Filter spatialFilter) {
        this.query         = "";
        this.spatialFilter = spatialFilter;
        nbField            = 0;
    }
    
    public SQLQuery(String query, Filter spatialFilter) {
        this.query         = query;
        this.spatialFilter = spatialFilter;
        nbField            = 0;
    }

    public String getQuery() {
        return query;
    }
    
    public void createSelect() {
        final StringBuilder select = new StringBuilder("SELECT distinct \"identifier\" FROM \"Storage\".\"Records\" ");
        for (int i = 1; i <= nbField; i++) {
            select.append(" , \"Storage\".\"TextValues\" v").append(i);
        }
        select.append(" WHERE ");
        query = select.toString() + query;
    }

    public Filter getSpatialFilter() {
        return spatialFilter;
    }

    public List<SQLQuery> getSubQueries() {
        if (subQueries == null)
            subQueries = new ArrayList<SQLQuery>();
        return subQueries;
    }

    public void setSubQueries(List<SQLQuery> subQueries) {
        this.subQueries = subQueries;
    }
    
    @Override
    public String toString() {
        final StringBuilder s =new StringBuilder("[SQLquery]").append('\n');
        if (query != null && !query.isEmpty())
            s.append("query= ").append(query).append('\n');
        if (spatialFilter != null) {
            s.append("spatialFilter").append(spatialFilter).append('\n');
        }
        if (subQueries != null && !subQueries.isEmpty()) {
            s.append("SubQueries:").append('\n');
            int i = 0;
            for (SQLQuery sq: subQueries) {
                s.append(i).append(": ").append(sq).append('\n'); 
                i++;
            }
        }
        return s.toString();
    }

}