summaryrefslogtreecommitdiff
path: root/src/main/java/com/redstoner/misc/mysql/elements/MysqlTable.java
blob: e6a7617cebda0fdfb0706b4348cb4e5872258e05 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
package com.redstoner.misc.mysql.elements;

import com.redstoner.misc.mysql.MysqlQueryHandler;

import java.sql.DatabaseMetaData;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

public class MysqlTable {
	private MysqlDatabase database;
	private String        name;

	public MysqlTable(MysqlDatabase database, String name) {
		this.database = database;
		this.name = name;
	}

	public String getName() {
		return this.name;
	}

	public boolean insert(String... values) {
		MysqlField[] description = describe();
		if (values.length > 0 && values.length == description.length) {
			String val = "(\"" + String.join("\",\"", values) + "\")";
			return MysqlQueryHandler.queryNoResult(
					database.getConnection(),
					"INSERT INTO `" + name + "` VALUES " + val + ";"
			);
		} else {
			return false;
		}
	}

	public MysqlField[] describe() {
		try {
			List<MysqlField> description  = new ArrayList<>();
			DatabaseMetaData metadata     = database.getConnection().getMetaData();
			ResultSet        queryResults = metadata.getColumns(null, null, name, null);
			while (queryResults.next()) {
				description.add(new MysqlField(
						queryResults.getString(4),
						queryResults.getString(6).split(" ")[0] + "(" + queryResults.getString(7) + ")",
						queryResults.getBoolean(11)
				));
			}
			return description.toArray(new MysqlField[0]);
		} catch (SQLException e) {
			e.printStackTrace();
			return null;
		}
	}

	public Object[] get(String fieldName, MysqlConstraint... constraints) {
		ResultSet results = MysqlQueryHandler.queryResult(
				database.getConnection(),
				"SELECT " + fieldName + " FROM `" + name + "`" + getConstraints(constraints) + ";"
		);
		List<Object> resObj = new ArrayList<>();
		try {
			while (results.next()) {
				resObj.add(results.getObject(1));
			}
		} catch (SQLException e) {
			e.printStackTrace();
			return new Object[0];
		}
		return resObj.toArray(new Object[0]);
	}

	private String getConstraints(MysqlConstraint... constraints) {
		String cons = "";
		if (constraints.length > 0) {
			cons += " WHERE ";
			for (int i = 0; i < constraints.length; i++) {
				MysqlConstraint constraint = constraints[i];
				cons += constraint.getFieldName() + constraint.getOperator().toString() + "\"" + constraint.getValue()
				        + "\"";
				if (i < constraints.length - 1) {
					cons += " AND ";
				}
			}
		}
		return cons;
	}

	public Object[] get(String statement) {
		ResultSet    results = MysqlQueryHandler.queryResult(database.getConnection(), statement);
		List<Object> resObj  = new ArrayList<>();
		try {
			while (results.next()) {
				resObj.add(results.getObject(1));
			}
		} catch (SQLException e) {
			e.printStackTrace();
			return new Object[0];
		}
		return resObj.toArray(new Object[0]);
	}

	public boolean delete(MysqlConstraint... constraints) {
		return MysqlQueryHandler.queryNoResult(
				database.getConnection(),
				"DELETE FROM `" + name + "`" + getConstraints(constraints) + ";"
		);
	}

	public boolean drop() {
		return MysqlQueryHandler.queryNoResult(database.getConnection(), "DROP TABLE `" + name + "`;");
	}
}