BaseEvalContext.java
/*
Copyright (c) 2018 James Ahlborn
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 com.healthmarketscience.jackcess.impl;
import java.io.IOException;
import java.math.BigDecimal;
import java.text.DecimalFormat;
import java.time.LocalDateTime;
import java.time.ZoneId;
import java.time.format.DateTimeFormatter;
import java.util.Collection;
import java.util.EnumMap;
import java.util.Map;
import javax.script.Bindings;
import com.healthmarketscience.jackcess.DataType;
import com.healthmarketscience.jackcess.JackcessException;
import com.healthmarketscience.jackcess.expr.EvalContext;
import com.healthmarketscience.jackcess.expr.EvalException;
import com.healthmarketscience.jackcess.expr.Expression;
import com.healthmarketscience.jackcess.expr.Identifier;
import com.healthmarketscience.jackcess.expr.LocaleContext;
import com.healthmarketscience.jackcess.expr.NumericConfig;
import com.healthmarketscience.jackcess.expr.TemporalConfig;
import com.healthmarketscience.jackcess.expr.Value;
import com.healthmarketscience.jackcess.impl.expr.Expressionator;
import com.healthmarketscience.jackcess.impl.expr.ValueSupport;
/**
*
* @author James Ahlborn
*/
public abstract class BaseEvalContext implements EvalContext
{
/** map of all non-string data types */
private static final Map<DataType,Value.Type> TYPE_MAP =
new EnumMap<DataType,Value.Type>(DataType.class);
static {
TYPE_MAP.put(DataType.BOOLEAN,Value.Type.LONG);
TYPE_MAP.put(DataType.BYTE,Value.Type.LONG);
TYPE_MAP.put(DataType.INT,Value.Type.LONG);
TYPE_MAP.put(DataType.LONG,Value.Type.LONG);
TYPE_MAP.put(DataType.MONEY,Value.Type.DOUBLE);
TYPE_MAP.put(DataType.FLOAT,Value.Type.DOUBLE);
TYPE_MAP.put(DataType.DOUBLE,Value.Type.DOUBLE);
TYPE_MAP.put(DataType.SHORT_DATE_TIME,Value.Type.DATE_TIME);
TYPE_MAP.put(DataType.NUMERIC,Value.Type.BIG_DEC);
TYPE_MAP.put(DataType.BIG_INT,Value.Type.BIG_DEC);
}
private final DBEvalContext _dbCtx;
private Expression _expr;
protected BaseEvalContext(DBEvalContext dbCtx) {
_dbCtx = dbCtx;
}
void setExpr(Expressionator.Type exprType, String exprStr) {
_expr = new RawExpr(exprType, exprStr);
}
protected DatabaseImpl getDatabase() {
return _dbCtx.getDatabase();
}
@Override
public TemporalConfig getTemporalConfig() {
return _dbCtx.getTemporalConfig();
}
@Override
public DateTimeFormatter createDateFormatter(String formatStr) {
return _dbCtx.createDateFormatter(formatStr);
}
@Override
public ZoneId getZoneId() {
return _dbCtx.getZoneId();
}
@Override
public NumericConfig getNumericConfig() {
return _dbCtx.getNumericConfig();
}
@Override
public DecimalFormat createDecimalFormat(String formatStr) {
return _dbCtx.createDecimalFormat(formatStr);
}
@Override
public float getRandom(Integer seed) {
return _dbCtx.getRandom(seed);
}
@Override
public Value.Type getResultType() {
return null;
}
@Override
public Value getThisColumnValue() {
throw new UnsupportedOperationException();
}
@Override
public Value getIdentifierValue(Identifier identifier) {
throw new UnsupportedOperationException();
}
@Override
public Bindings getBindings() {
return _dbCtx.getBindings();
}
@Override
public Object get(String key) {
return _dbCtx.getBindings().get(key);
}
@Override
public void put(String key, Object value) {
_dbCtx.getBindings().put(key, value);
}
public Object eval() throws IOException {
try {
return _expr.eval(this);
} catch(Exception e) {
String msg = withErrorContext(e.getMessage());
throw new JackcessException(msg, e);
}
}
public void collectIdentifiers(Collection<Identifier> identifiers) {
_expr.collectIdentifiers(identifiers);
}
@Override
public String toString() {
return _expr.toString();
}
protected Value toValue(Object val, DataType dType) {
try {
// expression engine always uses LocalDateTime, so force that date/time
// type
val = ColumnImpl.toInternalValue(dType, val, getDatabase(),
ColumnImpl.LDT_DATE_TIME_FACTORY);
if(val == null) {
return ValueSupport.NULL_VAL;
}
Value.Type vType = toValueType(dType);
switch(vType) {
case STRING:
return ValueSupport.toValue(val.toString());
case DATE:
case TIME:
case DATE_TIME:
return ValueSupport.toValue(vType, (LocalDateTime)val);
case LONG:
Integer i = ((val instanceof Integer) ? (Integer)val :
((Number)val).intValue());
return ValueSupport.toValue(i);
case DOUBLE:
Double d = ((val instanceof Double) ? (Double)val :
((Number)val).doubleValue());
return ValueSupport.toValue(d);
case BIG_DEC:
BigDecimal bd = ColumnImpl.toBigDecimal(val, getDatabase());
return ValueSupport.toValue(bd);
default:
throw new RuntimeException("Unexpected type " + vType);
}
} catch(IOException e) {
throw new EvalException("Failed converting value to type " + dType, e);
}
}
public static Value.Type toValueType(DataType dType) {
Value.Type type = TYPE_MAP.get(dType);
return ((type == null) ? Value.Type.STRING : type);
}
protected abstract String withErrorContext(String msg);
private class RawExpr implements Expression
{
private final Expressionator.Type _exprType;
private final String _exprStr;
private RawExpr(Expressionator.Type exprType, String exprStr) {
_exprType = exprType;
_exprStr = exprStr;
}
private Expression getExpr() {
// when the expression is parsed we replace the raw version
Expression expr = Expressionator.parse(
_exprType, _exprStr, getResultType(), _dbCtx);
_expr = expr;
return expr;
}
@Override
public Object eval(EvalContext ctx) {
return getExpr().eval(ctx);
}
@Override
public String toDebugString(LocaleContext ctx) {
return getExpr().toDebugString(ctx);
}
@Override
public String toRawString() {
return _exprStr;
}
@Override
public String toCleanString(LocaleContext ctx) {
return getExpr().toCleanString(ctx);
}
@Override
public boolean isConstant() {
return getExpr().isConstant();
}
@Override
public void collectIdentifiers(Collection<Identifier> identifiers) {
getExpr().collectIdentifiers(identifiers);
}
@Override
public String toString() {
return toRawString();
}
}
}