honeymoose/hapi-fhir
1
0
Fork 0
mirror of https://github.com/hapifhir/hapi-fhir.git synced 2025-02-09 06:24:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

Expand access to SqlQuery

Browse Source
This commit is contained in:
jamesagnew 2021-07-08 16:16:08 -04:00
parent ee1f1fb5b8
commit 6d4c2ef8f0
1 changed files with 30 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
hapi-fhir-jpaserver-base/src/main/java/ca/uhn/fhir/jpa/util/SqlQuery.java
View File

@ -21,6 +21,7 @@ package ca.uhn.fhir.jpa.util;
*/
import ca.uhn.fhir.util.UrlUtil;
import org.apache.commons.lang3.Validate;
import org.hibernate.engine.jdbc.internal.BasicFormatterImpl;
import java.util.ArrayList;
@ -37,14 +38,22 @@ public class SqlQuery {
private final long myElapsedTime;
private final StackTraceElement[] myStackTrace;
private final int mySize;
private final LanguageEnum myLanguage;
public SqlQuery(String theSql, List<String> theParams, long theQueryTimestamp, long theElapsedTime, StackTraceElement[] theStackTraceElements, int theSize) {
this(theSql, theParams, theQueryTimestamp, theElapsedTime, theStackTraceElements, theSize, LanguageEnum.SQL);
}
public SqlQuery(String theSql, List<String> theParams, long theQueryTimestamp, long theElapsedTime, StackTraceElement[] theStackTraceElements, int theSize, LanguageEnum theLanguage) {
Validate.notNull(theLanguage, "theLanguage must not be null");
SqlQuery(String theSql, List<String> theParams, long theQueryTimestamp, long theElapsedTime, StackTraceElement[] theStackTraceElements, int theSize) {
mySql = theSql;
myParams = Collections.unmodifiableList(theParams);
myQueryTimestamp = theQueryTimestamp;
myElapsedTime = theElapsedTime;
myStackTrace = theStackTraceElements;
mySize = theSize;
myLanguage = theLanguage;
}
public long getQueryTimestamp() {
@ -63,9 +72,14 @@ public class SqlQuery {
return getSql(theInlineParams, theFormat, false);
}
public LanguageEnum getLanguage() {
return myLanguage;
}
public String getSql(boolean theInlineParams, boolean theFormat, boolean theSanitizeParams) {
String retVal = mySql;
if (theFormat) {
if (getLanguage() == LanguageEnum.SQL) {
retVal = new BasicFormatterImpl().format(retVal);
// BasicFormatterImpl annoyingly adds a newline at the very start of its output
@ -73,6 +87,7 @@ public class SqlQuery {
retVal = retVal.substring(1);
}
}
}
if (theInlineParams) {
List<String> nextParams = new ArrayList<>(myParams);
@ -102,4 +117,14 @@ public class SqlQuery {
public int getSize() {
return mySize;
}
public enum LanguageEnum {
SQL,
JSON
}
}