jhannes / fluent-jdbc Goto Github PK

static class Field {
	String name;
	int start;
	int end;

	Field(String name, int start, int end) {
		this.name = name;
		this.start = start;
		this.end = end;
	}

	@Override
	public String toString() {
		return getClass().getSimpleName() + "<" + name + "," + start + "," + end + ">";
	}
}

private static final Pattern FIELD_REGEXP = Pattern.compile("(?:'[^']+')|:([a-zA-Z0-9_]+)");

private final String statement;

/** order of fields in statement. */
private final List<Field> fieldOrder = new ArrayList<>();
private final Map<String, List<?>> fieldParams = new LinkedHashMap<>();

public DbStatement(String stmt) {
	this.statement = Objects.requireNonNull(stmt);
	Matcher m = FIELD_REGEXP.matcher(stmt);
	while (m.find()) {
		if (m.group(1) != null) {
			fieldOrder.add(new Field(m.group(1), m.start(), m.end()));
		}
	}
}

void executeQuery() {
	Set<String> fields = fieldOrder.stream().map(f -> f.name).collect(Collectors.toSet());
	if (!fieldParams.keySet().equals(fields)) {
		throw new IllegalStateException("Forventet : " + fields + ", men har " + fieldParams.keySet());
	}
	StringBuilder sql = new StringBuilder();

	int last = 0;
	for (Field f : fieldOrder) {
		sql.append(statement.subSequence(last, f.start));
		String[] args = new String[fieldParams.get(f.name).size()];
		Arrays.fill(args, "?");
		String qs = String.join(",", Arrays.asList(args));
		sql = sql.append(qs);
		last = f.end;
	}
	sql.append(statement.substring(last));

	int i = 1;
	for (Field f : fieldOrder) {
		for (Object v : fieldParams.get(f.name)) {
			// TODO: populate statement
			System.out.println(i++ + ": " + f.name + " =" + v);
		}
	}

	// TODO: execute
	System.out.println(sql);

}

public void setField(String fieldName, Object val) {
	this.fieldParams.put(fieldName, List.of(val));
}

public void setField(String fieldName, List<Object> vals) {
	fieldParams.put(fieldName, vals);
}

List<Field> getFieldOrder() {
	return Collections.unmodifiableList(fieldOrder);
}

Map<String, List<?>> getFieldParams() {
	return Collections.unmodifiableMap(fieldParams);
}

	System.out.println(stmt.getFieldOrder());

	stmt.setField("myField1", 999);
	stmt.setField("myField_2", List.of("A", "B", "C"));

	stmt.executeQuery();
}