Repository navigation
Expand file tree
/
Copy pathQueryColumn.java
More file actions
64 lines (50 loc) · 1.63 KB
/
Copy pathQueryColumn.java
File metadata and controls
64 lines (50 loc) · 1.63 KB
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
package org.javawebstack.orm.query;
import org.javawebstack.orm.TableInfo;
import org.javawebstack.orm.exception.ORMQueryException;
import java.util.Arrays;
import java.util.Objects;
import java.util.regex.Pattern;
import java.util.stream.Collectors;
public class QueryColumn {
private static final Pattern NAME_PATTERN = Pattern.compile("[A-Za-z0-9_\\-.]+");
private final String name;
private final boolean raw;
public QueryColumn(String name) {
this(name, false);
}
public QueryColumn(String name, boolean raw) {
if(!raw)
validateName(name);
this.name = name;
this.raw = raw;
}
public String getName() {
return name;
}
public boolean isRaw() {
return raw;
}
public String toString() {
return toString(null);
}
public String toString(TableInfo info) {
if (raw)
return name;
return Arrays.stream((info != null ? info.getColumnName(name) : name).split("\\.")).map(s -> "`" + s + "`").collect(Collectors.joining("."));
}
private static void validateName(String name) {
if(!NAME_PATTERN.matcher(name).matches())
throw new ORMQueryException("Invalid column name '" + name + "' (Use raw in case you know what you're doing)");
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
QueryColumn that = (QueryColumn) o;
return toString().equals(that.toString());
}
@Override
public int hashCode() {
return Objects.hash(toString());
}
}