Examples with Literal org.teiid.language.Literal used on opensource projects

Example 66 with Literal

use of org.teiid.language.Literal in project teiid by teiid.

the class SwaggerProcedureExecution method addArgumentValue.

private void addArgumentValue(String argName, Array value, String collectionFormat, StringBuilder queryStr) {
    List<Expression> exprs = value.getExpressions();
    if (collectionFormat.equalsIgnoreCase("multi")) {
        for (int i = 0; i < exprs.size(); i++) {
            if (i > 0) {
                queryStr.append("&");
            }
            queryStr.append(argName);
            queryStr.append(Tokens.EQ);
            Literal l = (Literal) exprs.get(i);
            queryStr.append(getURLValue(l));
        }
    } else {
        String delimiter = ",";
        if (collectionFormat.equalsIgnoreCase("csv")) {
            delimiter = ",";
        } else if (collectionFormat.equalsIgnoreCase("ssv")) {
            delimiter = " ";
        } else if (collectionFormat.equalsIgnoreCase("tsv")) {
            delimiter = "\t";
        } else if (collectionFormat.equalsIgnoreCase("pipes")) {
            delimiter = "|";
        }
        queryStr.append(argName);
        queryStr.append(Tokens.EQ);
        for (int i = 0; i < exprs.size(); i++) {
            Literal l = (Literal) exprs.get(i);
            if (i > 0) {
                queryStr.append(delimiter);
            }
            queryStr.append(getURLValue(l));
        }
    }
}

Example 67 with Literal

use of org.teiid.language.Literal in project teiid by teiid.

the class TestSingleInsert method testDateTypes.

@Test
public void testDateTypes() throws Exception {
    NamedTable table = new NamedTable("temp", null, Mockito.mock(Table.class));
    ArrayList<ColumnReference> elements = new ArrayList<ColumnReference>();
    elements.add(new ColumnReference(table, "one", Mockito.mock(Column.class), Integer.class));
    elements.add(new ColumnReference(table, "two", Mockito.mock(Column.class), Date.class));
    elements.add(new ColumnReference(table, "three", Mockito.mock(Column.class), Timestamp.class));
    List<Expression> values = new ArrayList<Expression>();
    values.add(new Literal(1, DataTypeManager.DefaultDataClasses.INTEGER));
    values.add(new Literal(TimestampUtil.createDate(100, 01, 1), DataTypeManager.DefaultDataClasses.DATE));
    values.add(new Literal(TimestampUtil.createTimestamp(100, 01, 1, 0, 4, 0, 0), DataTypeManager.DefaultDataClasses.TIMESTAMP));
    ExpressionValueSource valueSource = new ExpressionValueSource(values);
    Insert insert = new Insert(table, elements, valueSource);
    SalesforceConnection connection = Mockito.mock(SalesforceConnection.class);
    Mockito.stub(connection.create(Mockito.any(DataPayload.class))).toAnswer(new Answer<Integer>() {

        @Override
        public Integer answer(InvocationOnMock invocation) throws Throwable {
            DataPayload payload = (DataPayload) invocation.getArguments()[0];
            List<DataPayload.Field> fields = payload.getMessageElements();
            assertEquals(3, fields.size());
            assertEquals(1, fields.get(0).value);
            assertEquals(TimestampUtil.createDate(100, 01, 1), fields.get(1).value);
            Calendar cal = (Calendar) fields.get(2).value;
            assertEquals(TimeZone.getTimeZone("GMT-1"), cal.getTimeZone());
            return 1;
        }
    });
    Mockito.stub(connection.upsert(Mockito.any(DataPayload.class))).toReturn(1);
    SalesForceExecutionFactory config = new SalesForceExecutionFactory();
    config.setMaxBulkInsertBatchSize(1);
    InsertExecutionImpl updateExecution = new InsertExecutionImpl(config, insert, connection, Mockito.mock(RuntimeMetadata.class), Mockito.mock(ExecutionContext.class));
    while (true) {
        try {
            updateExecution.execute();
            org.junit.Assert.assertArrayEquals(new int[] { 1 }, updateExecution.getUpdateCounts());
            break;
        } catch (DataNotAvailableException e) {
            continue;
        }
    }
    insert.setUpsert(true);
    updateExecution = new InsertExecutionImpl(config, insert, connection, Mockito.mock(RuntimeMetadata.class), Mockito.mock(ExecutionContext.class));
    while (true) {
        try {
            updateExecution.execute();
            org.junit.Assert.assertArrayEquals(new int[] { 1 }, updateExecution.getUpdateCounts());
            break;
        } catch (DataNotAvailableException e) {
            continue;
        }
    }
    Mockito.verify(connection).upsert(Mockito.any(DataPayload.class));
}

