DatabaseTypeTable.java

/*
 * Copyright 2016 nerzid.
 *
 * 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 com.nerzid.autocomment.database;

import static com.nerzid.autocomment.database.DataTypeModel.COLUMN_IDENTIFIER;
import static com.nerzid.autocomment.database.DataTypeModel.COLUMN_LEMMA;
import static com.nerzid.autocomment.database.DataTypeModel.COLUMN_POSTAG;
import static com.nerzid.autocomment.database.DataTypeModel.COLUMN_SIMPLIFIED_IDENTIFIER;
import java.util.ArrayList;
import java.util.List;

/**
 *
 * @author nerzid
 */
public class DataTypeTable {

    private int dtid;
    private String identifier;
    private String simplifiedIdentifier;
    private String lemma;
    private String postag;

    public DataTypeTable() {
        identifier = "";
        simplifiedIdentifier = "";
        lemma = "";
        postag = "";
    }

    public DataTypeTable(String identifier, String simplifiedIdentifier, String lemma, String postag) {
        this.identifier = identifier;
        this.simplifiedIdentifier = simplifiedIdentifier;
        this.lemma = lemma;
        this.postag = postag;
    }

    public static DataTypeModel insertOrGet(DataTypeTable dt) {
        DataTypeModel data_type = null;
        try {
            data_type = DataTypeModel.findFirst(
                    DataTypeModel.COLUMN_IDENTIFIER + " = ? AND "
                    + DataTypeModel.COLUMN_SIMPLIFIED_IDENTIFIER + " = ? AND "
                    + DataTypeModel.COLUMN_LEMMA + " = ? AND "
                    + DataTypeModel.COLUMN_POSTAG + " = ?",
                    dt.getIdentifier(),
                    dt.getSimplifiedIdentifier(),
                    dt.getLemma(),
                    dt.getPostag());
            if (data_type == null) {
                data_type = new DataTypeModel().set(
                        COLUMN_IDENTIFIER, dt.getIdentifier(),
                        COLUMN_SIMPLIFIED_IDENTIFIER, dt.getSimplifiedIdentifier(),
                        COLUMN_LEMMA, dt.getLemma(),
                        COLUMN_POSTAG, dt.getPostag());
                if (data_type.saveIt()) {
                    return data_type;
                } else {
                    return null;
                }
            } else {
                return DataTypeModel.getDataTypeModelUsingIdentifier(dt.getIdentifier());
            }
        } catch (NullPointerException e) {
            if (data_type == null) {
                data_type = new DataTypeModel().set(
                        COLUMN_IDENTIFIER, dt.getIdentifier(),
                        COLUMN_SIMPLIFIED_IDENTIFIER, dt.getSimplifiedIdentifier(),
                        COLUMN_LEMMA, dt.getLemma(),
                        COLUMN_POSTAG, dt.getPostag());
                if (data_type.saveIt()) {
                    return data_type;
                } else {
                    return null;
                }
            } else {
                return DataTypeModel.getDataTypeModelUsingIdentifier(dt.getIdentifier());
            }
        }

    }

    /**
     * Gets all data type's identifiers
     *
     * @param param_models
     * @return List of identifiers as string
     */
    public static List<String> getAll(List<ParameterModel> param_models) {
        List<String> dtm = new ArrayList<>();

        for (ParameterModel pm : param_models) {

            int dtid = pm.getInteger(ParameterModel.COLUMN_FK_DTID);

            String data_type = (String) DataTypeModel
                    .findFirst(DataTypeModel.COLUMN_DTID + " = ?", dtid)
                    .get(DataTypeModel.COLUMN_IDENTIFIER);
            if (data_type != null) {
                dtm.add(data_type);
            }

        }
        return dtm;
    }

    public String addSimplifiedIdentifier(String si) {
        return simplifiedIdentifier += si + " ";
    }

    public String addLastSimplifiedIdentifier(String si) {
        return simplifiedIdentifier += si;
    }

    public String addLemma(String l) {
        return lemma += l + " ";
    }

    public String addLastLemma(String l) {
        return lemma += l;
    }

    public String addPostag(String p) {
        return postag += p + " ";
    }

    public String addLastPostag(String p) {
        return postag += p;
    }

    public int getDtid() {
        return dtid;
    }

    public void setDtid(int dtid) {
        this.dtid = dtid;
    }

    public String getIdentifier() {
        return identifier;
    }

    public void setIdentifier(String identifier) {
        this.identifier = identifier;
    }

    public String getSimplifiedIdentifier() {
        return simplifiedIdentifier;
    }

    public void setSimplifiedIdentifier(String simplifiedIdentifier) {
        this.simplifiedIdentifier = simplifiedIdentifier;
    }

    public String getLemma() {
        return lemma;
    }

    public void setLemma(String lemma) {
        this.lemma = lemma;
    }

    public String getPostag() {
        return postag;
    }

    public void setPostag(String postag) {
        this.postag = postag;
    }
}