use of com.rdbcache.helpers.Context in project rdbcache by rdbcache.
the class ParserTest method prepareStandardClauseParams.
@Test
public void prepareStandardClauseParams() {
InputStream inputStream = this.getClass().getClassLoader().getResourceAsStream("test-table.json");
assertNotNull(inputStream);
String text = null;
try (final Reader reader = new InputStreamReader(inputStream)) {
text = CharStreams.toString(reader);
} catch (Exception e) {
e.printStackTrace();
fail(e.getCause().getMessage());
}
assertNotNull(text);
Map<String, Object> testTable = Utils.toMap(text);
assertNotNull(testTable);
DbaseOps dbaseOps = new DbaseOps();
AppCtx.setDbaseOps(dbaseOps);
Context context = new Context();
String json = "{\n" + " \"id\" : 12467,\n" + " \"email\" : \"kevin@example.com\",\n" + " \"name\" : \"Kevin B.\",\n" + " \"dob\" : \"1980-07-21\"\n" + " }";
KvPair pair = new KvPair("*", "data", Utils.toMap(json));
try {
KeyInfo keyInfo = new KeyInfo();
keyInfo.setExpire("100");
keyInfo.setTable("user_table");
keyInfo.setCreatedAt(1522367710621L);
String json2 = "{\"table\":\"user_table\",\"conditions\":{\"id\":{\"=\":[\"1\",\"2\",\"3\"]}},\"limit\":2}";
QueryInfo queryInfo = Utils.toPojo(Utils.toMap(json2), QueryInfo.class);
keyInfo.setQuery(queryInfo);
keyInfo.setColumns((Map<String, Object>) testTable.get("table_columns::user_table"));
keyInfo.setPrimaryIndexes(Arrays.asList("id"));
Parser.prepareStandardClauseParams(context, pair, keyInfo);
// System.out.println(Utils.toJsonMap(keyInfo));
assertEquals("{\"expire\":\"100\",\"table\":\"user_table\",\"clause\":\"id = ?\"," + "\"params\":[12467],\"query\":{\"table\":\"user_table\",\"conditions\":{\"id\":{\"=\":" + "[\"1\",\"2\",\"3\"]}},\"limit\":2},\"query_key\":\"87677684c30a46c6e5afec88d0131410\"," + "\"is_new\":false,\"expire_old\":\"180\",\"created_at\":1522367710621}", Utils.toJsonMap(keyInfo));
keyInfo.cleanup();
// System.out.println(Utils.toJsonMap(keyInfo));
assertEquals("{\"expire\":\"100\",\"table\":\"user_table\",\"clause\":\"id = ?\",\"params\":[12467]" + ",\"query_key\":\"87677684c30a46c6e5afec88d0131410\",\"is_new\":false,\"created_at\":1522367710621}", Utils.toJsonMap(keyInfo));
} catch (Exception e) {
e.printStackTrace();
fail(e.getCause().getMessage());
}
}
use of com.rdbcache.helpers.Context in project rdbcache by rdbcache.
the class QueryTest method updateTest.
@Test
public void updateTest() {
try {
Context context = new Context();
KvPairs pairs = new KvPairs();
AnyKey anyKey = new AnyKey();
KeyInfo keyInfo = new KeyInfo();
keyInfo.setExpire("100");
keyInfo.setTable("user_table");
String json1 = "{\"table\":\"user_table\",\"conditions\":{\"email\":{\"=\":[\"david@example.com\"]}}}";
QueryInfo queryInfo1 = Utils.toPojo(Utils.toMap(json1), QueryInfo.class);
keyInfo.setQuery(queryInfo1);
anyKey.setKeyInfo(keyInfo);
Query query = new Query(context, jdbcTemplate, pairs, anyKey);
assertTrue(query.ifSelectOk());
assertTrue(query.executeSelect());
// System.out.println(Utils.toJsonMap(pairs));
assertEquals(1, pairs.size());
String key = pairs.getPair().getId();
Integer id = (Integer) pairs.getPair().getData().get("id");
assertNotNull(id);
// System.out.println("id = "+id);
String json2 = "{\"name\" : \"David Copper\"}";
Map<String, Object> map1 = Utils.toMap(json2);
KvPair pair = new KvPair(key, map1);
pairs.setPair(pair);
KeyInfo keyInfo2 = new KeyInfo();
keyInfo2.setExpire("100");
keyInfo2.setTable("user_table");
keyInfo2.setClause("id = ?");
keyInfo2.setParams(Arrays.asList(id));
anyKey.setKeyInfo(keyInfo2);
query = new Query(context, jdbcTemplate, pairs, anyKey);
assertTrue(query.ifUpdateOk());
assertTrue(query.executeUpdate());
pairs.clear();
query = new Query(context, jdbcTemplate, pairs, anyKey);
assertTrue(query.ifSelectOk());
assertTrue(query.executeSelect());
map1 = pairs.getPair().getData();
assertEquals("David Copper", map1.get("name"));
} catch (Exception e) {
e.printStackTrace();
fail(e.getCause().getMessage());
}
}
use of com.rdbcache.helpers.Context in project rdbcache by rdbcache.
the class QueryTest method selectTest.
@Test
public void selectTest() {
try {
Context context = new Context();
KvPairs pairs = new KvPairs();
AnyKey anyKey = new AnyKey();
KeyInfo keyInfo = new KeyInfo();
keyInfo.setExpire("100");
keyInfo.setTable("user_table");
String json = "{\"table\":\"user_table\",\"conditions\":{\"id\":{\">\":[\"1\"]}},\"limit\":2}";
QueryInfo queryInfo = Utils.toPojo(Utils.toMap(json), QueryInfo.class);
keyInfo.setQuery(queryInfo);
anyKey.setKeyInfo(keyInfo);
Query query = new Query(context, jdbcTemplate, pairs, anyKey);
assertTrue(query.ifSelectOk());
assertTrue(query.executeSelect());
// System.out.println(Utils.toJsonMap(pairs));
assertEquals(2, pairs.size());
} catch (Exception e) {
e.printStackTrace();
fail(e.getCause().getMessage());
}
}
use of com.rdbcache.helpers.Context in project rdbcache by rdbcache.
the class TaskQueue method onReceiveTask.
public void onReceiveTask(String task) {
LOGGER.debug("Received Task: " + task);
String[] parts = task.split("::");
if (parts.length < 3) {
LOGGER.error("invalid task format");
return;
}
String action = parts[0];
String key = parts[1];
String traceId = parts[2];
Context context = new Context(traceId);
if (enableMonitor)
context.enableMonitor(task, "queue", action);
KvPair pair = new KvPair(key);
KvPairs pairs = new KvPairs(pair);
AnyKey anyKey = new AnyKey();
if (!AppCtx.getKeyInfoRepo().find(context, pairs, anyKey)) {
String msg = "keyInfo not found";
LOGGER.error(msg);
context.logTraceMessage(msg);
return;
}
KeyInfo keyInfo = anyKey.getKeyInfo();
// ...
String msg = "unknown task action:" + action;
LOGGER.error(msg);
context.logTraceMessage(msg);
context.closeMonitor();
}
use of com.rdbcache.helpers.Context in project rdbcache by rdbcache.
the class ParserTest method prepareQueryClauseParams.
@Test
public void prepareQueryClauseParams() {
Context context = new Context();
KvPair pair = new KvPair("hash_key");
KeyInfo keyInfo = new KeyInfo();
keyInfo.setExpire("100");
keyInfo.setTable("user_table");
String json = "{\"table\":\"user_table\",\"conditions\":{\"id\":{\"=\":[\"1\",\"2\",\"3\"]}},\"limit\":2}";
Map<String, Object> map = Utils.toMap(json);
QueryInfo queryInfo = Utils.toPojo(map, QueryInfo.class);
keyInfo.setQuery(queryInfo);
Parser.prepareQueryClauseParams(context, pair, keyInfo);
assertEquals("(id = ? OR id = ? OR id = ?)", keyInfo.getClause());
assertEquals("[1, 2, 3]", keyInfo.getParams().toString());
}
Aggregations