QUERY_START = new ThreadLocal<>();
18 |
19 | /**
20 | */
21 | public static void start() {
22 | QUERY_START.set(System.currentTimeMillis());
23 | }
24 |
25 | /**
26 | * @param loggerTime
27 | * @param resultSize
28 | */
29 | public static void stop(int loggerTime, int resultSize) {
30 | stop(loggerTime, resultSize, null);
31 | }
32 |
33 | /**
34 | * @param loggerTime
35 | * @param resultSize
36 | * @param sql
37 | */
38 | public static void stop(int loggerTime, int resultSize, String sql) {
39 | if (!LOG.isInfoEnabled() || loggerTime <= 0) {
40 | return;
41 | }
42 |
43 | Long start = QUERY_START.get();
44 | if (start == null) {
45 | start = 0L;
46 | }
47 | QUERY_START.remove();
48 |
49 | int time = start == 0 ? loggerTime : (int) (System.currentTimeMillis() - start);
50 | if (time <= loggerTime && resultSize <= 10000) {
51 | return;
52 | }
53 |
54 | StringBuilder logs = new StringBuilder()
55 | .append("\n** SQL Slower/Many ***********************************************************************")
56 | .append("\nSQL: ").append(sql == null ? "Unknow" : sql);
57 | if (time > loggerTime) {
58 | logs.append("\nTime: ").append(time).append("ms");
59 | }
60 | if (resultSize > 10000) {
61 | logs.append("\nResult(s): ").append(resultSize);
62 | }
63 | LOG.warn(logs);
64 | }
65 | }
66 |
--------------------------------------------------------------------------------
/src/main/java/cn/devezhao/persist4j/query/compiler/CompileException.java:
--------------------------------------------------------------------------------
1 | package cn.devezhao.persist4j.query.compiler;
2 |
3 | import cn.devezhao.persist4j.PersistException;
4 |
5 | /**
6 | *
7 | *
8 | * @author FANGFANG ZHAO
9 | * @since 0.1, Feb 4, 2009
10 | * @version $Id: CompileException.java 8 2015-06-08 09:09:03Z zhaofang123@gmail.com $
11 | */
12 | public class CompileException extends PersistException {
13 | private static final long serialVersionUID = -1344641792787972588L;
14 |
15 | public CompileException(String message) {
16 | super(message);
17 | }
18 |
19 | public CompileException(String message, Throwable ex) {
20 | super(message, ex);
21 | }
22 | }
23 |
--------------------------------------------------------------------------------
/src/main/java/cn/devezhao/persist4j/query/compiler/JoinField.java:
--------------------------------------------------------------------------------
1 | package cn.devezhao.persist4j.query.compiler;
2 |
3 | import cn.devezhao.persist4j.Field;
4 | import cn.devezhao.persist4j.dialect.Dialect;
5 | import cn.devezhao.persist4j.query.compiler.JoinTree.JoinNode;
6 |
7 | import java.util.Objects;
8 |
9 | /**
10 | *
11 | * @author FANGFANG ZHAO
12 | * @since 0.1, 10/11/08
13 | * @version $Id: JoinField.java 120 2016-01-07 11:49:05Z zhaofang123@gmail.com $
14 | */
15 | public class JoinField implements SelectItem {
16 | private static final long serialVersionUID = 6578159656188301525L;
17 |
18 | private final JoinNode tableNode;
19 | private final Field field;
20 | private final String fieldPath;
21 | private final SelectItemType type;
22 |
23 | private String name;
24 | private String alias;
25 | private Integer index;
26 |
27 | private String fullPath = null;
28 |
29 | private String aggregator;
30 | private String aggregatorSibling;
31 | private String aggregatorMode;
32 |
33 | JoinField(JoinNode tableNode, Field field, String fieldPath, SelectItemType type) {
34 | this.tableNode = tableNode;
35 | this.field = field;
36 | this.fieldPath = fieldPath;
37 | this.type = type;
38 | }
39 |
40 | JoinField(JoinField joinField, SelectItemType type) {
41 | this.tableNode = joinField.tableNode;
42 | this.field = joinField.field;
43 | this.fieldPath = joinField.fieldPath;
44 | this.type = type;
45 | }
46 |
47 | // --------------------------------------------------------------------- Implements SelectItem
48 |
49 | @Override
50 | public String getFieldPath() {
51 | return fieldPath;
52 | }
53 |
54 | @Override
55 | public String getName() {
56 | return name;
57 | }
58 |
59 | @Override
60 | public String getAlias() {
61 | return alias;
62 | }
63 |
64 | @Override
65 | public int getIndex() {
66 | return index;
67 | }
68 |
69 | @Override
70 | public Field getField() {
71 | return field;
72 | }
73 |
74 | @Override
75 | public SelectItemType getType() {
76 | return type;
77 | }
78 |
79 |
80 | // ---------------------------------------------------------------------
81 |
82 | protected void setAggregator(String aggregator, String aggregatorSibling) {
83 | this.aggregator = aggregator;
84 | this.aggregatorSibling = aggregatorSibling;
85 | }
86 |
87 | public String getAggregator() {
88 | return aggregator;
89 | }
90 |
91 | public String getAggregatorSibling() {
92 | return aggregatorSibling;
93 | }
94 |
95 | protected void setAggregatorMode(String mode) {
96 | this.aggregatorMode = mode;
97 | }
98 |
99 | public String getAggregatorMode() {
100 | return aggregatorMode;
101 | }
102 |
103 | protected String as(int increase, Dialect dialect) {
104 | if (fullPath != null) {
105 | return fullPath;
106 | }
107 |
108 | index = increase;
109 | if (tableNode == null) {
110 | name = "$";
111 | } else {
112 | name = String.format("%s.%s",
113 | tableNode.getAlias(), dialect.quote(field.getPhysicalName()));
114 | }
115 | alias = JoinTree.COLUMN_ALIAS_PREFIX + index;
116 | return (fullPath = name + " as " + alias);
117 | }
118 |
119 | @Override
120 | public String toString() {
121 | if (getAggregator() != null) {
122 | return String.format("%s%s[%s->%s->%s]",
123 | getAggregator(), (getAggregatorMode() == null ? "" : ("," + getAggregatorMode())),
124 | fieldPath, name, alias);
125 | } else {
126 | return String.format("[%s->%s->%s]", fieldPath, name, alias);
127 | }
128 | }
129 |
130 | @Override
131 | public boolean equals(Object o) {
132 | if (this == o) return true;
133 | if (o == null || getClass() != o.getClass()) return false;
134 | return o.hashCode() == hashCode();
135 | }
136 |
137 | @Override
138 | public int hashCode() {
139 | return Objects.hash(tableNode, field, fieldPath, aggregator, aggregatorMode);
140 | }
141 | }
142 |
--------------------------------------------------------------------------------
/src/main/java/cn/devezhao/persist4j/query/compiler/NestedSelectContext.java:
--------------------------------------------------------------------------------
1 | package cn.devezhao.persist4j.query.compiler;
2 |
3 | import cn.devezhao.persist4j.Entity;
4 | import cn.devezhao.persist4j.query.compiler.JoinTree.JoinNode;
5 |
6 | import java.io.Serializable;
7 |
8 | /**
9 | * @author zhaofang123@gmail.com
10 | * @since 10/27/2018
11 | */
12 | public class NestedSelectContext implements Serializable {
13 | private static final long serialVersionUID = 7585161604688620934L;
14 |
15 | private Entity root;
16 | private JoinNode rootNode;
17 | private int tableIncrease;
18 |
19 | protected NestedSelectContext(Entity root, JoinNode rootNode, int tableIncrease) {
20 | this.root = root;
21 | this.rootNode = rootNode;
22 | this.tableIncrease = tableIncrease;
23 | }
24 |
25 | public Entity getRoot() {
26 | return root;
27 | }
28 |
29 | public JoinNode getRootNode() {
30 | return rootNode;
31 | }
32 |
33 | public int getTableIncrease() {
34 | return tableIncrease;
35 | }
36 | }
37 |
--------------------------------------------------------------------------------
/src/main/java/cn/devezhao/persist4j/query/compiler/ParameterItem.java:
--------------------------------------------------------------------------------
1 | package cn.devezhao.persist4j.query.compiler;
2 |
3 | import cn.devezhao.persist4j.Field;
4 |
5 | import java.io.Serializable;
6 | import java.util.Objects;
7 |
8 | /**
9 | * @author FANGFANG ZHAO
10 | * @since 0.1, Apr 3, 2009
11 | * @version $Id: ParameterItem.java 8 2015-06-08 09:09:03Z zhaofang123@gmail.com $
12 | */
13 | public class ParameterItem implements Serializable {
14 | private static final long serialVersionUID = 6919840306703366608L;
15 |
16 | private String named;
17 | private int index;
18 | private Field field;
19 |
20 | public ParameterItem(String named, int index, Field field) {
21 | this.named = named;
22 | this.index = index;
23 | this.field = field;
24 | }
25 |
26 | public String getNamed() {
27 | return named;
28 | }
29 |
30 | public int getIndex() {
31 | return index;
32 | }
33 |
34 | public Field getField() {
35 | return field;
36 | }
37 |
38 | @Override
39 | public boolean equals(Object o) {
40 | if (this == o) return true;
41 | if (o == null || getClass() != o.getClass()) return false;
42 | return hashCode() == o.hashCode();
43 | }
44 |
45 | @Override
46 | public int hashCode() {
47 | return Objects.hash(named, index);
48 | }
49 | }
50 |
--------------------------------------------------------------------------------
/src/main/java/cn/devezhao/persist4j/query/compiler/SelectItem.java:
--------------------------------------------------------------------------------
1 | package cn.devezhao.persist4j.query.compiler;
2 |
3 | import cn.devezhao.persist4j.Field;
4 |
5 | import java.io.Serializable;
6 |
7 | /**
8 | *
9 | *
10 | * @author FANGFANG ZHAO
11 | * @since 0.1, Feb 15, 2009
12 | * @version $Id: SelectItem.java 120 2016-01-07 11:49:05Z zhaofang123@gmail.com $
13 | */
14 | public interface SelectItem extends Serializable {
15 |
16 | Field getField();
17 |
18 | int getIndex();
19 |
20 | String getName();
21 |
22 | String getAlias();
23 |
24 | String getFieldPath();
25 |
26 | SelectItemType getType();
27 | }
28 |
--------------------------------------------------------------------------------
/src/main/java/cn/devezhao/persist4j/query/compiler/SelectItemType.java:
--------------------------------------------------------------------------------
1 | package cn.devezhao.persist4j.query.compiler;
2 |
3 | /**
4 | * @author FANGFANG ZHAO
5 | * @since 0.1, Feb 15, 2009
6 | * @version $Id: SelectItemType.java 8 2015-06-08 09:09:03Z zhaofang123@gmail.com $
7 | */
8 | public enum SelectItemType {
9 |
10 | Aggregator,
11 |
12 | Field,
13 |
14 | NameField,
15 |
16 | IDLabelField
17 | }
18 |
--------------------------------------------------------------------------------
/src/main/java/cn/devezhao/persist4j/query/compiler/antlr/AjQLParserTokenTypes.java:
--------------------------------------------------------------------------------
1 | // $ANTLR 2.7.7 (2006-11-01): "ajql.g" -> "AjQLParser.java"$
2 |
3 | package cn.devezhao.persist4j.query.compiler.antlr;
4 |
5 | public interface AjQLParserTokenTypes {
6 | int EOF = 1;
7 | int NULL_TREE_LOOKAHEAD = 3;
8 | int SELECT = 4;
9 | int DISTINCT = 5;
10 | int FROM = 6;
11 | int WHERE = 7;
12 | int AND = 8;
13 | int OR = 9;
14 | int ORDER = 10;
15 | int ASC = 11;
16 | int DESC = 12;
17 | int GROUP = 13;
18 | int BY = 14;
19 | int HAVING = 15;
20 | int WITH = 16;
21 | int ROLLUP = 17;
22 | int MAX = 18;
23 | int MIN = 19;
24 | int AVG = 20;
25 | int SUM = 21;
26 | int COUNT = 22;
27 | int DATE_FORMAT = 23;
28 | int YEAR = 24;
29 | int QUARTER = 25;
30 | int MONTH = 26;
31 | int WEEK = 27;
32 | int CONCAT = 28;
33 | int IS = 29;
34 | int NOT = 30;
35 | int NULL = 31;
36 | int IN = 32;
37 | int LIKE = 33;
38 | int EXISTS = 34;
39 | int BETWEEN = 35;
40 | int TRUE = 36;
41 | int FALSE = 37;
42 | int MATCH = 38;
43 | int GROUP_CONCAT = 39;
44 | int SEPARATOR = 40;
45 | int COMMA = 41;
46 | int LPAREN = 42;
47 | int QUOTED_STRING = 43;
48 | int RPAREN = 44;
49 | int IDENT = 45;
50 | int STAR = 46;
51 | int LITERAL = 47;
52 | int NAMED_PARAM = 48;
53 | int QUESTION_MARK = 49;
54 | int PLUS = 50;
55 | int MINUS = 51;
56 | int DIVIDE = 52;
57 | int MOD = 53;
58 | int EQ = 54;
59 | int LT = 55;
60 | int GT = 56;
61 | int LE = 57;
62 | int GE = 58;
63 | int SQL_NE = 59;
64 | int BAND = 60;
65 | int NBAND = 61;
66 | int DOT = 62;
67 | int COLON = 63;
68 | int IDENT_START = 64;
69 | int IDENT_LETTER = 65;
70 | int DIGIT = 66;
71 | int INT = 67;
72 | int NUM = 68;
73 | int ESCqs = 69;
74 | int WS = 70;
75 | }
76 |
--------------------------------------------------------------------------------
/src/main/java/cn/devezhao/persist4j/query/compiler/antlr/ParserException.java:
--------------------------------------------------------------------------------
1 | package cn.devezhao.persist4j.query.compiler.antlr;
2 |
3 | import antlr.ANTLRException;
4 | import org.apache.commons.lang.exception.NestableRuntimeException;
5 |
6 | /**
7 | * @author FANGFANG ZHAO
8 | * @since 0.1, Feb 10, 2009
9 | * @version $Id: ParserException.java 8 2015-06-08 09:09:03Z zhaofang123@gmail.com $
10 | */
11 | public class ParserException extends NestableRuntimeException {
12 | private static final long serialVersionUID = -3556487161116352482L;
13 |
14 | public ParserException(String message) {
15 | super(message);
16 | }
17 |
18 | public ParserException(ANTLRException ex) {
19 | super(ex);
20 | }
21 | }
22 |
--------------------------------------------------------------------------------
/src/main/java/cn/devezhao/persist4j/query/compiler/antlr/ParserHelper.java:
--------------------------------------------------------------------------------
1 | package cn.devezhao.persist4j.query.compiler.antlr;
2 |
3 | import java.io.StringReader;
4 |
5 | /**
6 | * @author FANGFANG ZHAO
7 | * @since 0.1, Feb 4, 2009
8 | * @version $Id: ParserHelper.java 8 2015-06-08 09:09:03Z zhaofang123@gmail.com $
9 | */
10 | public class ParserHelper {
11 |
12 | /**
13 | * @param ajql
14 | * @return
15 | */
16 | public static AjQLParser createAjQLParser(String ajql) {
17 | return createAjQLParser(ajql, false);
18 | }
19 |
20 | /**
21 | * @param ajql
22 | * @param isThrow
23 | * @return
24 | */
25 | public static AjQLParser createAjQLParser(String ajql, boolean isThrow) {
26 | AjQLLexer l = new AjQLLexer(new StringReader(ajql));
27 | return (isThrow) ? new ThrowerAjQLParser(l) : new AjQLParser(l);
28 | }
29 |
30 | /**
31 | * @param type
32 | * @return
33 | * @see #isAggregatorWithMode(int)
34 | * @see #isAggregatorWithNested(int)
35 | */
36 | public static boolean isAggregator(int type) {
37 | return type == AjQLParserTokenTypes.MIN
38 | || type == AjQLParserTokenTypes.MAX
39 | || type == AjQLParserTokenTypes.AVG
40 | || type == AjQLParserTokenTypes.SUM
41 | || type == AjQLParserTokenTypes.COUNT
42 | || type == AjQLParserTokenTypes.YEAR
43 | || type == AjQLParserTokenTypes.QUARTER
44 | || type == AjQLParserTokenTypes.MONTH
45 | || type == AjQLParserTokenTypes.WEEK
46 | || isAggregatorWithMode(type)
47 | || isAggregatorWithNested(type);
48 | }
49 |
50 | /**
51 | * @param type
52 | * @return
53 | */
54 | public static boolean isAggregatorWithMode(int type) {
55 | return type == AjQLParserTokenTypes.DATE_FORMAT || type == AjQLParserTokenTypes.GROUP_CONCAT;
56 | }
57 |
58 | /**
59 | * @param type
60 | * @return
61 | */
62 | public static boolean isAggregatorWithNested(int type) {
63 | return type == AjQLParserTokenTypes.CONCAT;
64 | }
65 |
66 | /**
67 | * 运算符
68 | *
69 | * @param type
70 | * @return
71 | */
72 | public static boolean isOperator(int type) {
73 | return type == AjQLParserTokenTypes.EQ
74 | || type == AjQLParserTokenTypes.LT
75 | || type == AjQLParserTokenTypes.GT
76 | || type == AjQLParserTokenTypes.LE
77 | || type == AjQLParserTokenTypes.GE
78 | || type == AjQLParserTokenTypes.SQL_NE
79 | || type == AjQLParserTokenTypes.LIKE
80 | || type == AjQLParserTokenTypes.BETWEEN
81 | || type == AjQLParserTokenTypes.BAND
82 | || type == AjQLParserTokenTypes.NBAND
83 | || type == AjQLParserTokenTypes.MATCH;
84 | }
85 |
86 | /**
87 | * 逻辑符
88 | *
89 | * @param type
90 | * @return
91 | */
92 | public static boolean isInIgnore(int type) {
93 | return type == AjQLParserTokenTypes.AND
94 | || type == AjQLParserTokenTypes.OR
95 | || type == AjQLParserTokenTypes.IS
96 | || type == AjQLParserTokenTypes.NOT
97 | || type == AjQLParserTokenTypes.LPAREN
98 | || type == AjQLParserTokenTypes.RPAREN
99 | || type == AjQLParserTokenTypes.COMMA
100 | || type == AjQLParserTokenTypes.MINUS
101 | || isOperator(type)
102 | || isInIgnoreValue(type);
103 | }
104 |
105 | /**
106 | * 特殊值
107 | *
108 | * @param type
109 | * @return
110 | */
111 | public static boolean isInIgnoreValue(int type) {
112 | return type == AjQLParserTokenTypes.TRUE
113 | || type == AjQLParserTokenTypes.FALSE
114 | || type == AjQLParserTokenTypes.LITERAL
115 | || type == AjQLParserTokenTypes.NULL;
116 | }
117 | }
118 |
--------------------------------------------------------------------------------
/src/main/java/cn/devezhao/persist4j/query/compiler/antlr/ThrowerAjQLParser.java:
--------------------------------------------------------------------------------
1 | package cn.devezhao.persist4j.query.compiler.antlr;
2 |
3 | import antlr.RecognitionException;
4 | import antlr.TokenStream;
5 |
6 | /**
7 | * @author FANGFANG ZHAO
8 | * @since 0.1, Feb 10, 2009
9 | * @version $Id: ThrowerAjQLParser.java 8 2015-06-08 09:09:03Z zhaofang123@gmail.com $
10 | */
11 | public class ThrowerAjQLParser extends AjQLParser {
12 |
13 | public ThrowerAjQLParser(TokenStream stream) {
14 | super(stream);
15 | }
16 |
17 | @Override
18 | public void reportError(RecognitionException ex) {
19 | throw new ParserException(ex);
20 | }
21 | }
22 |
--------------------------------------------------------------------------------
/src/main/java/cn/devezhao/persist4j/record/FieldValueException.java:
--------------------------------------------------------------------------------
1 | package cn.devezhao.persist4j.record;
2 |
3 | import cn.devezhao.persist4j.PersistException;
4 |
5 | /**
6 | * Invald value of field
7 | *
8 | * @author FANGFANG ZHAO
9 | * @since 0.1, 06/14/08
10 | * @version $Id: FieldValueException.java 8 2015-06-08 09:09:03Z zhaofang123@gmail.com $
11 | */
12 | public class FieldValueException extends PersistException {
13 | private static final long serialVersionUID = -2063367127310548765L;
14 |
15 | public FieldValueException(String message) {
16 | super(message);
17 | }
18 |
19 | public FieldValueException(String message, Throwable ex) {
20 | super(message, ex);
21 | }
22 | }
23 |
--------------------------------------------------------------------------------
/src/main/java/cn/devezhao/persist4j/record/JsonRecordCreator.java:
--------------------------------------------------------------------------------
1 | /*
2 | Copyright 2018 DEVEZHAO(zhaofang123@gmail.com)
3 |
4 | Licensed under the Apache License, Version 2.0 (the "License");
5 | you may not use this file except in compliance with the License.
6 | You may obtain a copy of the License at
7 |
8 | http://www.apache.org/licenses/LICENSE-2.0
9 |
10 | Unless required by applicable law or agreed to in writing, software
11 | distributed under the License is distributed on an "AS IS" BASIS,
12 | WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | See the License for the specific language governing permissions and
14 | limitations under the License.
15 | */
16 |
17 | package cn.devezhao.persist4j.record;
18 |
19 | import cn.devezhao.persist4j.Entity;
20 | import cn.devezhao.persist4j.Field;
21 | import cn.devezhao.persist4j.Record;
22 | import cn.devezhao.persist4j.engine.ID;
23 | import cn.devezhao.persist4j.engine.StandardRecord;
24 | import com.alibaba.fastjson.JSONObject;
25 | import org.apache.commons.lang.Validate;
26 | import org.apache.commons.logging.Log;
27 | import org.apache.commons.logging.LogFactory;
28 |
29 | import java.util.Map;
30 |
31 | /**
32 | * JSON 记录解析
33 | *
34 | * @author zhaofang123@gmail.com
35 | * @since 08/01/2018
36 | */
37 | public class JsonRecordCreator implements RecordCreator {
38 |
39 | private static final Log LOG = LogFactory.getLog(JsonRecordCreator.class);
40 |
41 | public static final String META_FIELD = "metadata";
42 |
43 | final protected Entity entity;
44 | final protected ID editor;
45 | final protected JSONObject source;
46 |
47 | /**
48 | * Create a new JsonRecordCreator. the format of document like
49 | *
50 | * {
51 | * metadata: { entity: 'User', [id='xx'] },
52 | * loginName: 'xx',
53 | * email: 'xx'
54 | * }
55 | *
56 | *
57 | * @param entity
58 | * @param source
59 | */
60 | public JsonRecordCreator(Entity entity, JSONObject source) {
61 | this(entity, source, null);
62 | }
63 |
64 | /**
65 | *
66 | * @param entity
67 | * @param source
68 | * @param editor
69 | */
70 | public JsonRecordCreator(Entity entity, JSONObject source, ID editor) {
71 | Validate.notNull(entity);
72 | Validate.notNull(source);
73 | this.entity = entity;
74 | this.source = source;
75 | this.editor = editor;
76 | }
77 |
78 | /**
79 | * @return
80 | */
81 | @Override
82 | public Record create() {
83 | return create(true);
84 | }
85 |
86 | /**
87 | * @param ignoreNullValueWhenNew 忽略空值（仅在新建时有效）
88 | * @return
89 | */
90 | public Record create(boolean ignoreNullValueWhenNew) {
91 | Record record = new StandardRecord(entity, editor);
92 |
93 | JSONObject metadata = source.getJSONObject(META_FIELD);
94 | if (metadata == null) {
95 | throw new FieldValueException("The `metadata` element must not be null");
96 | }
97 |
98 | String id = metadata.getString("id");
99 | if (ID.isId(id)) {
100 | record.setID(entity.getPrimaryField().getName(), ID.valueOf(id));
101 | }
102 |
103 | for (Map.Entry e : source.entrySet()) {
104 | String fileName = e.getKey();
105 | if (META_FIELD.equals(fileName)) continue;
106 |
107 | final Field field = entity.containsField(fileName) ? entity.getField(fileName) : null;
108 | if (field == null) {
109 | LOG.warn("Unable found field [ " + entity.getName() + '#' + fileName + " ], will ignore");
110 | continue;
111 | }
112 |
113 | Object fieldValue = e.getValue();
114 | if (fieldValue != null) fieldValue = fieldValue.toString();
115 |
116 | if (ignoreNullValueWhenNew && record.getPrimary() == null
117 | && (fieldValue == null || isNoValue((String) fieldValue, field))) {
118 | continue;
119 | }
120 |
121 | setFieldValue(field, (String) fieldValue, record);
122 | }
123 |
124 | afterCreate(record);
125 | return record;
126 | }
127 |
128 | protected void afterCreate(Record record) {
129 | verify(record);
130 | }
131 | }
132 |
--------------------------------------------------------------------------------
/src/main/java/cn/devezhao/persist4j/record/RecordCreator.java:
--------------------------------------------------------------------------------
1 | package cn.devezhao.persist4j.record;
2 |
3 | import cn.devezhao.persist4j.Field;
4 | import cn.devezhao.persist4j.Record;
5 | import cn.devezhao.persist4j.dialect.FieldType;
6 | import cn.devezhao.persist4j.engine.NullValue;
7 | import org.apache.commons.lang.StringUtils;
8 | import org.apache.commons.logging.Log;
9 | import org.apache.commons.logging.LogFactory;
10 |
11 | import java.util.ArrayList;
12 | import java.util.List;
13 |
14 | /**
15 | * @author FANGFANG ZHAO
16 | * @version $Id: RecordCreator.java 8 2015-06-08 09:09:03Z zhaofang123@gmail.com $
17 | * @since 0.1, Feb 14, 2009
18 | */
19 | public interface RecordCreator {
20 |
21 | Log LOG = LogFactory.getLog(RecordCreator.class);
22 |
23 | /**
24 | * 创建 Record
25 | *
26 | * @return
27 | */
28 | Record create();
29 |
30 | /**
31 | * #setFieldValue 产生警告时调用
32 | *
33 | * @param field
34 | * @param value
35 | * @param record
36 | * @return
37 | */
38 | default boolean onSetFieldValueWarn(Field field, String value, Record record) {
39 | return false;
40 | }
41 |
42 | /**
43 | * 设置字段值
44 | *
45 | * @param field
46 | * @param value
47 | * @param record
48 | * @return
49 | */
50 | default boolean setFieldValue(Field field, String value, Record record) {
51 | final boolean isNew = record.getPrimary() == null;
52 |
53 | // 忽略更新
54 | if (!isNew && !field.isUpdatable()) {
55 | if (!onSetFieldValueWarn(field, value, record)) {
56 | LOG.warn("Could not put value to un-updatable field : " + field);
57 | return false;
58 | }
59 | }
60 |
61 | // 忽略新建
62 | if (isNew && !field.isCreatable()) {
63 | if (!onSetFieldValueWarn(field, value, record)) {
64 | LOG.warn("Could not put value to un-creatable field : " + field);
65 | return false;
66 | }
67 | }
68 |
69 | final boolean noValue = isNoValue(value, field);
70 |
71 | // 无值
72 | if (noValue) {
73 | if (isNew && !field.isNullable() && !field.isAutoValue()) {
74 | throw new FieldValueException("Field [ " + field + " ] must not be null");
75 | }
76 | }
77 |
78 | if (noValue) {
79 | record.setNull(field.getName());
80 | } else {
81 | try {
82 | RecordVisitor.setValueByLiteral(field, value, record);
83 | } catch (Exception ex) {
84 | LOG.error("Invalid field value by literal : " + field.getName() + " = " + value);
85 | throw ex;
86 | }
87 | }
88 | return true;
89 | }
90 |
91 | /**
92 | * 验证字段约束
93 | *
94 | * @param record
95 | */
96 | default void verify(Record record) {
97 | if (record.getPrimary() == null) return;
98 |
99 | List notNullable = new ArrayList<>();
100 | for (Field field : record.getEntity().getFields()) {
101 | if (FieldType.PRIMARY.equals(field.getType())) continue;
102 |
103 | Object val = record.getObjectValue(field.getName());
104 | if (!field.isNullable() && !field.isAutoValue() && (val == null || NullValue.is(val))) {
105 | notNullable.add(field.getName());
106 | }
107 | }
108 |
109 | if (!notNullable.isEmpty()) {
110 | throw new FieldValueException("Must not be null. Entity [ " + record.getEntity().getName()
111 | + " ], Fields [ " + StringUtils.join(notNullable.toArray(new String[0]), ", ") + " ]");
112 | }
113 | }
114 |
115 | /**
116 | * 是否无值/空值
117 | *
118 | * @param value
119 | * @param field
120 | * @return
121 | */
122 | default boolean isNoValue(String value, Field field) {
123 | return value == null || StringUtils.isEmpty(value);
124 | }
125 | }
126 |
--------------------------------------------------------------------------------
/src/main/java/cn/devezhao/persist4j/record/XmlBean.java:
--------------------------------------------------------------------------------
1 | package cn.devezhao.persist4j.record;
2 |
3 | import cn.devezhao.persist4j.Entity;
4 | import cn.devezhao.persist4j.engine.ID;
5 |
6 | /**
7 | *
8 | *
9 | * @author FANGFANG ZHAO
10 | * @since 0.1, Feb 14, 2009
11 | * @version $Id: XMLBean.java 8 2015-06-08 09:09:03Z zhaofang123@gmail.com $
12 | */
13 | public class XmlBean {
14 |
15 | public XmlBean(Entity entity) {
16 | }
17 |
18 | public XmlBean(Entity entity, ID recordId) {
19 | }
20 |
21 | public XmlBean addValue(String fieldName, String value) {
22 | return this;
23 | }
24 | }
25 |
--------------------------------------------------------------------------------
/src/main/java/cn/devezhao/persist4j/record/XmlRecordCreator.java:
--------------------------------------------------------------------------------
1 | package cn.devezhao.persist4j.record;
2 |
3 | import cn.devezhao.persist4j.Entity;
4 | import cn.devezhao.persist4j.Field;
5 | import cn.devezhao.persist4j.Record;
6 | import cn.devezhao.persist4j.engine.ID;
7 | import cn.devezhao.persist4j.engine.StandardRecord;
8 | import org.apache.commons.lang.StringUtils;
9 | import org.apache.commons.lang.Validate;
10 | import org.apache.commons.logging.Log;
11 | import org.apache.commons.logging.LogFactory;
12 | import org.dom4j.Element;
13 |
14 | /**
15 | * @author FANGFANG ZHAO
16 | * @since 0.1, Feb 14, 2009
17 | * @version $Id: XMLRecordCreator.java 8 2015-06-08 09:09:03Z zhaofang123@gmail.com $
18 | */
19 | @Deprecated
20 | public class XmlRecordCreator implements RecordCreator {
21 |
22 | private static final Log LOG = LogFactory.getLog(XmlRecordCreator.class);
23 |
24 | final protected Entity entity;
25 | final protected ID editor;
26 | final protected Element source;
27 |
28 | /**
29 | * Create a new XmlRecordCreator. the format of document like
30 | *
31 | * <entity name="..." [id="..."]>
32 | * <userName value="..." [label"..."]/>
33 | * <description>
34 | * ...
35 | * </description>
36 | * </entity>
37 | *
38 | *
39 | * @param entity
40 | * @param source
41 | */
42 | public XmlRecordCreator(Entity entity, Element source) {
43 | this(entity, source, null);
44 | }
45 |
46 | /**
47 | *
48 | * @param entity
49 | * @param source
50 | * @param editor
51 | */
52 | public XmlRecordCreator(Entity entity, Element source, ID editor) {
53 | Validate.notNull(entity);
54 | Validate.notNull(source);
55 | this.entity = entity;
56 | this.source = source;
57 | this.editor = editor;
58 | }
59 |
60 | @Override
61 | public Record create() {
62 | Record record = new StandardRecord(entity, editor);
63 |
64 | String id = source.valueOf("@id");
65 | if (!StringUtils.isBlank(id)) {
66 | record.setID(entity.getPrimaryField().getName(), ID.valueOf(id));
67 | }
68 |
69 | for (Object o : source.elements()) {
70 | Element el = (Element) o;
71 | String fieldName = el.getName();
72 | String fieldValue = el.attributeValue("value");
73 | if (fieldValue == null) {
74 | fieldValue = el.getText();
75 | }
76 |
77 | Field field = entity.containsField(fieldName) ? entity.getField(fieldName) : null;
78 | if (field == null) {
79 | LOG.warn("Unable found field [ " + entity.getName() + '#' + fieldName + " ], will ignore");
80 | continue;
81 | }
82 |
83 | setFieldValue(field, fieldValue, record);
84 | }
85 |
86 | afterCreate(record);
87 | return record;
88 | }
89 |
90 | protected void afterCreate(Record record) {
91 | verify(record);
92 | }
93 | }
94 |
--------------------------------------------------------------------------------
/src/main/java/cn/devezhao/persist4j/util/CaseInsensitiveMap.java:
--------------------------------------------------------------------------------
1 | package cn.devezhao.persist4j.util;
2 |
3 | import java.util.*;
4 |
5 | /**
6 | * 大小写不敏感 Map
7 | *
8 | * @author FANGFANG ZHAO
9 | * @since 08/17/2019
10 | */
11 | public class CaseInsensitiveMap implements Map {
12 |
13 | private Map originalKeys = new HashMap<>();
14 |
15 | private Map delegate = new org.apache.commons.collections4.map.CaseInsensitiveMap<>();
16 |
17 | public CaseInsensitiveMap() {
18 | super();
19 | }
20 |
21 | public CaseInsensitiveMap(Map m) {
22 | this();
23 | this.putAll(m);
24 | }
25 |
26 | @Override
27 | public int size() {
28 | return delegate.size();
29 | }
30 |
31 | @Override
32 | public boolean isEmpty() {
33 | return delegate.isEmpty();
34 | }
35 |
36 | @Override
37 | public boolean containsKey(Object key) {
38 | return delegate.containsKey(key);
39 | }
40 |
41 | @Override
42 | public boolean containsValue(Object value) {
43 | return delegate.containsValue(value);
44 | }
45 |
46 | @Override
47 | public V get(Object key) {
48 | return delegate.get(key);
49 | }
50 |
51 | @Override
52 | public V put(String key, V value) {
53 | originalKeys.put(convertKey(key), key);
54 | return delegate.put(key, value);
55 | }
56 |
57 | @Override
58 | public V remove(Object key) {
59 | originalKeys.remove(convertKey(key));
60 | return delegate.remove(key);
61 | }
62 |
63 | @Override
64 | public void putAll(Map m) {
65 | for (String key : m.keySet()) {
66 | originalKeys.put(convertKey(key), key);
67 | }
68 | delegate.putAll(m);
69 | }
70 |
71 | @Override
72 | public void clear() {
73 | originalKeys.clear();
74 | delegate.clear();
75 | }
76 |
77 | @Override
78 | public Set keySet() {
79 | return new HashSet<>(originalKeys.values());
80 | }
81 |
82 | @Override
83 | public Collection values() {
84 | return delegate.values();
85 | }
86 |
87 | @Override
88 | public Set> entrySet() {
89 | Set> es = new HashSet<>();
90 | for (Entry e : delegate.entrySet()) {
91 | String oKey = originalKeys.get(convertKey(e.getKey()));
92 | es.add(new CIEntry(oKey, e.getValue()));
93 | }
94 | return es;
95 | }
96 |
97 | private String convertKey(final Object key) {
98 | return key.toString().toUpperCase();
99 | }
100 |
101 | /**
102 | */
103 | class CIEntry implements Entry {
104 |
105 | private String key;
106 | private V value;
107 |
108 | protected CIEntry(String key, V value) {
109 | this.key = key;
110 | this.value = value;
111 | }
112 |
113 | @Override
114 | public String getKey() {
115 | return key;
116 | }
117 | @Override
118 | public V getValue() {
119 | return value;
120 | }
121 | @Override
122 | public V setValue(V value) {
123 | this.value = value;
124 | return value;
125 | }
126 | @Override
127 | public String toString() {
128 | return String.format("%s=%s", key, value);
129 | }
130 |
131 | @Override
132 | public boolean equals(Object o) {
133 | if (this == o) return true;
134 | if (o == null || getClass() != o.getClass()) return false;
135 | return o.hashCode() == hashCode();
136 | }
137 |
138 | @Override
139 | public int hashCode() {
140 | return Objects.hash(key, value);
141 | }
142 | }
143 | }
144 |
--------------------------------------------------------------------------------
/src/main/java/cn/devezhao/persist4j/util/StringHelper.java:
--------------------------------------------------------------------------------
1 | package cn.devezhao.persist4j.util;
2 |
3 | import org.apache.commons.lang.CharSet;
4 | import org.apache.commons.lang.StringUtils;
5 |
6 | /**
7 | * @author FANGFANG ZHAO
8 | * @version $Id: StringHelper.java 119 2011-05-15 16:56:18Z
9 | * zhaofang123@gmail.com $
10 | * @since 0.1, Feb 17, 2009
11 | */
12 | public class StringHelper {
13 |
14 | private StringHelper() {
15 | }
16 |
17 | /**
18 | * 验证字符有效
19 | * 规则：['a-zA-Z', '_', '#'], can contains ['a-zA-Z', '_', '#', '0-9']
20 | *
21 | * @param ident
22 | * @return
23 | */
24 | public static boolean isIdentifier(String ident) {
25 | if (StringUtils.isBlank(ident)) {
26 | return false;
27 | }
28 |
29 | char start = ident.charAt(0);
30 | if (!(CharSet.ASCII_ALPHA.contains(start) || start == '_' || start == '#')) {
31 | return false;
32 | }
33 |
34 | for (char ch : ident.toCharArray()) {
35 | if (!(CharSet.ASCII_ALPHA.contains(ch) || CharSet.ASCII_NUMERIC.contains(ch) || ch == '_' || ch == '#')) {
36 | return false;
37 | }
38 | }
39 |
40 | return true;
41 | }
42 |
43 | /**
44 | * 根据驼峰加入下划线并大些，例如：userName > USER_NAME
45 | *
46 | * @param s
47 | * @return
48 | */
49 | public static String hyphenate(final String s) {
50 | char[] chars = s.toCharArray();
51 | int lenth = s.length();
52 |
53 | StringBuilder sb = new StringBuilder();
54 | boolean lastAZ = false;
55 | boolean lastInsert = false;
56 | for (int i = 0; i < lenth; i++) {
57 | char ch = chars[i];
58 | if (ch >= 'A' && ch <= 'Z') {
59 | if (i > 0) {
60 | if (!lastAZ) {
61 | sb.append('_');
62 | lastInsert = true;
63 | } else {
64 | lastInsert = false;
65 | }
66 | }
67 |
68 | lastAZ = true;
69 | } else {
70 | if (lastAZ) {
71 | if (!lastInsert && i > 1) {
72 | if (!(ch >= '0' && ch <= '9')) {
73 | sb.insert(sb.length() - 1, '_');
74 | }
75 | }
76 | }
77 |
78 | lastAZ = false;
79 | lastInsert = false;
80 | }
81 | sb.append(ch);
82 | }
83 | return sb.toString().replaceAll("__", "_").toLowerCase();
84 | }
85 |
86 | }
87 |
--------------------------------------------------------------------------------
/src/main/java/cn/devezhao/persist4j/util/support/QueryHelper.java:
--------------------------------------------------------------------------------
1 | package cn.devezhao.persist4j.util.support;
2 |
3 | import cn.devezhao.persist4j.Query;
4 | import cn.devezhao.persist4j.Record;
5 | import cn.devezhao.persist4j.query.IQuery;
6 |
7 | import java.util.ArrayList;
8 | import java.util.Collections;
9 | import java.util.List;
10 |
11 | /**
12 | * @author Zhao Fangfang
13 | * @version $Id: QueryHelper.java 8 2015-06-08 09:09:03Z zhaofang123@gmail.com $
14 | * @since 2.6, 2012-10-22
15 | */
16 | public class QueryHelper {
17 |
18 | /**
19 | * 分页读取
20 | *
21 | * @param query
22 | * @return
23 | */
24 | public static Object[][] readArray(IQuery query) {
25 | return readArray(query, 10000);
26 | }
27 |
28 | /**
29 | * 分页读取
30 | *
31 | * @param query
32 | * @param pageSize
33 | * @return
34 | */
35 | public static Object[][] readArray(IQuery query, int pageSize) {
36 | List tempList = new ArrayList<>();
37 | int pageNo = 1;
38 | while (true) {
39 | query.reset();
40 | query.setLimit(pageSize, pageNo++ * pageSize - pageSize);
41 | Object[][] temp = query.array();
42 | Collections.addAll(tempList, temp);
43 |
44 | if (temp.length < pageSize) {
45 | break;
46 | }
47 | }
48 | return tempList.toArray(new Object[tempList.size()][]);
49 | }
50 |
51 | /**
52 | * 分页读取
53 | *
54 | * @param query
55 | * @return
56 | */
57 | public static List readList(Query query) {
58 | return readList(query, 10000);
59 | }
60 |
61 | /**
62 | * 分页读取
63 | *
64 | * @param query
65 | * @param pageSize
66 | * @return
67 | */
68 | public static List readList(Query query, int pageSize) {
69 | List tempList = new ArrayList<>();
70 | int pageNo = 1;
71 | while (true) {
72 | query.result().reset();
73 | query.setLimit(pageSize, pageNo++ * pageSize - pageSize);
74 | List temp = query.list();
75 | tempList.addAll(temp);
76 |
77 | if (temp.size() < pageSize) {
78 | break;
79 | }
80 | }
81 | return tempList;
82 | }
83 | }
84 |
--------------------------------------------------------------------------------
/src/main/java/cn/devezhao/persist4j/util/support/SchemaExport.java:
--------------------------------------------------------------------------------
1 | package cn.devezhao.persist4j.util.support;
2 |
3 | import cn.devezhao.persist4j.Entity;
4 | import cn.devezhao.persist4j.PersistException;
5 | import cn.devezhao.persist4j.dialect.Dialect;
6 | import cn.devezhao.persist4j.metadata.MetadataFactory;
7 | import cn.devezhao.persist4j.metadata.impl.ConfigurationMetadataFactory;
8 | import cn.devezhao.persist4j.util.SqlHelper;
9 | import org.dom4j.Element;
10 |
11 | import java.sql.Connection;
12 | import java.sql.SQLException;
13 | import java.sql.Statement;
14 | import java.util.List;
15 |
16 | /**
17 | * 生成建表语句
18 | *
19 | * @author FANGFANG ZHAO
20 | * @since 0.1, 06/15/08
21 | * @version $Id: SchemaExport.java 8 2015-06-08 09:09:03Z zhaofang123@gmail.com $
22 | */
23 | public class SchemaExport {
24 |
25 | final private MetadataFactory metadataFactory;
26 | final private Dialect dialect;
27 | final private Connection connection;
28 |
29 | /**
30 | * @param metadataFactory
31 | * @param dialect
32 | * @param connect
33 | */
34 | public SchemaExport(MetadataFactory metadataFactory, Dialect dialect, Connection connect) {
35 | this.metadataFactory = metadataFactory;
36 | this.dialect = dialect;
37 | this.connection = connect;
38 | }
39 |
40 | /**
41 | * @param dropExists
42 | * @param createFK
43 | */
44 | @SuppressWarnings("deprecation")
45 | public void export(boolean dropExists, boolean createFK) {
46 | Entity[] entities = metadataFactory.getEntities();
47 | Element cfgRoot = ((ConfigurationMetadataFactory) metadataFactory).getConfigDocument().getRootElement();
48 |
49 | try {
50 | for (Entity entity : entities) {
51 | Element entityElement = (Element) cfgRoot.selectSingleNode("//entity[@name='" + entity.getName() + "']");
52 | List ix = entityElement.selectNodes("index");
53 | String[] sqls = new Table(entity, dialect, ix).generateDDL(dropExists, createFK);
54 |
55 | for (String sql : sqls) {
56 | try (Statement stmt = connection.createStatement()) {
57 | stmt.execute(sql);
58 | }
59 | }
60 | }
61 | } catch (SQLException e) {
62 | throw new PersistException(null, e);
63 | } finally {
64 | SqlHelper.close(connection);
65 | }
66 | }
67 | }
68 |
--------------------------------------------------------------------------------
/src/main/java/cn/devezhao/persist4j/util/support/SecurityPreferencesConfigurer.java:
--------------------------------------------------------------------------------
1 | package cn.devezhao.persist4j.util.support;
2 |
3 | import org.springframework.beans.factory.config.PreferencesPlaceholderConfigurer;
4 |
5 | import java.io.IOException;
6 | import java.util.Properties;
7 |
8 | /**
9 | *
10 | *
11 | * @author FANGFANG ZHAO
12 | * @since 0.1, 12/04/08
13 | * @version $Id: SecurityPreferencesConfigurer.java 8 2015-06-08 09:09:03Z zhaofang123@gmail.com $
14 | */
15 | public class SecurityPreferencesConfigurer extends PreferencesPlaceholderConfigurer {
16 |
17 | @Override
18 | protected void loadProperties(Properties props) throws IOException {
19 | super.loadProperties(props);
20 | this.afterLoad(props);
21 | }
22 |
23 | /*-
24 | */
25 | protected void afterLoad(Properties props) {
26 | String dbUrl = props.getProperty("db.url");
27 |
28 | if (dbUrl.startsWith("jdbc:mysql://")) {
29 | String append = "";
30 | // Fix MySQL Connector/J 3.1 or higher Bugs
31 | if (!dbUrl.contains("zeroDateTimeBehavior")) {
32 | append = "&zeroDateTimeBehavior=convertToNull";
33 | }
34 |
35 | // MySQL using PreparedStatement. MUST BE MYSQL5 OR HIGHER
36 | if (!dbUrl.contains("useServerPrepStmts")) {
37 | append += "&useServerPrepStmts=true";
38 | }
39 |
40 | if (append.length() > 1) {
41 | props.put("db.url", dbUrl + append);
42 | }
43 | }
44 | }
45 | }
46 |
--------------------------------------------------------------------------------
/src/main/resources/metadata.dtd:
--------------------------------------------------------------------------------
1 |
2 |
5 |
6 |
9 |
10 |
11 |
15 |
19 |
20 |
23 |
24 |
40 |
55 |
56 |
59 |
60 |
79 |
98 |
99 |
102 |
103 |
108 |
--------------------------------------------------------------------------------
/src/test/java/cn/devezhao/persist4j/BaseTest.java:
--------------------------------------------------------------------------------
1 | package cn.devezhao.persist4j;
2 |
3 | import cn.devezhao.persist4j.dialect.Dialect;
4 | import cn.devezhao.persist4j.dialect.h2.H2Dialect;
5 | import cn.devezhao.persist4j.engine.PersistManagerFactoryImpl;
6 | import cn.devezhao.persist4j.metadata.MetadataFactory;
7 | import cn.devezhao.persist4j.metadata.impl.ConfigurationMetadataFactory;
8 | import cn.devezhao.persist4j.util.support.SchemaExport;
9 | import com.alibaba.druid.pool.DruidDataSourceFactory;
10 |
11 | import javax.sql.DataSource;
12 | import java.util.HashMap;
13 | import java.util.Map;
14 |
15 | /**
16 | * @author zhaofang123@gmail.com
17 | * @since 11/27/2019
18 | */
19 | public class BaseTest {
20 |
21 | private DataSource ds;
22 | private PersistManagerFactory persistManagerFactory;
23 |
24 | /**
25 | * 获取持久化工厂
26 | *
27 | * @return
28 | */
29 | protected PersistManagerFactory getPersistManagerFactory() {
30 | if (persistManagerFactory == null) {
31 | Dialect dialect = new H2Dialect();
32 | MetadataFactory metadata = new ConfigurationMetadataFactory("metadata-test.xml", dialect);
33 | this.persistManagerFactory = new PersistManagerFactoryImpl(getDataSource(), dialect, metadata);
34 | }
35 | return persistManagerFactory;
36 | }
37 |
38 | /**
39 | * 获取 H2 数据源（内存表）。已知于 MySQL 兼容问题：
40 | * 1. 不支持 double（不支持指定小数位）
41 | * 2. 不支付全文索引，及对 CLOB/TEXT 字段建立索引
42 | *
43 | * @return
44 | */
45 | protected DataSource getDataSource() {
46 | if (ds == null) {
47 | Map dbProps = new HashMap<>();
48 | dbProps.put("url", "jdbc:h2:mem:test;MODE=MYSQL;DATABASE_TO_LOWER=TRUE;IGNORECASE=TRUE");
49 | dbProps.put("testWhileIdle", "false");
50 | try {
51 | this.ds = DruidDataSourceFactory.createDataSource(dbProps);
52 | } catch (Exception e) {
53 | throw new RuntimeException(e);
54 | }
55 | }
56 | return ds;
57 | }
58 |
59 | /**
60 | * 建表
61 | *
62 | * @throws Exception
63 | */
64 | protected void readySchemes() throws Exception {
65 | PersistManagerFactory factory = getPersistManagerFactory();
66 | SchemaExport schemaExport = new SchemaExport(
67 | factory.getMetadataFactory(), factory.getDialect(), factory.getDataSource().getConnection());
68 | schemaExport.export(false, false);
69 | }
70 | }
71 |
--------------------------------------------------------------------------------
/src/test/java/cn/devezhao/persist4j/RecordTest.java:
--------------------------------------------------------------------------------
1 | package cn.devezhao.persist4j;
2 |
3 | import cn.devezhao.persist4j.dialect.Dialect;
4 | import cn.devezhao.persist4j.dialect.MySQL5Dialect;
5 | import cn.devezhao.persist4j.engine.ID;
6 | import cn.devezhao.persist4j.engine.StandardRecord;
7 | import cn.devezhao.persist4j.metadata.MetadataFactory;
8 | import cn.devezhao.persist4j.metadata.impl.ConfigurationMetadataFactory;
9 | import org.junit.Test;
10 |
11 | import java.math.BigDecimal;
12 | import java.util.Iterator;
13 |
14 | /**
15 | *
16 | * @author zhaofang123@gmail.com
17 | * @since 12/28/2018
18 | */
19 | public class RecordTest {
20 |
21 | @Test
22 | public void testAnyReference() throws Exception {
23 | Dialect dialect = new MySQL5Dialect();
24 | MetadataFactory metadataFactory = new ConfigurationMetadataFactory("metadata-test.xml", dialect);
25 | Entity allTypes = metadataFactory.getEntity(100);
26 |
27 | Record record = new StandardRecord(allTypes);
28 | record.setID("tAnyReference2", ID.newId(102));
29 | record.setID("tAnyReference2", ID.newId(103));
30 |
31 | for (Iterator iter = record.getAvailableFieldIterator(); iter.hasNext(); ) {
32 | System.out.println(iter.next());
33 | }
34 | }
35 |
36 | @Test
37 | public void testTypeCompatible() throws Exception {
38 | Dialect dialect = new MySQL5Dialect();
39 | MetadataFactory metadataFactory = new ConfigurationMetadataFactory("metadata-test.xml", dialect);
40 | Entity allTypes = metadataFactory.getEntity(100);
41 |
42 | Record record = new StandardRecord(allTypes);
43 |
44 | record.setInt("tLong", 123);
45 | System.out.println(record.getInt("tLong"));
46 | record.setLong("tInt", 123L);
47 | System.out.println(record.getLong("tInt"));
48 |
49 | record.setDecimal("tDouble", BigDecimal.valueOf(123.123d));
50 | System.out.println(record.getDouble("tDouble"));
51 | record.setDouble("tDecimal", 123.123D);
52 | System.out.println(record.getDecimal("tDecimal"));
53 | }
54 | }
55 |
--------------------------------------------------------------------------------
/src/test/java/cn/devezhao/persist4j/engine/PersistManagerTest.java:
--------------------------------------------------------------------------------
1 | package cn.devezhao.persist4j.engine;
2 |
3 | import cn.devezhao.commons.CalendarUtils;
4 | import cn.devezhao.persist4j.BaseTest;
5 | import cn.devezhao.persist4j.PersistManagerFactory;
6 | import cn.devezhao.persist4j.Record;
7 | import org.apache.commons.lang.StringUtils;
8 | import org.junit.Test;
9 |
10 | import java.time.LocalTime;
11 |
12 | /**
13 | * @author devezhao zhaofang123@gmail.com
14 | * @since 2019年11月28日
15 | */
16 | public class PersistManagerTest extends BaseTest {
17 |
18 | @Test
19 | public void testComplicated() throws Exception {
20 | this.readySchemes();
21 | PersistManagerFactory factory = getPersistManagerFactory();
22 |
23 | Record record = new StandardRecord(factory.getMetadataFactory().getEntity("TestAllType"));
24 | record.setString("tText", "CLOB/Text");
25 | record.setIDArray("tReferenceList", new ID[] { ID.newId(100), ID.newId(100) });
26 | record.setObjectValue("tTime", LocalTime.now());
27 | record.setDate("tTime", CalendarUtils.now());
28 | factory.createPersistManager().save(record);
29 |
30 | Object[][] array = factory.createQuery("select tText,tReferenceList,tTime from TestAllType").setLimit(1).array();
31 | for (Object[] o : array) {
32 | System.out.println(o[0]);
33 | System.out.println(StringUtils.join((ID[]) o[1], ","));
34 | System.out.println(o[2]);
35 | }
36 | }
37 |
38 | }
39 |
--------------------------------------------------------------------------------
/src/test/java/cn/devezhao/persist4j/metadata/MetadataTest.java:
--------------------------------------------------------------------------------
1 | package cn.devezhao.persist4j.metadata;
2 |
3 | import cn.devezhao.persist4j.Entity;
4 | import cn.devezhao.persist4j.Field;
5 | import cn.devezhao.persist4j.dialect.Dialect;
6 | import cn.devezhao.persist4j.dialect.MySQL5Dialect;
7 | import cn.devezhao.persist4j.metadata.impl.ConfigurationMetadataFactory;
8 | import org.apache.commons.lang.StringUtils;
9 | import org.junit.Test;
10 |
11 | import java.util.Arrays;
12 |
13 | /**
14 | * @author zhaofang123@gmail.com
15 | * @since 12/28/2018
16 | */
17 | public class MetadataTest {
18 |
19 | @Test
20 | public void testAnyReference() throws Exception {
21 | final MetadataFactory metadataFactory = createMetadataFactory();
22 |
23 | Entity allTypes = metadataFactory.getEntity(100);
24 | Field anyReference1 = allTypes.getField("tanyreference");
25 | System.out.println(anyReference1.getType() + " : " + StringUtils.join(anyReference1.getReferenceEntities(), " | "));
26 |
27 | Field anyReference2 = allTypes.getField("tanyreference2");
28 | System.out.println(anyReference2.getType() + " : " + StringUtils.join(anyReference2.getReferenceEntities(), " | "));
29 | }
30 |
31 | @Test
32 | public void testReferenceToFields() throws Exception {
33 | final MetadataFactory metadataFactory = createMetadataFactory();
34 |
35 | Entity test = metadataFactory.getEntity(102);
36 | for (Field field : test.getReferenceToFields()) {
37 | System.out.println(field);
38 | }
39 | }
40 |
41 | @Test
42 | public void testCommon() {
43 | final MetadataFactory metadataFactory = createMetadataFactory();
44 |
45 | Field commonReference = metadataFactory.getEntity("Test2").getField("commonReference");
46 | System.out.println(commonReference);
47 | }
48 |
49 | @Test
50 | public void testNDetails() {
51 | final MetadataFactory metadataFactory = createMetadataFactory();
52 |
53 | Entity main = metadataFactory.getEntity(200);
54 | System.out.println(main.getDetailEntity());
55 | System.out.println(Arrays.toString(main.getDetialEntities()));
56 |
57 | Entity detail1 = metadataFactory.getEntity(201);
58 | System.out.println(detail1.getMainEntity());
59 |
60 | Entity detail2 = metadataFactory.getEntity(202);
61 | System.out.println(detail2.getMainEntity());
62 |
63 | System.out.println(detail1.getMainEntity() == (detail2.getMainEntity()));
64 | }
65 |
66 | static MetadataFactory createMetadataFactory() {
67 | Dialect dialect = new MySQL5Dialect();
68 | return new ConfigurationMetadataFactory("metadata-test.xml", dialect);
69 | }
70 | }
71 |
--------------------------------------------------------------------------------
/src/test/java/cn/devezhao/persist4j/query/QueryTest.java:
--------------------------------------------------------------------------------
1 | package cn.devezhao.persist4j.query;
2 |
3 | import cn.devezhao.persist4j.BaseTest;
4 | import cn.devezhao.persist4j.Query;
5 | import cn.devezhao.persist4j.engine.ID;
6 | import org.junit.Test;
7 |
8 | /**
9 | * @author FANGFANG ZHAO
10 | * @since 04/16/2020
11 | */
12 | public class QueryTest extends BaseTest {
13 |
14 | @Test
15 | public void testParameterInNested() throws Exception {
16 | readySchemes();
17 | createQuery("select tPrimary from TestAllType where tDouble > ? and"
18 | + " tInt in (select tInt from TestAllType where tPrimary = ?)")
19 | .setParameter(1, 1d)
20 | .setParameter(2, ID.newId(0))
21 | .unique();
22 | }
23 |
24 | @Test
25 | public void testParameterExistsNested() throws Exception {
26 | readySchemes();
27 | createQuery("select tPrimary from TestAllType where"
28 | + " exists (select t2Primary from Test2 where ^tReference = t2Reference and t2Int = ?) and"
29 | + " tText = ?")
30 | .setParameter(1, 1)
31 | .setParameter(2, "123")
32 | .unique();
33 | }
34 |
35 | @Test
36 | public void testParameterInAndExistsNested() throws Exception {
37 | readySchemes();
38 | createQuery("select tPrimary from TestAllType where"
39 | + " exists (select t2Primary from Test2 where ^tReference = t2Reference and t2Int = ?) and"
40 | + " tText = ? and"
41 | + " tInt in (select tInt from TestAllType where tPrimary = ?)")
42 | .setParameter(1, 1)
43 | .setParameter(2, "123")
44 | .setParameter(3, ID.newId(0))
45 | .unique();
46 | }
47 |
48 | /**
49 | * 创建查询
50 | *
51 | * @param ajql
52 | * @return
53 | */
54 | private Query createQuery(String ajql) {
55 | return getPersistManagerFactory().createQuery(ajql);
56 | }
57 | }
58 |
--------------------------------------------------------------------------------
/src/test/java/cn/devezhao/persist4j/query/compiler/Compiler.java:
--------------------------------------------------------------------------------
1 | package cn.devezhao.persist4j.query.compiler;
2 |
3 | import cn.devezhao.persist4j.PersistManagerFactory;
4 | import cn.devezhao.persist4j.dialect.Dialect;
5 | import cn.devezhao.persist4j.dialect.MySQL5Dialect;
6 | import cn.devezhao.persist4j.engine.PersistManagerFactoryImpl;
7 | import cn.devezhao.persist4j.metadata.MetadataFactory;
8 | import cn.devezhao.persist4j.metadata.impl.ConfigurationMetadataFactory;
9 |
10 | /**
11 | *
12 | * @author zhaofang123@gmail.com
13 | * @since 03/05/2018
14 | */
15 | public class Compiler {
16 |
17 | /**
18 | * @param ajql
19 | * @return
20 | */
21 | protected String compile(String ajql) {
22 | Dialect dialect = new MySQL5Dialect();
23 | MetadataFactory metadataFactory = new ConfigurationMetadataFactory("metadata-test.xml", dialect);
24 | PersistManagerFactory persistManagerFactory = new PersistManagerFactoryImpl(null, dialect, metadataFactory);
25 | QueryCompiler queryCompiler = new QueryCompiler(ajql);
26 | String rawSql = queryCompiler.compile(persistManagerFactory.getSQLExecutorContext());
27 | queryCompiler.getSelectItems();
28 | return rawSql;
29 | }
30 |
31 | /**
32 | * @param ajql
33 | * @return
34 | */
35 | protected QueryCompiler createCompiler(String ajql) {
36 | Dialect dialect = new MySQL5Dialect();
37 | MetadataFactory metadataFactory = new ConfigurationMetadataFactory("metadata-test.xml", dialect);
38 | PersistManagerFactory persistManagerFactory = new PersistManagerFactoryImpl(null, dialect, metadataFactory);
39 | QueryCompiler queryCompiler = new QueryCompiler(ajql);
40 | queryCompiler.compile(persistManagerFactory.getSQLExecutorContext());
41 | return queryCompiler;
42 | }
43 | }
44 |
--------------------------------------------------------------------------------
/src/test/java/cn/devezhao/persist4j/record/RecordVisitorTest.java:
--------------------------------------------------------------------------------
1 | package cn.devezhao.persist4j.record;
2 |
3 | import org.junit.Test;
4 |
5 | public class RecordVisitorTest {
6 |
7 | @Test
8 | public void testClearNumber() {
9 | System.out.println(RecordVisitor.clearNumber("123 ", Boolean.TRUE));
10 | System.out.println(RecordVisitor.clearNumber("123d ", Boolean.TRUE));
11 | System.out.println(RecordVisitor.clearNumber("1,123 ", Boolean.TRUE));
12 | System.out.println(RecordVisitor.clearNumber("111,1,11,,123.00 ", Boolean.FALSE));
13 | System.out.println(RecordVisitor.clearNumber("NaN", Boolean.TRUE));
14 | }
15 | }
--------------------------------------------------------------------------------
/src/test/java/cn/devezhao/persist4j/util/CaseInsensitiveMapTest.java:
--------------------------------------------------------------------------------
1 | package cn.devezhao.persist4j.util;
2 |
3 | import org.junit.Test;
4 |
5 | import java.util.Map;
6 |
7 | import static org.junit.Assert.assertEquals;
8 | import static org.junit.Assert.assertFalse;
9 |
10 | /**
11 | * @author zhaofang123@gmail.com
12 | * @since 12/21/2018
13 | */
14 | public class CaseInsensitiveMapTest {
15 |
16 | @Test
17 | public void testMap() throws Exception {
18 | Map map = new CaseInsensitiveMap<>();
19 |
20 | map.put("abc", "1");
21 | map.put("Abc", "11");
22 | map.put("AbC", "111");
23 | map.put("abcd", "2");
24 | map.remove("ABCd");
25 |
26 | System.out.println(map.keySet());
27 | System.out.println(map.entrySet());
28 |
29 | assertEquals(map.entrySet().size(), map.size());
30 | assertFalse(map.containsKey("abcd"));
31 | assertEquals(map.get("abc"), "111");
32 | }
33 | }
34 |
--------------------------------------------------------------------------------
/src/test/java/cn/devezhao/persist4j/util/StringHelperTest.java:
--------------------------------------------------------------------------------
1 | package cn.devezhao.persist4j.util;
2 |
3 | import cn.hutool.core.util.RandomUtil;
4 | import junit.framework.TestCase;
5 |
6 | public class StringHelperTest extends TestCase {
7 |
8 | public void testHyphenate() {
9 | System.out.println(StringHelper.hyphenate("TEST999"));
10 | System.out.println(StringHelper.hyphenate("TEST999Id"));
11 | System.out.println(StringHelper.hyphenate("TEST999Name"));
12 | System.out.println(StringHelper.hyphenate("ishttp_Req"));
13 | System.out.println(StringHelper.hyphenate("ishttp_req"));
14 | System.out.println(StringHelper.hyphenate("ISHTTP_REQ"));
15 | System.out.println(StringHelper.hyphenate("Bp0B2"));
16 | System.out.println(StringHelper.hyphenate("IshttpReq"));
17 | System.out.println(StringHelper.hyphenate("isHttpReq"));
18 | System.out.println(StringHelper.hyphenate("isHTTPReq"));
19 | System.out.println(StringHelper.hyphenate("ISHTTPREQ"));
20 | System.out.println(StringHelper.hyphenate("_123A"));
21 |
22 | for (int i = 0; i < 100; i++) {
23 | String rnd = RandomUtil.randomString(20);
24 | String h1 = StringHelper.hyphenate(rnd).toLowerCase();
25 | System.out.println(rnd + " > " + h1);
26 | }
27 | }
28 | }
--------------------------------------------------------------------------------
/src/test/java/cn/devezhao/persist4j/util/support/SchemaExportTest.java:
--------------------------------------------------------------------------------
1 | package cn.devezhao.persist4j.util.support;
2 |
3 | import cn.devezhao.persist4j.BaseTest;
4 | import cn.devezhao.persist4j.Entity;
5 | import cn.devezhao.persist4j.PersistManagerFactory;
6 | import cn.devezhao.persist4j.metadata.impl.ConfigurationMetadataFactory;
7 | import org.apache.commons.lang.StringUtils;
8 | import org.dom4j.Element;
9 | import org.junit.Test;
10 |
11 | import java.util.List;
12 |
13 | /**
14 | *
15 | * @author zhaofang123@gmail.com
16 | * @since 02/27/2017
17 | */
18 | public class SchemaExportTest extends BaseTest {
19 |
20 | @Test
21 | public void testExportDDL() throws Exception {
22 | PersistManagerFactory factory = getPersistManagerFactory();
23 |
24 | SchemaExport schemaExport = new SchemaExport(
25 | factory.getMetadataFactory(), factory.getDialect(), factory.getDataSource().getConnection());
26 | schemaExport.export(true, false);
27 | }
28 |
29 | @Test
30 | public void testGenDDL() throws Exception {
31 | PersistManagerFactory factory = getPersistManagerFactory();
32 |
33 | Element cfgRoot = ((ConfigurationMetadataFactory) factory.getMetadataFactory()).getConfigDocument().getRootElement();
34 | Entity entity = factory.getMetadataFactory().getEntity("TestAllType");
35 | List ix = cfgRoot.selectSingleNode("//entity[@name='" + entity.getName() + "']").selectNodes("index");
36 |
37 | String[] sqls = new Table(entity, factory.getDialect(), ix).generateDDL(false, false);
38 | System.out.println(StringUtils.join(sqls, "\n"));
39 | }
40 | }
41 |
--------------------------------------------------------------------------------
/src/test/resources/metadata-test.xml:
--------------------------------------------------------------------------------
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 |
--------------------------------------------------------------------------------