Example 68 with Literal

use of org.teiid.language.Literal in project teiid by teiid.

the class CoherenceUpdateExecution method addChildObject.

private void addChildObject(Table t) throws TranslatorException {
    List<ForeignKey> fks = t.getForeignKeys();
    ForeignKey fk = fks.get(0);
    Table parentTable = fk.getParent();
    // the name of the method to obtain the collection is the nameInSource of the foreginKey
    String parentToChildMethod = fk.getNameInSource();
    if (parentToChildMethod == null) {
        // $NON-NLS-1$
        final String msg = CoherencePlugin.Util.getString("CoherenceUpdateExecution.noNameInSourceForForeingKey", new Object[] { fk.getName() });
        throw new TranslatorException(msg);
    }
    // there must only be 1 column in the primary key
    String parentColName = visitor.getNameFromElement(fk.getPrimaryKey().getColumns().get(0));
    Insert icommand = (Insert) command;
    List<ColumnReference> insertElementList = icommand.getColumns();
    List<Expression> insertValueList = ((ExpressionValueSource) icommand.getValueSource()).getValues();
    if (insertElementList.size() != insertValueList.size()) {
        throw new TranslatorException("Error:  columns.size and values.size are not the same.");
    }
    ColumnReference insertElement;
    String[] nameOfElement = new String[insertElementList.size()];
    int parentValueLoc = -1;
    for (int i = 0; i < insertElementList.size(); i++) {
        insertElement = insertElementList.get(i);
        // // call utility class to get NameInSource/Name of element
        nameOfElement[i] = visitor.getNameFromElement(insertElement.getMetadataObject());
        // match the parent column to the colum in the insert statement
        if (nameOfElement[i].equalsIgnoreCase(parentColName)) {
            parentValueLoc = i;
        }
    }
    if (parentColName != null && parentValueLoc == -1) {
        // $NON-NLS-1$
        final String msg = CoherencePlugin.Util.getString("CoherenceUpdateExecution.noColumnMatchedForeignColumn", new Object[] { t.getName(), parentColName });
        throw new TranslatorException(msg);
    }
    // get the parent key and find the root object
    Object parentValue = insertValueList.get(parentValueLoc);
    Object val;
    if (parentValue instanceof Literal) {
        Literal literalValue = (Literal) parentValue;
        val = literalValue.getValue();
    } else {
        val = parentValue;
    }
    Object parentObject = null;
    // get the parent object from the cache
    try {
        List<Object> result = this.connection.get(CoherenceFilterUtil.createCompareFilter(parentColName, val, Operator.EQ, val.getClass()));
        // visitor.createFilter(parentColName + " = " + val));
        if (result == null || result.isEmpty()) {
            // $NON-NLS-1$
            final String msg = CoherencePlugin.Util.getString("CoherenceUpdateExecution.noobjectfound", new Object[] { parentTable.getName(), parentColName, val });
            throw new TranslatorException(msg);
        }
        parentObject = result.get(0);
    } catch (ResourceException e) {
        throw new TranslatorException(e);
    }
    // create and load the child object data
    Object newChildObject = cacheTranslator.createObject(insertElementList, insertValueList, this.visitor, t);
    // /--- questions
    // / --- how to not process - setvalue for parent column
    // / --- need to get the key value off the object of the parent
    // get the key value to use to for adding to the cache
    Object parentContainer = ObjectSourceMethodManager.getValue("get" + parentToChildMethod, parentObject);
    if (parentContainer == null) {
        // $NON-NLS-1$
        final String msg = CoherencePlugin.Util.getString("CoherenceUpdateExecution.noParentContainerObjectFound", new Object[] { parentTable.getName(), parentToChildMethod });
        throw new TranslatorException(msg);
    }
    if (parentContainer.getClass().isArray()) {
    } else if (parentContainer instanceof Collection) {
        Collection c = (Collection) parentContainer;
        c.add(newChildObject);
    } else if (parentContainer instanceof Map) {
        Map m = (Map) parentContainer;
        m.put(1, newChildObject);
    }
    try {
        this.connection.update(parentValue, parentObject);
    } catch (ResourceException e) {
        throw new TranslatorException(e);
    }
}

