aswi2020horky-gitlab/project/Deserializer/src/io/Database.java @ 460852db

package io;

import java.sql.Blob;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;

public class Database {
	
	private static class DB_Messenger {
		public String url; // jdbc:mysql://localhost/mydb
		public String user; // root
		public String password; // 1234
		public String query; // SELECT * FROM table
		public String blob; // image
		
		public DB_Messenger(String url, String user, String password, String query, String blob) {
			this.url = url;
			this.user = user;
			this.password = password;
			this.query = query;
			this.blob = blob;
		}
	}
	
	public byte[] doWork(DB_Messenger crate) {
		Connection conn = null;
		Statement stmt = null;
		
		try {
			System.out.println("Connecting to the database...");
			conn = DriverManager.getConnection(crate.url, crate.user, crate.password);
			System.out.println("Connected.");
		} catch (SQLException e) {
			System.out.println("Connection error.");
			e.printStackTrace();
			return null;
		}
		
		System.out.println("Getting the BLOB...");
		try {
			stmt = conn.createStatement();
			ResultSet rs = stmt.executeQuery(crate.query);
			while (rs.next()) {
				Blob blob = rs.getBlob(crate.blob);
				if (blob != null) {
					byte[] bytes = blob.getBytes(1l, (int) blob.length());
					System.out.println("The BLOB was found.");
					return bytes;
				}
			}
		} catch (SQLException e) {
			System.out.println("Error getting the BLOB.");
			e.printStackTrace();
			
		}
		
		System.out.println("The BLOB wasn't found.");
		return null;
	}

}