io.leopard.data.dfs.DfsMysqlImpl Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of leopard-dfs Show documentation

分布式文件系统实现，目前只是做了基于GridFS的实现，可以在本机文件系统进行缓存。

There is a newer version: 0.9.12

package io.leopard.data.dfs;

import io.leopard.jdbc.Jdbc;
import io.leopard.jdbc.builder.ReplaceBuilder;

import java.io.FileNotFoundException;
import java.io.IOException;
import java.sql.ResultSet;
import java.sql.SQLException;

import org.springframework.beans.factory.DisposableBean;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.dao.DataAccessException;
import org.springframework.jdbc.core.ResultSetExtractor;

public class DfsMysqlImpl implements Dfs, InitializingBean, DisposableBean {

	private static final String TABLE = "file";

	private Jdbc jdbc;

	public void setJdbc(Jdbc jdbc) {
		this.jdbc = jdbc;
	}

	@Override
	public boolean create(String filename, byte[] data) {
		ReplaceBuilder builder = new ReplaceBuilder(TABLE);
		builder.setString("filename", filename);
		builder.setInt("size", data.length);
		builder.setBytes("data", data);
		return this.jdbc.insertForBoolean(builder);
	}

	@Override
	public byte[] read(String filename) throws IOException {
		String sql = "select data from " + TABLE + " where filename='" + filename + "';";
		byte[] data = this.jdbc.getJdbcTemplate().query(sql, new ResultSetExtractor() {
			@Override
			public byte[] extractData(ResultSet rs) throws SQLException, DataAccessException {
				if (rs.first()) {
					return rs.getBytes(1);
				}
				else {
					return null;
				}
			}
		});
		if (data == null) {
			throw new FileNotFoundException(filename);
		}
		return data;
	}

	@Override
	public boolean delete(String filename) {
		String sql = "delete from " + TABLE + " where filename=?;";
		return this.jdbc.updateForBoolean(sql, filename);
	}

	@Override
	public void destroy() throws Exception {

	}

	@Override
	public void afterPropertiesSet() throws Exception {

	}

}