de.mhus.db.karaf.xdb.cmd.CmdXRawSelect Maven / Gradle / Ivy
/**
* Copyright 2018 Mike Hummel
*
* Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file
* except in compliance with the License. You may obtain a copy of the License at
*
*
http://www.apache.org/licenses/LICENSE-2.0
*
*
Unless required by applicable law or agreed to in writing, software distributed under the
* License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
* express or implied. See the License for the specific language governing permissions and
* limitations under the License.
*/
package de.mhus.db.karaf.xdb.cmd;
import java.util.Comparator;
import java.util.HashMap;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import org.apache.karaf.shell.api.action.Argument;
import org.apache.karaf.shell.api.action.Command;
import org.apache.karaf.shell.api.action.Option;
import org.apache.karaf.shell.api.action.lifecycle.Reference;
import org.apache.karaf.shell.api.action.lifecycle.Service;
import org.apache.karaf.shell.api.console.Session;
import de.mhus.db.karaf.xdb.adb.XdbKarafUtil;
import de.mhus.lib.adb.DbCollection;
import de.mhus.lib.core.M;
import de.mhus.lib.core.MCast;
import de.mhus.lib.core.MString;
import de.mhus.lib.core.console.ConsoleTable;
import de.mhus.lib.core.matcher.Condition;
import de.mhus.lib.errors.MException;
import de.mhus.lib.xdb.XdbType;
import de.mhus.osgi.api.karaf.AbstractCmd;
@Command(
scope = "xdb",
name = "rawselect",
description = "Select data from DB DataSource and print the results")
// @Parsing(XdbParser.class) see
// https://github.com/apache/karaf/tree/master/jdbc/src/main/java/org/apache/karaf/jdbc/command/parsing
@Service
public class CmdXRawSelect extends AbstractCmd {
@Argument(
index = 0,
name = "type",
required = true,
description = "Type to select",
multiValued = false)
String typeName;
@Argument(
index = 1,
name = "qualification",
required = false,
description = "Select qualification",
multiValued = false)
String qualification;
@Option(
name = "-l",
aliases = "--oneline",
description = "Disable multi line output in table cells",
required = false)
boolean oneLine = false;
@Option(
name = "-f",
aliases = "--filter",
description = "Additional filters after loading or results",
required = false)
String filter;
@Option(
name = "-m",
aliases = "--max",
description = "Maximum amount of chars for a value (if not full)",
required = false)
int max = 40;
@Option(
name = "-o",
aliases = "--out",
description = "Comma separated list of fields to print",
required = false)
String fieldsComma = null;
@Option(name = "-x", description = "Output parameter", required = false)
String outputParam = null;
@Option(name = "-a", description = "Api Name", required = false)
String apiName = null;
@Option(name = "-s", description = "Service Name", required = false)
String serviceName = null;
@Option(name = "-v", aliases = "--csv", description = "CSV Style", required = false)
boolean csv = false;
@Option(
name = "-n",
aliases = "--lines",
description =
"Number of lines f (first n lines) or l (last n lines) or p[,]",
required = false)
String page = null;
@Option(
name = "-p",
aliases = "--parameter",
description = "Define a parameter key=value",
required = false,
multiValued = true)
String[] parameters = null;
@Option(name = "-q", description = "xdb query parser", required = false)
boolean xdbQuery = false;
@Reference private Session session;
private Condition condition;
@Override
public Object execute2() throws Exception {
Object output = null;
if (MString.isSet(filter)) condition = new Condition(filter);
apiName = XdbKarafUtil.getApiName(session, apiName);
serviceName = XdbKarafUtil.getServiceName(session, serviceName);
XdbType type = XdbKarafUtil.getType(apiName, serviceName, typeName);
// sort columns to print
final LinkedList fieldNames = new LinkedList<>();
if (fieldsComma == null) {
for (String name : type.getAttributeNames()) {
fieldNames.add(name);
}
fieldNames.sort(
new Comparator() {
@Override
public int compare(String o1, String o2) {
boolean pk1 = type.isPrimaryKey(o1);
boolean pk2 = type.isPrimaryKey(o2);
if (pk1 == pk2) return o1.compareTo(o2);
if (pk1) return -1;
// if (pk2) return 1;
return 1;
}
});
} else {
for (String name : fieldsComma.split(",")) fieldNames.add(name);
}
ConsoleTable out = new ConsoleTable(tblOpt);
if (csv) {
out.setColSeparator(";");
out.setCellSpacer(false);
}
if (oneLine) out.setMultiLine(false);
// if (!full)
// out.setMaxColSize(max);
for (String name : fieldNames) {
if (type.isPrimaryKey(name)) name = name + "*";
out.addHeader(name);
}
HashMap queryParam = null;
if (parameters != null) {
queryParam = new HashMap<>();
for (String p : parameters) {
String k = MString.beforeIndex(p, '=');
String v = MString.afterIndex(p, '=');
queryParam.put(k, v);
}
}
// if (xdbQuery) {
// AQuery query = Db.parse(type, qualification);
// }
if (page == null) {
for (Object object : type.getByQualification(qualification, queryParam)) {
if (skipResult(type, object)) continue;
ConsoleTable.Row row = out.addRow();
for (String name : fieldNames) {
Object value = getValueValue(type, object, name);
row.add(value);
}
output = object;
}
} else if (page.startsWith("f")) {
int lines = MCast.toint(page.substring(1), 100);
DbCollection res = type.getByQualification(qualification, null);
for (Object object : res) {
if (skipResult(type, object)) continue;
ConsoleTable.Row row = out.addRow();
for (String name : fieldNames) {
Object value = getValueValue(type, object, name);
row.add(value);
}
output = object;
lines--;
if (lines <= 0) {
res.close();
break;
}
}
} else if (page.startsWith("l")) {
int lines = MCast.toint(page.substring(1), 100);
for (Object object : type.getByQualification(qualification, null)) {
if (skipResult(type, object)) continue;
ConsoleTable.Row row = out.addRow();
for (String name : fieldNames) {
Object value = getValueValue(type, object, name);
row.add(value);
}
output = object;
if (out.size() > lines) out.removeFirstRow();
}
} else if (page.startsWith("p")) {
int lines = 100;
int p = 0;
if (MString.isIndex(page, ',')) {
lines = MCast.toint(MString.beforeIndex(page, ','), lines);
p = MCast.toint(MString.afterIndex(page, ','), p);
} else {
p = MCast.toint(page, p);
}
System.out.println("Page size: " + lines + ", Page: " + p);
DbCollection res = type.getByQualification(qualification, null);
int cnt = 0;
Iterator iter = res.iterator();
while (iter.hasNext()) {
iter.next();
cnt++;
if (cnt >= p * lines) break;
}
while (iter.hasNext()) {
Object object = iter.next();
if (skipResult(type, object)) continue;
ConsoleTable.Row row = out.addRow();
for (String name : fieldNames) {
Object value = getValueValue(type, object, name);
row.add(value);
}
output = object;
lines--;
if (lines <= 0) {
res.close();
break;
}
}
}
out.print(System.out);
if (outputParam != null) session.put(outputParam, output);
return null;
}
private boolean skipResult(XdbType type, Object object) throws MException {
if (condition == null) return false;
return condition.matches(new ConditionMap(type, object));
}
private class ConditionMap extends HashMap {
private static final long serialVersionUID = 1L;
private Object object;
private XdbType type;
ConditionMap(XdbType type, Object object) {
this.type = type;
this.object = object;
}
@Override
public Object get(Object key) {
try {
return getValueValue(type, object, String.valueOf(key));
} catch (MException e) {
return null;
}
}
}
@SuppressWarnings("rawtypes")
private Object getValueValue(XdbType type, Object object, String name) throws MException {
int pos = name.indexOf('.');
Object value = null;
if (pos < 0) {
value = type.get(object, name);
} else {
String key = name.substring(pos + 1);
name = name.substring(0, pos);
value = type.get(object, name);
if (value == null) {
// nothing
} else if (value instanceof List) {
int idx = M.to(key, 0);
List c = (List) value;
if (idx < c.size()) value = c.get(idx);
else value = null;
} else if (value.getClass().isArray()) {
int idx = M.to(key, 0);
Object[] a = (Object[]) value;
if (idx < a.length) value = a[idx];
else a = null;
} else if (value instanceof Map) {
Map m = (Map) value;
value = m.get(key);
}
}
if (value == null) return "[null]";
return value;
}
}