Example 69 with Literal

use of org.teiid.language.Literal in project teiid by teiid.

the class CoherenceVisitor method getExpressionString.

// GHH 20080326 - found that code to fall back on Name if NameInSource
// was null wasn't working properly, so replaced with tried and true
// code from another custom connector.
private String getExpressionString(Expression e) throws TranslatorException {
    String expressionName = null;
    // - the rest of this method is unchanged
    if (e instanceof ColumnReference) {
        Column mdIDElement = ((ColumnReference) e).getMetadataObject();
        expressionName = mdIDElement.getNameInSource();
        if (expressionName == null || expressionName.equals("")) {
            // $NON-NLS-1$
            expressionName = mdIDElement.getName();
        }
    } else if (e instanceof Literal) {
        // try {
        // if(((Literal)e).getType().equals(Class.forName(Timestamp.class.getName()))) {
        // LogManager.logTrace(LogConstants.CTX_CONNECTOR, "Found an expression that uses timestamp; converting to LDAP string format."); //$NON-NLS-1$
        // Timestamp ts = (Timestamp)((Literal)e).getValue();
        // Date dt = new Date(ts.getTime());
        // //TODO: Fetch format if provided.
        // SimpleDateFormat sdf = new SimpleDateFormat(LDAPConnectorConstants.ldapTimestampFormat);
        // expressionName = sdf.format(dt);
        // LogManager.logTrace(LogConstants.CTX_CONNECTOR, "Timestamp to stsring is: " + expressionName); //$NON-NLS-1$
        // }
        // else {
        // expressionName = ((Literal)e).getValue().toString();
        // }
        expressionName = ((Literal) e).getValue().toString();
    // } catch (ClassNotFoundException cce) {
    // final String msg = LDAPPlugin.Util.getString("IQueryToLdapSearchParser.timestampClassNotFoundError"); //$NON-NLS-1$
    // throw new TranslatorException(cce, msg);
    // }
    // 
    } else {
        if (e instanceof AggregateFunction) {
            // $NON-NLS-1$
            LogManager.logError(LogConstants.CTX_CONNECTOR, "Received IAggregate, but it is not supported. Check capabilities.");
        } else if (e instanceof Function) {
            // $NON-NLS-1$
            LogManager.logError(LogConstants.CTX_CONNECTOR, "Received IFunction, but it is not supported. Check capabilties.");
        } else if (e instanceof ScalarSubquery) {
            // $NON-NLS-1$
            LogManager.logError(LogConstants.CTX_CONNECTOR, "Received IScalarSubquery, but it is not supported. Check capabilties.");
        } else if (e instanceof SearchedCase) {
            // $NON-NLS-1$
            LogManager.logError(LogConstants.CTX_CONNECTOR, "Received ISearchedCaseExpression, but it is not supported. Check capabilties.");
        }
        // $NON-NLS-1$
        final String msg = CoherencePlugin.Util.getString("CoherenceVisitory.unsupportedElementError", e.toString());
        throw new TranslatorException(msg);
    }
    expressionName = escapeReservedChars(expressionName);
    return expressionName;
}

Example 70 with Literal

use of org.teiid.language.Literal in project teiid by teiid.

the class CoherenceVisitor method addParmFromExpression.

private Class addParmFromExpression(Expression expr, List parms) {
    Class type = null;
    if (expr instanceof Literal) {
        Long longparm = null;
        Literal literal = (Literal) expr;
        parms.add(literal);
        type = literal.getType();
    } else {
        // $NON-NLS-1$
        this.exception = new TranslatorException("CoherenceVisitor.Unsupported_expression" + expr);
    }
    return type;
}