RegimenRepository.java

/*
 * This program is part of the OpenLMIS logistics management information
 * system platform software.
 *
 * Copyright © 2015 ThoughtWorks, Inc.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published
 * by the Free Software Foundation, either version 3 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 Affero General Public License for more details. You should
 * have received a copy of the GNU Affero General Public License along with
 * this program. If not, see http://www.gnu.org/licenses. For additional
 * information contact info@OpenLMIS.org
 */

package org.openlmis.core.model.repository;


import android.content.Context;

import com.google.inject.Inject;
import com.j256.ormlite.dao.Dao;

import org.openlmis.core.exceptions.LMISException;
import org.openlmis.core.model.RegimeShortCode;
import org.openlmis.core.model.Regimen;
import org.openlmis.core.persistence.DbUtil;
import org.openlmis.core.persistence.GenericDao;

import java.sql.SQLException;
import java.util.List;

public class RegimenRepository {

    GenericDao<Regimen> regimenGenericDao;

    @Inject
    DbUtil dbUtil;

    @Inject
    public RegimenRepository(Context context) {
        this.regimenGenericDao = new GenericDao<>(Regimen.class, context);
    }

    public Regimen getByNameAndCategory(final String name, final Regimen.RegimeType category) throws LMISException {
        return dbUtil.withDao(Regimen.class, new DbUtil.Operation<Regimen, Regimen>() {
            @Override
            public Regimen operate(Dao<Regimen, String> dao) throws SQLException {
                return dao.queryBuilder().where().eq("name", name).and().eq("type", category).queryForFirst();
            }
        });
    }

    public void create(Regimen regimen) throws LMISException {
        regimenGenericDao.create(regimen);
    }

    public List<Regimen> listDefaultRegime() throws LMISException {
        return dbUtil.withDao(Regimen.class, new DbUtil.Operation<Regimen, List<Regimen>>() {
            @Override
            public List<Regimen> operate(Dao<Regimen, String> dao) throws SQLException {
                return dao.queryBuilder()
                        .where()
                        .eq("isCustom", false)
                        .query();
            }
        });
    }

    public List<RegimeShortCode> listRegimeShortCode() throws LMISException {
        return dbUtil.withDao(RegimeShortCode.class, new DbUtil.Operation<RegimeShortCode, List<RegimeShortCode>>() {
            @Override
            public List<RegimeShortCode> operate(Dao<RegimeShortCode, String> dao) throws SQLException {
                return dao.queryBuilder().query();
            }
        });
    }
}