modeling.ocl.sql.SimpleService Maven / Gradle / Ivy

Go to download
/**************************************************************************
Copyright 2019 Vietnamese-German-University

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.

@author: thian
***************************************************************************/

package modeling.ocl.sql;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileReader;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import org.json.simple.JSONArray;
import org.json.simple.parser.JSONParser;
import org.json.simple.parser.ParseException;

import com.vgu.se.jocl.expressions.Expression;
import com.vgu.se.jocl.expressions.Variable;
import com.vgu.se.jocl.parser.simple.SimpleParser;
import com.vgu.se.jocl.types.Type;

import modeling.data.entities.DataModel;
import modeling.ocl.sql.parser.simple.SimpleOclParser;
import modeling.ocl.sql.statement.select.PlainSelect;
import modeling.ocl.sql.statement.select.ResSelectExpression;
import modeling.ocl.sql.statement.select.Select;
import modeling.ocl.sql.statement.select.ValSelectExpression;
import modeling.ocl.sql.utils.SQLAsStringUtils;
import net.sf.jsqlparser.statement.select.SelectItem;

public class SimpleService extends OCL2PSQLService {

    protected DataModel dm;

    protected SimpleParser parser;
    protected SimpleOclParser o2pParser;

    public SimpleService() {
        parser = new SimpleParser();
        o2pParser = new SimpleOclParser();
        setDescriptionMode(false);
    }

    @Override
    protected String mapToString(String oclExp) {
        Select finalStatement = this.mapToSQL(oclExp);
        return m2t(finalStatement);
    }

    protected String m2t(Select finalStatement) {
        if (descriptionMode == true) {
            String finalStatementString = finalStatement
                .toStringWithDescription();

            return SQLAsStringUtils.applyIndent(finalStatementString);

        } else {
            return finalStatement.toString();
        }
    }

    @Override
    protected Select mapToSQL(String oclExp) {

        Expression newExp = parser.parse(oclExp, dm);

        o2pParser.setDataModel(dm);
        newExp.accept(o2pParser);

        return cookFinalStatement(o2pParser.getSelect());
    }

    @Override
    protected void setContextualType(String varName, String varType) {
        parser.putAdhocContextualSet(new Variable(varName, new Type(varType)));
    }

    private Select cookFinalStatement(Select finalStatement) {
        PlainSelect finalPlainSelect = (PlainSelect) finalStatement
            .getSelectBody();

        List newSelectItems = new ArrayList();

        for (SelectItem item : finalPlainSelect.getSelectItems()) {
            if (item instanceof ResSelectExpression
                || item instanceof ValSelectExpression) {
                newSelectItems.add(item);
            }
        }

        finalPlainSelect.getSelectItems().clear();
        finalPlainSelect.getSelectItems().addAll(newSelectItems);

        return finalStatement;
    }

    @Override
    protected void setDataModelFromFilePath(String filePath)
        throws FileNotFoundException, IOException, ParseException, Exception {
        File dataModelFile = new File(filePath);
        DataModel dataModel = new DataModel(new JSONParser()
            .parse(new FileReader(dataModelFile.getAbsolutePath())));
        this.setDataModel(dataModel);
    }

    @Override
    protected void setDataModel(Object dm) {
        this.dm = (DataModel) dm;
    }

    @Override
    protected void setDataModelFromFile(JSONArray json) throws Exception {
        DataModel dataModel = new DataModel(json);
        this.setDataModel(dataModel);
    }

    @Override
    protected String mapToString(Expression oclExp) {
        Select finalStatement = this.mapToSQL(oclExp);

        return m2t(finalStatement);
    }

    @Override
    protected Select mapToSQL(Expression oclExp) {
        o2pParser.setDataModel(dm);
        oclExp.accept(o2pParser);

        return cookFinalStatement(o2pParser.getSelect());
    }

}