D-Net

package eu.dnetlib.goldoa.service.dao;

import eu.dnetlib.goldoa.domain.File;

import eu.dnetlib.goldoa.domain.Invoice;

import org.apache.commons.io.IOUtils;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.hibernate.criterion.Restrictions;

import org.hibernate.type.StringType;

import org.springframework.stereotype.Repository;

import java.io.ByteArrayOutputStream;

import java.io.IOException;

import java.io.InputStream;

import java.util.List;

/**

 * Created by panagiotis on 28/1/2017.

 */

@Repository

public class InvoiceDAO extends AbstractDao<String, Invoice>{

    private Log logger = LogFactory.getLog(InvoiceDAO.class);

    public Invoice saveInvoice(Invoice invoice) {

        Invoice i = getByKey(invoice.getId());

        if(i!=null) {

            if(i.getFile()!=null)

                invoice.setFile(i.getFile());

            Invoice persistened_invoice = (Invoice) getSession().merge(invoice);

        }else

            persist(invoice);

        return invoice;

    }

    @SuppressWarnings("unchecked")

    public Invoice getByKey(String invoiceId){

        List<Invoice> inv =  createEntityCriteria()

                .add(Restrictions.eq("id",invoiceId)).list();

        if(inv.size() > 0)

            return inv.get(0);

        return null;

    }

    public Invoice uploadInvoice(String invoiceId, String mimetype, InputStream inputStream) throws IOException {

        Invoice inv = getByKey(invoiceId);

        ByteArrayOutputStream baos = null;

        baos = new ByteArrayOutputStream();

        IOUtils.copy(inputStream, baos);

        IOUtils.closeQuietly(baos);

        File file = new File();

        file.setId(getFileId());

        file.setMimetype(mimetype);

        file.setFile(baos.toByteArray());

        inv.setFile(file);

        return inv;

    }

    private String getFileId(){

        return (String) getSession().createSQLQuery("select nextval('file_id_seq') as id").addScalar("id",StringType.INSTANCE)

                .list().get(0);

    }

}