Package org.teiid.language

Examples of org.teiid.language.Insert

90
91
92
93
94
95
96
97
98
99
            if (l.isBindValue() || isBindEligible(l)) {
                return true;
            }
        }
      if (command instanceof Insert) {
          Insert insert = (Insert)command;
          return insert.getValueSource() instanceof IteratorValueSource<?>;
      }
        return false;
    }
View Full Code Here
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
                addStatementWarnings();
            } else {
              PreparedStatement pstatement = getPreparedStatement(sql);
             
              if (command instanceof Insert) {
                  Insert insert = (Insert)command;
                  if (insert.getValueSource() instanceof IteratorValueSource) {
                        commitType = getAutoCommit(translatedComm);
                        if (commitType) {
                            connection.setAutoCommit(false);
                        }
                   
                    IteratorValueSource<List<Object>> ivs = (IteratorValueSource)insert.getValueSource();
                    List<Object>[] values = new List[ivs.getColumnCount()];
                    for (int i = 0; i < ivs.getColumnCount(); i++) {
                      values[i] = new ArrayList<Object>();
                      Literal literal = new Literal(values[i], insert.getColumns().get(i).getType());
                      literal.setMultiValued(true);
                      translatedComm.getPreparedValues().add(literal);
                    }
                    Iterator<List<Object>> i = ivs.getIterator();
                    int maxBatchSize = this.executionFactory.getMaxPreparedInsertBatchSize();
View Full Code Here
389
390
391
392
393
394
395
396
397
398
399
    }
   
    @Test public void testInsertWithQuery() throws Exception {
      String sql = "insert into pm1.g1 values (null, null, null, null)"; //$NON-NLS-1$

      Insert insert = (Insert)FakeTranslationFactory.getInstance().getExampleTranslationUtility().parseCommand(sql);
     
      Select command = (Select)FakeTranslationFactory.getInstance().getExampleTranslationUtility().parseCommand("select * from pm1.g2"); //$NON-NLS-1$
      insert.setValueSource(command);
      assertEquals("INSERT INTO g1 (e1, e2, e3, e4) SELECT g2.e1, g2.e2, g2.e3, g2.e4 FROM g2", insert.toString()); //$NON-NLS-1$
    }
View Full Code Here
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
    updateExecution.execute();
    Mockito.verify(p, Mockito.times(2)).addBatch();
  }
 
  @Test public void testInsertIteratorUpdate() throws Exception {
    Insert command = (Insert)TranslationHelper.helpTranslate(TranslationHelper.BQT_VDB, "insert into BQT1.SmallA (IntKey, IntNum) values (1, 2)"); //$NON-NLS-1$
    List<List<Integer>> values = new ArrayList<List<Integer>>();
    values.add(Arrays.asList(1, 2));
    values.add(Arrays.asList(2, 3));
    command.setValueSource(new IteratorValueSource(values.iterator(), 2));
   
    Connection connection = Mockito.mock(Connection.class);
    PreparedStatement p = Mockito.mock(PreparedStatement.class);
    Mockito.stub(p.executeBatch()).toReturn(new int [] {1, 1});
    Mockito.stub(connection.prepareStatement("INSERT INTO SmallA (IntKey, IntNum) VALUES (?, ?)")).toReturn(p); //$NON-NLS-1$
View Full Code Here
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
            assertTrue(i.next() instanceof Expression);
        }
    }
   
    public void testExpressionsInInsert() throws Exception {
        Insert insert = example2("a.b"); //$NON-NLS-1$
        assertNotNull(insert.getColumns());
        assertEquals(1, insert.getColumns().size());
        for (Iterator i = insert.getColumns().iterator(); i.hasNext();) {
            assertTrue(i.next() instanceof ColumnReference);
        }
        assertNotNull(insert.getValueSource());
        assertEquals(1, ((ExpressionValueSource)insert.getValueSource()).getValues().size());
        for (Iterator i = ((ExpressionValueSource)insert.getValueSource()).getValues().iterator(); i.hasNext();) {
            assertTrue(i.next() instanceof Expression);
        }
    }
View Full Code Here
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
    public void testGetGroup() throws Exception {
        assertNotNull(example("a.b").getTable()); //$NON-NLS-1$
    }

    public void testGetElements() throws Exception {
        Insert insert = example("a.b"); //$NON-NLS-1$
        assertNotNull(insert.getColumns());
        assertEquals(4, insert.getColumns().size());
        for (Iterator i = insert.getColumns().iterator(); i.hasNext();) {
            assertTrue(i.next() instanceof ColumnReference);
        }

        // verify that elements are not qualified by group
        String sInsertSQL = insert.toString();
        assertTrue(sInsertSQL.substring(sInsertSQL.indexOf('(')).indexOf( '.') == -1 );                       
    }
View Full Code Here
102
103
104
105
106
107
108
109
110
111
112
113
        String sInsertSQL = insert.toString();
        assertTrue(sInsertSQL.substring(sInsertSQL.indexOf('(')).indexOf( '.') == -1 );                       
    }

    public void testGetValues() throws Exception {
        Insert insert = example("a.b"); //$NON-NLS-1$
        assertNotNull(insert.getValueSource());
        assertEquals(4, ((ExpressionValueSource)insert.getValueSource()).getValues().size());
        for (Iterator i = ((ExpressionValueSource)insert.getValueSource()).getValues().iterator(); i.hasNext();) {
            assertTrue(i.next() instanceof Expression);
        }
    }
View Full Code Here
TOP

Related Classes of org.teiid.language.Insert

  • org.teiid.connector.visitor.util.TestSQLStringVisitor
  • org.teiid.dqp.internal.datamgr.TestInsertImpl
  • org.teiid.translator.jdbc.JDBCUpdateExecution
  • org.teiid.translator.jdbc.TestJDBCUpdateExecution
  • org.teiid.translator.jdbc.TranslatedCommand
    • Copyright © 2018 www.massapicom. All rights reserved.
    All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.