|
1 | 1 | package org.tarantool;
|
2 | 2 |
|
| 3 | +import static org.tarantool.AbstractTarantoolOps.ResolvableArgument.ofResolved; |
| 4 | +import static org.tarantool.AbstractTarantoolOps.ResolvableArgument.ofUnresolved; |
3 | 5 |
|
4 |
| -public abstract class AbstractTarantoolOps<Space, Tuple, Operation, Result> |
5 |
| - implements TarantoolClientOps<Space, Tuple, Operation, Result> { |
| 6 | +import java.util.Objects; |
| 7 | +import java.util.function.Function; |
| 8 | +import java.util.stream.Stream; |
| 9 | + |
| 10 | +public abstract class AbstractTarantoolOps<Tuple, Operation, Result> |
| 11 | + implements TarantoolClientOps<Tuple, Operation, Result> { |
| 12 | + |
| 13 | + private final Function<Object, Object> defaultSpaceResolver = space -> resolveSpace((String) space); |
6 | 14 |
|
7 | 15 | private Code callCode = Code.OLD_CALL;
|
8 | 16 |
|
9 |
| - protected abstract Result exec(Code code, Object... args); |
| 17 | + protected Result exec(Code code, Object... args) { |
| 18 | + return exec(code, ResolvableArgument.ofAllResolved(args)); |
| 19 | + } |
| 20 | + |
| 21 | + protected abstract Result exec(Code code, ResolvableArgument... args); |
| 22 | + |
| 23 | + protected abstract int resolveSpace(String space); |
| 24 | + |
| 25 | + protected abstract int resolveSpaceIndex(String space, String index); |
| 26 | + |
| 27 | + protected static class ResolvableArgument { |
| 28 | + |
| 29 | + private final Object value; |
| 30 | + private final Function<Object, Object> resolver; |
| 31 | + |
| 32 | + private ResolvableArgument(Object value, Function<Object, Object> resolver) { |
| 33 | + Objects.requireNonNull(value); |
| 34 | + this.value = value; |
| 35 | + this.resolver = resolver; |
| 36 | + } |
| 37 | + |
| 38 | + public static ResolvableArgument ofResolved(Object resolvedValue) { |
| 39 | + return new ResolvableArgument(resolvedValue, null); |
| 40 | + } |
| 41 | + |
| 42 | + public static ResolvableArgument[] ofAllResolved(Object[] resolvedValue) { |
| 43 | + return Stream.of(resolvedValue).map(ResolvableArgument::ofResolved).toArray(ResolvableArgument[]::new); |
| 44 | + } |
| 45 | + |
| 46 | + public static ResolvableArgument ofUnresolved(Object unresolvedValue, Function<Object, Object> resolver) { |
| 47 | + return new ResolvableArgument(unresolvedValue, resolver); |
| 48 | + } |
| 49 | + |
| 50 | + public boolean isUnresolved() { |
| 51 | + try { |
| 52 | + return resolver != null && resolver.apply(value) == null; |
| 53 | + } catch (Exception ignored) { |
| 54 | + return true; |
| 55 | + } |
| 56 | + } |
| 57 | + |
| 58 | + public Object getValue() { |
| 59 | + return resolver != null ? resolver.apply(value) : value; |
| 60 | + } |
10 | 61 |
|
11 |
| - public Result select(Space space, Space index, Tuple key, int offset, int limit, Iterator iterator) { |
| 62 | + @Override |
| 63 | + public String toString() { |
| 64 | + if (isUnresolved()) { |
| 65 | + return "{unresolved: " + value + "}"; |
| 66 | + } |
| 67 | + return value.toString(); |
| 68 | + } |
| 69 | + |
| 70 | + } |
| 71 | + |
| 72 | + @Override |
| 73 | + public Result select(Integer space, Integer index, Tuple key, int offset, int limit, Iterator iterator) { |
12 | 74 | return select(space, index, key, offset, limit, iterator.getValue());
|
13 | 75 | }
|
14 | 76 |
|
15 |
| - public Result select(Space space, Space index, Tuple key, int offset, int limit, int iterator) { |
16 |
| - return exec( |
17 |
| - Code.SELECT, |
18 |
| - Key.SPACE, space, |
19 |
| - Key.INDEX, index, |
20 |
| - Key.KEY, key, |
21 |
| - Key.ITERATOR, iterator, |
22 |
| - Key.LIMIT, limit, |
23 |
| - Key.OFFSET, offset |
| 77 | + @Override |
| 78 | + public Result select(String space, String index, Tuple key, int offset, int limit, Iterator iterator) { |
| 79 | + return select(space, index, key, offset, limit, iterator.getValue()); |
| 80 | + } |
| 81 | + |
| 82 | + @Override |
| 83 | + public Result select(Integer space, Integer index, Tuple key, int offset, int limit, int iterator) { |
| 84 | + return doSelect(ofResolved(space), ofResolved(index), key, offset, limit, iterator); |
| 85 | + } |
| 86 | + |
| 87 | + @Override |
| 88 | + public Result select(String space, String index, Tuple key, int offset, int limit, int iterator) { |
| 89 | + return doSelect( |
| 90 | + ofUnresolved(space, defaultSpaceResolver), |
| 91 | + ofUnresolved(index, indexName -> resolveSpaceIndex(space, (String) indexName)), |
| 92 | + key, offset, limit, iterator |
24 | 93 | );
|
25 | 94 | }
|
26 | 95 |
|
27 |
| - public Result insert(Space space, Tuple tuple) { |
28 |
| - return exec(Code.INSERT, Key.SPACE, space, Key.TUPLE, tuple); |
| 96 | + @Override |
| 97 | + public Result insert(Integer space, Tuple tuple) { |
| 98 | + return doInsert(ofResolved(space), tuple); |
| 99 | + } |
| 100 | + |
| 101 | + @Override |
| 102 | + public Result insert(String space, Tuple tuple) { |
| 103 | + return doInsert(ofUnresolved(space, defaultSpaceResolver), tuple); |
| 104 | + } |
| 105 | + |
| 106 | + @Override |
| 107 | + public Result replace(Integer space, Tuple tuple) { |
| 108 | + return doReplace(ofResolved(space), tuple); |
| 109 | + } |
| 110 | + |
| 111 | + @Override |
| 112 | + public Result replace(String space, Tuple tuple) { |
| 113 | + return doReplace(ofUnresolved(space, defaultSpaceResolver), tuple); |
| 114 | + } |
| 115 | + |
| 116 | + @Override |
| 117 | + public Result update(Integer space, Tuple key, Operation... args) { |
| 118 | + return doUpdate(ofResolved(space), key, args); |
| 119 | + } |
| 120 | + |
| 121 | + @Override |
| 122 | + public Result update(String space, Tuple key, Operation... tuple) { |
| 123 | + return doUpdate(ofUnresolved(space, defaultSpaceResolver), key, tuple); |
29 | 124 | }
|
30 | 125 |
|
31 |
| - public Result replace(Space space, Tuple tuple) { |
32 |
| - return exec(Code.REPLACE, Key.SPACE, space, Key.TUPLE, tuple); |
| 126 | + @Override |
| 127 | + public Result upsert(Integer space, Tuple key, Tuple defTuple, Operation... ops) { |
| 128 | + return doUpsert(ofResolved(space), key, defTuple, ops); |
33 | 129 | }
|
34 | 130 |
|
35 |
| - public Result update(Space space, Tuple key, Operation... args) { |
36 |
| - return exec(Code.UPDATE, Key.SPACE, space, Key.KEY, key, Key.TUPLE, args); |
| 131 | + @Override |
| 132 | + public Result upsert(String space, Tuple key, Tuple defTuple, Operation... ops) { |
| 133 | + return doUpsert(ofUnresolved(space, defaultSpaceResolver), key, defTuple, ops); |
37 | 134 | }
|
38 | 135 |
|
39 |
| - public Result upsert(Space space, Tuple key, Tuple def, Operation... args) { |
40 |
| - return exec(Code.UPSERT, Key.SPACE, space, Key.KEY, key, Key.TUPLE, def, Key.UPSERT_OPS, args); |
| 136 | + @Override |
| 137 | + public Result delete(Integer space, Tuple key) { |
| 138 | + return doDelete(ofResolved(space), key); |
41 | 139 | }
|
42 | 140 |
|
43 |
| - public Result delete(Space space, Tuple key) { |
44 |
| - return exec(Code.DELETE, Key.SPACE, space, Key.KEY, key); |
| 141 | + @Override |
| 142 | + public Result delete(String space, Tuple key) { |
| 143 | + return doDelete(ofUnresolved(space, defaultSpaceResolver), key); |
45 | 144 | }
|
46 | 145 |
|
| 146 | + @Override |
47 | 147 | public Result call(String function, Object... args) {
|
48 | 148 | return exec(callCode, Key.FUNCTION, function, Key.TUPLE, args);
|
49 | 149 | }
|
50 | 150 |
|
| 151 | + @Override |
51 | 152 | public Result eval(String expression, Object... args) {
|
52 | 153 | return exec(Code.EVAL, Key.EXPRESSION, expression, Key.TUPLE, args);
|
53 | 154 | }
|
54 | 155 |
|
| 156 | + @Override |
55 | 157 | public void ping() {
|
56 | 158 | exec(Code.PING);
|
57 | 159 | }
|
58 | 160 |
|
59 | 161 | public void setCallCode(Code callCode) {
|
60 | 162 | this.callCode = callCode;
|
61 | 163 | }
|
| 164 | + |
| 165 | + private Result doDelete(ResolvableArgument space, Tuple key) { |
| 166 | + return exec(Code.DELETE, ofResolved(Key.SPACE), space, ofResolved(Key.KEY), ofResolved(key)); |
| 167 | + } |
| 168 | + |
| 169 | + private Result doUpsert(ResolvableArgument space, Tuple key, Tuple defTuple, Operation... ops) { |
| 170 | + return exec( |
| 171 | + Code.UPSERT, |
| 172 | + ofResolved(Key.SPACE), space, |
| 173 | + ofResolved(Key.KEY), ofResolved(key), |
| 174 | + ofResolved(Key.TUPLE), ofResolved(defTuple), |
| 175 | + ofResolved(Key.UPSERT_OPS), ofResolved(ops) |
| 176 | + ); |
| 177 | + } |
| 178 | + |
| 179 | + private Result doUpdate(ResolvableArgument space, Tuple key, Operation... ops) { |
| 180 | + return exec( |
| 181 | + Code.UPDATE, |
| 182 | + ofResolved(Key.SPACE), space, |
| 183 | + ofResolved(Key.KEY), ofResolved(key), |
| 184 | + ofResolved(Key.TUPLE), ofResolved(ops) |
| 185 | + ); |
| 186 | + } |
| 187 | + |
| 188 | + private Result doReplace(ResolvableArgument space, Tuple tuple) { |
| 189 | + return exec(Code.REPLACE, ofResolved(Key.SPACE), space, ofResolved(Key.TUPLE), ofResolved(tuple)); |
| 190 | + } |
| 191 | + |
| 192 | + private Result doInsert(ResolvableArgument space, Tuple tuple) { |
| 193 | + return exec(Code.INSERT, ofResolved(Key.SPACE), space, ofResolved(Key.TUPLE), ofResolved(tuple)); |
| 194 | + } |
| 195 | + |
| 196 | + private Result doSelect(ResolvableArgument space, |
| 197 | + ResolvableArgument index, |
| 198 | + Tuple key, |
| 199 | + int offset, |
| 200 | + int limit, |
| 201 | + int iterator) { |
| 202 | + return exec( |
| 203 | + Code.SELECT, |
| 204 | + ofResolved(Key.SPACE), space, |
| 205 | + ofResolved(Key.INDEX), index, |
| 206 | + ofResolved(Key.KEY), ofResolved(key), |
| 207 | + ofResolved(Key.ITERATOR), ofResolved(iterator), |
| 208 | + ofResolved(Key.LIMIT), ofResolved(limit), |
| 209 | + ofResolved(Key.OFFSET), ofResolved(offset) |
| 210 | + ); |
| 211 | + } |
| 212 | + |
62 | 213 | }
|
0 commit comments