com.lioland.harmony.services.dao.ODBClass.java Source code

Introduction

Here is the source code for com.lioland.harmony.services.dao.ODBClass.java

Source

/*
 * Copyright (C) 2014 Chandima
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 */
package com.lioland.harmony.services.dao;

import com.orientechnologies.orient.core.id.ORecordId;
import com.orientechnologies.orient.core.record.impl.ODocument;
import com.orientechnologies.orient.core.sql.query.OSQLSynchQuery;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.util.ArrayList;
import java.util.List;
import java.util.logging.Level;
import java.util.logging.Logger;
import org.apache.commons.beanutils.PropertyUtils;

/**
 *
 * @author Chandima
 */
public abstract class ODBClass {

    private String rid;
    private final DBFactory factory = DBFactory.getInstance();

    public abstract String getUniqueFieldName();

    public void loadObject() {
        Class cls = this.getClass();
        try {
            String query = "select * from " + cls.getSimpleName() + " where " + getUniqueFieldName() + "='"
                    + PropertyUtils.getProperty(this, getUniqueFieldName()) + "'";
            System.out.println("Query: " + query);
            List<ODocument> docs = DBFactory.getInstance().getDb().query(new OSQLSynchQuery<ODocument>(query));
            if (!docs.isEmpty()) {
                fillObject(docs.get(0), this);
            }
        } catch (IllegalAccessException | InvocationTargetException | NoSuchMethodException
                | ClassNotFoundException ex) {
            Logger.getLogger(ODBClass.class.getName()).log(Level.SEVERE, null, ex);
        }
    }

    public ODBClass(String rid) {
        this.rid = rid;
        ODocument doc = getODocument();
        transform(doc, this.getClass());
    }

    public ODBClass() {
    }

    public void save() {
        ODBClass odbc = this;
        ODocument doc = createODocument(odbc);
        doc.save();
        odbc.setRid(doc.getIdentity().toString());
    }

    private ODocument createODocument(ODBClass odbc) {
        Class cls = odbc.getClass();
        ODocument doc;
        if (rid == null) {
            doc = new ODocument(cls.getSimpleName());
        } else {
            doc = getODocument();
        }
        for (Field field : cls.getDeclaredFields()) {
            try {
                String fieldName = field.getName();
                Object fieldValue = PropertyUtils.getProperty(odbc, fieldName);
                if (fieldValue instanceof ODBClass) {
                    fieldValue = createODocument((ODBClass) fieldValue);
                }
                if (fieldValue != null) {
                    doc.field(fieldName, fieldValue);
                }
            } catch (IllegalArgumentException | IllegalAccessException | InvocationTargetException
                    | NoSuchMethodException ex) {
                Logger.getLogger(ODBClass.class.getName()).log(Level.SEVERE, null, ex);
            }
        }
        return doc;
    }

    public String getRid() {
        return rid;
    }

    public void setRid(String rid) {
        this.rid = rid;
    }

    protected Object querySingle(String query, Class cls) {
        List results = queryList(query, cls);
        return results.get(0);
    }

    protected List queryList(String query, Class cls) {
        List<ODocument> docs = DBFactory.getInstance().getDb().query(new OSQLSynchQuery<ODocument>(query));
        List results = new ArrayList();
        for (ODocument doc : docs) {
            Object ob = transform(doc, cls);
            results.add(ob);
        }
        return results;
    }

    private Object transform(ODocument doc, Class cls) {
        Object ob = null;
        try {
            ob = cls.newInstance();
            fillObject(doc, ob);
        } catch (InstantiationException | IllegalAccessException | InvocationTargetException | NoSuchMethodException
                | ClassNotFoundException ex) {
            Logger.getLogger(ODBClass.class.getName()).log(Level.SEVERE, null, ex);
        }
        return ob;
    }

    private void fillObject(ODocument doc, Object ob) throws NoSuchMethodException, ClassNotFoundException,
            IllegalAccessException, InvocationTargetException {
        for (String fieldName : doc.fieldNames()) {
            Object fieldValue = doc.field(fieldName);
            if (fieldValue instanceof ODocument) {
                fieldValue = transform((ODocument) fieldValue,
                        Class.forName(PACKAGE_PREFIX + ((ODocument) fieldValue).getClassName()));
            }
            if (fieldValue != null) {
                PropertyUtils.setProperty(ob, fieldName, fieldValue);
            }
        }
    }

    public void remove() {
        factory.getDb().delete(new ORecordId(rid));
    }

    private ODocument getODocument() {
        return factory.getDb().getRecord(new ORecordId(rid));
    }

    private static final String PACKAGE_PREFIX = "com.lioland.harmony.dataaccess.";

}