JiriOndrusek commented on a change in pull request #2694: URL: https://github.com/apache/camel-quarkus/pull/2694#discussion_r642455214
########## File path: integration-tests/sql/src/main/java/org/apache/camel/quarkus/component/sql/it/SqlRoutes.java ########## @@ -0,0 +1,148 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You 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 org.apache.camel.quarkus.component.sql.it; + +import java.io.*; +import java.nio.file.Path; +import java.sql.SQLException; +import java.util.HashMap; +import java.util.List; +import java.util.Map; +import java.util.concurrent.CopyOnWriteArrayList; + +import javax.enterprise.context.ApplicationScoped; +import javax.enterprise.inject.Produces; +import javax.inject.Inject; +import javax.inject.Named; +import javax.transaction.TransactionManager; +import javax.transaction.UserTransaction; + +import io.agroal.api.AgroalDataSource; +import org.apache.camel.AggregationStrategy; +import org.apache.camel.Exchange; +import org.apache.camel.builder.RouteBuilder; +import org.apache.camel.processor.aggregate.jdbc.JdbcAggregationRepository; +import org.apache.camel.processor.idempotent.jdbc.JdbcMessageIdRepository; +import org.springframework.transaction.jta.JtaTransactionManager; + +@ApplicationScoped +public class SqlRoutes extends RouteBuilder { + + @Inject + @Named("results") + Map<String, List> results; + + @Inject + TransactionManager tm; + + @Inject + UserTransaction userTransaction; + + @Inject + AgroalDataSource dataSource; + + @Inject + SqlDbInitializer sqlDbInitializer; + + @Override + public void configure() throws IOException, SQLException { + //db has to be initialized before routes are started + sqlDbInitializer.initDb(); + + from("sql:select * from projects where processed = false order by id?initialDelay=0&delay=50&consumer.onConsume=update projects set processed = true where id = :#id") + .id("consumerRoute").autoStartup(false) + .process(e -> results.get("consumerRoute").add(e.getMessage().getBody(Map.class))); + + from("sql:classpath:sql/selectProjects.sql?initialDelay=0&delay=50&consumer.onConsume=update projects set processed = true") + .id("consumerClasspathRoute").autoStartup(false) + .process(e -> results.get("consumerClasspathRoute").add(e.getMessage().getBody(Map.class))); + + Path tmpFile = createTmpFileFrom("sql/selectProjects.sql"); + from("sql:file:" + tmpFile + + "?initialDelay=0&delay=50&consumer.onConsume=update projects set processed = true") + .id("consumerFileRoute").autoStartup(false) + .process(e -> results.get("consumerFileRoute").add(e.getMessage().getBody(Map.class))); + + from("direct:transacted") + .transacted("PROPAGATION_REQUIRED") + .to("sql:overriddenByTheHeader") + .process(e -> { + if (e.getIn().getHeader("rollback", boolean.class)) { + throw new Exception("forced Exception"); + } + }); + + // Idempotent Repository + JdbcMessageIdRepository repo = new JdbcMessageIdRepository(dataSource, "idempotentRepo"); + from("direct:idempotent") + .idempotentConsumer(header("messageId"), repo) + .process(e -> results.get("idempotentRoute").add(e.getMessage().getBody(String.class))); + + //aggregation repository + JdbcAggregationRepository aggregationRepo = new JdbcAggregationRepository( + new JtaTransactionManager(userTransaction, tm), "aggregation", dataSource); + from("direct:aggregation") + .aggregate(header("messageId"), new MyAggregationStrategy()) + // use our created jdbc repo as aggregation repository + .completionSize(4).aggregationRepository(aggregationRepo) + .process(e -> results.get("aggregationRoute").add(e.getMessage().getBody(String.class))); + + } + + private Path createTmpFileFrom(String file) throws IOException { + File tmpFile = File.createTempFile("selectProjects-", ".sql"); + tmpFile.deleteOnExit(); + try (InputStream is = Thread.currentThread().getContextClassLoader().getResourceAsStream(file)) { //todo quick workaround + + ByteArrayOutputStream out = new ByteArrayOutputStream(); + int c; + while ((c = is.read()) >= 0) { + out.write(c); + } + new FileOutputStream(tmpFile).write(out.toByteArray()); Review comment: My fault, I'll fix it. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org
