org.lgb.service.FileService.java Source code

Introduction

Here is the source code for org.lgb.service.FileService.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package org.lgb.service;

import java.io.IOException;
import java.io.InputStream;
import java.util.Date;
import java.util.UUID;
import org.apache.log4j.Logger;
import org.hibernate.Session;
import org.hibernate.SessionFactory;
import org.hibernate.Transaction;
import org.lgb.util.Hibernate;
import org.lgb.model.Version;
import org.lgb.model.File;
import org.lgb.model.Content;

/**
 *
 * @author mward
 */
public class FileService {

    protected static Logger logger = Logger.getLogger("service");
    protected static SessionFactory sessionFactory = Hibernate.getSessionFactory();

    public static Version upload(UUID id, InputStream uploadedInputStream) throws IOException {
        Content content = ContentService.store(uploadedInputStream);
        Session session = sessionFactory.getCurrentSession();
        Transaction trans = session.beginTransaction();
        File file = (File) session.load(File.class, id);
        //Only create a new version if the current version content is different
        if (file.getLastestVersion() == null || file.getLastestVersion().getContent().getId() != content.getId()) {
            Version version = new Version(file, content);
            file.setModified(new Date());
            file.setLatestVersion(version);
            session.persist(file);
            session.persist(version);
            trans.commit();
            return version;
        }
        trans.commit();

        return file.getLastestVersion();
    }

    public static File addFile(String name, String path, String type) {
        Session session = sessionFactory.getCurrentSession();
        Transaction trans = session.beginTransaction();
        File file = new File();
        file.setName(name);
        file.setPath(path);
        file.setType(type);
        session.save(file);
        trans.commit();
        return file;
    }

    public static byte[] getContent(UUID id) {
        Session session = sessionFactory.getCurrentSession();
        Transaction trans = session.beginTransaction();
        File file = (File) session.load(File.class, id);
        byte[] bytes = "".getBytes();
        if ((file.getLastestVersion()) != null && file.getLastestVersion().getContent() != null) {
            bytes = file.getLastestVersion().getContent().getFile();
        }
        trans.commit();
        return bytes;
    }

    public static File getFile(UUID id) {
        Session session = sessionFactory.getCurrentSession();
        Transaction trans = session.beginTransaction();
        File file = (File) session.get(File.class, id);
        trans.commit();
        return file;
    }
}