com.clicktravel.cheddar.infrastructure.persistence.filestore.FileItem.java Source code

Introduction

Here is the source code for com.clicktravel.cheddar.infrastructure.persistence.filestore.FileItem.java

Source

/*
 * Copyright 2014 Click Travel Ltd
 *
 * 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.clicktravel.cheddar.infrastructure.persistence.filestore;

import java.io.*;
import java.nio.charset.StandardCharsets;
import java.util.Arrays;
import java.util.UUID;

import org.joda.time.DateTime;

public class FileItem {

    private final String filename;

    private final DateTime lastUpdatedTime;

    private final byte[] contents;

    /**
     * Create a file item from any given string, assumes UTF-encoding
     *
     * @param filename the name of the file you want to create
     * @param contents the string in UTF-8 encoding
     */
    public FileItem(final String filename, final String contents) {
        this.filename = filename;
        this.contents = contents.getBytes(StandardCharsets.UTF_8);
        lastUpdatedTime = DateTime.now();
    }

    public FileItem(final String filename, final InputStream inputStream, final DateTime lastUpdatedTime)
            throws IOException {
        if (inputStream == null) {
            throw new IllegalArgumentException("Input stream cannot be null");
        }
        this.filename = filename;
        if (lastUpdatedTime == null) {
            this.lastUpdatedTime = DateTime.now();
        } else {
            this.lastUpdatedTime = lastUpdatedTime;
        }
        final ByteArrayOutputStream bout = new ByteArrayOutputStream();
        int read = -1;
        final byte[] buf = new byte[1024 * 50];
        try {
            while ((read = inputStream.read(buf)) != -1) {
                bout.write(buf, 0, read);
            }
            contents = bout.toByteArray();
        } catch (final IOException e) {
            throw new IllegalStateException(e);
        }
    }

    public FileItem(final String contents) {
        this(UUID.randomUUID().toString(), contents);
    }

    public FileItem(final String filename, final File file) throws IOException {
        this(filename, new FileInputStream(file), DateTime.now());
    }

    public String filename() {
        return filename;
    }

    public DateTime lastUpdatedTime() {
        return lastUpdatedTime;
    }

    public byte[] getBytes() {
        return contents;
    }

    /**
     * @return the contents of the bytes[] as a string, assumes the bytes are UTF-8 encoded
     */
    public String getContentsAsString() {
        return new String(getBytes(), StandardCharsets.UTF_8);
    }

    @Override
    public int hashCode() {
        final int prime = 31;
        int result = 1;
        result = prime * result + Arrays.hashCode(contents);
        result = prime * result + ((filename == null) ? 0 : filename.hashCode());
        result = prime * result + ((lastUpdatedTime == null) ? 0 : lastUpdatedTime.hashCode());
        return result;
    }

    @Override
    public boolean equals(final Object obj) {
        if (this == obj) {
            return true;
        }
        if (obj == null) {
            return false;
        }
        if (getClass() != obj.getClass()) {
            return false;
        }
        final FileItem other = (FileItem) obj;
        if (!Arrays.equals(contents, other.contents)) {
            return false;
        }
        if (filename == null) {
            if (other.filename != null) {
                return false;
            }
        } else if (!filename.equals(other.filename)) {
            return false;
        }
        if (lastUpdatedTime == null) {
            if (other.lastUpdatedTime != null) {
                return false;
            }
        } else if (!lastUpdatedTime.equals(other.lastUpdatedTime)) {
            return false;
        }
        return true;
    }

}