/*
* 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.activemq.artemis.tests.integration.ra;
import javax.jms.Message;
import javax.jms.MessageListener;
import javax.resource.ResourceException;
import javax.resource.spi.ActivationSpec;
import javax.resource.spi.BootstrapContext;
import javax.resource.spi.UnavailableException;
import javax.resource.spi.XATerminator;
import javax.resource.spi.endpoint.MessageEndpoint;
import javax.resource.spi.endpoint.MessageEndpointFactory;
import javax.resource.spi.work.ExecutionContext;
import javax.resource.spi.work.Work;
import javax.resource.spi.work.WorkException;
import javax.resource.spi.work.WorkListener;
import javax.resource.spi.work.WorkManager;
import javax.transaction.xa.XAResource;
import java.lang.reflect.Method;
import java.util.Map;
import java.util.Timer;
import java.util.concurrent.CountDownLatch;
import org.apache.activemq.artemis.api.core.SimpleString;
import org.apache.activemq.artemis.api.core.client.ServerLocator;
import org.apache.activemq.artemis.core.settings.impl.AddressSettings;
import org.apache.activemq.artemis.jms.client.ActiveMQMessage;
import org.apache.activemq.artemis.ra.ActiveMQResourceAdapter;
import org.apache.activemq.artemis.ra.inflow.ActiveMQActivation;
import org.apache.activemq.artemis.tests.util.JMSTestBase;
import org.junit.Before;
public abstract class ActiveMQRATestBase extends JMSTestBase {
protected ServerLocator locator;
protected static final String MDBQUEUE = "mdbQueue";
protected static final String DLQ = "dlqQueue";
protected static final String MDBQUEUEPREFIXED = "mdbQueue";
protected static final SimpleString MDBQUEUEPREFIXEDSIMPLE = new SimpleString("mdbQueue");
@Override
@Before
public void setUp() throws Exception {
super.setUp();
locator = createInVMNonHALocator();
createQueue(true, MDBQUEUE);
createQueue(DLQ);
setupDLQ(1);
}
protected void setupDLQ(int maxDeliveries) {
AddressSettings settings = new AddressSettings().setDeadLetterAddress(SimpleString.toSimpleString(DLQ)).setMaxDeliveryAttempts(maxDeliveries);
server.getAddressSettingsRepository().addMatch("#", settings);
}
protected ActiveMQActivation lookupActivation(ActiveMQResourceAdapter qResourceAdapter) {
Map<ActivationSpec, ActiveMQActivation> activations = qResourceAdapter.getActivations();
assertEquals(1, activations.size());
return activations.values().iterator().next();
}
protected ActiveMQResourceAdapter newResourceAdapter() {
ActiveMQResourceAdapter qResourceAdapter = new ActiveMQResourceAdapter();
qResourceAdapter.setConnectorClassName(INVM_CONNECTOR_FACTORY);
return qResourceAdapter;
}
protected class DummyMessageEndpointFactory implements MessageEndpointFactory {
private DummyMessageEndpoint endpoint;
private final boolean isDeliveryTransacted;
public DummyMessageEndpointFactory(DummyMessageEndpoint endpoint, boolean deliveryTransacted) {
this.endpoint = endpoint;
isDeliveryTransacted = deliveryTransacted;
}
@Override
public MessageEndpoint createEndpoint(XAResource xaResource) throws UnavailableException {
if (xaResource != null) {
endpoint.setXAResource(xaResource);
}
return endpoint;
}
@Override
public boolean isDeliveryTransacted(Method method) throws NoSuchMethodException {
return isDeliveryTransacted;
}
}
protected class DummyMessageEndpoint implements MessageEndpoint, MessageListener {
public CountDownLatch latch;
public ActiveMQMessage lastMessage;
public boolean released = false;
public XAResource xaResource;
volatile boolean inAfterDelivery = false;
public DummyMessageEndpoint(CountDownLatch latch) {
this.latch = latch;
}
@Override
public void beforeDelivery(Method method) throws NoSuchMethodException, ResourceException {
}
@Override
public void afterDelivery() throws ResourceException {
if (latch != null) {
latch.countDown();
}
}
@Override
public void release() {
released = true;
}
@Override
public void onMessage(Message message) {
lastMessage = (ActiveMQMessage) message;
}
public void reset(CountDownLatch latch) {
this.latch = latch;
lastMessage = null;
}
public void setXAResource(XAResource xaResource) {
this.xaResource = xaResource;
}
}
public class MyBootstrapContext implements BootstrapContext {
WorkManager workManager = new DummyWorkManager();
@Override
public Timer createTimer() throws UnavailableException {
return null;
}
@Override
public WorkManager getWorkManager() {
return workManager;
}
@Override
public XATerminator getXATerminator() {
return null;
}
class DummyWorkManager implements WorkManager {
@Override
public void doWork(Work work) throws WorkException {
}
@Override
public void doWork(Work work,
long l,
ExecutionContext executionContext,
WorkListener workListener) throws WorkException {
}
@Override
public long startWork(Work work) throws WorkException {
return 0;
}
@Override
public long startWork(Work work,
long l,
ExecutionContext executionContext,
WorkListener workListener) throws WorkException {
return 0;
}
@Override
public void scheduleWork(Work work) throws WorkException {
work.run();
}
@Override
public void scheduleWork(Work work,
long l,
ExecutionContext executionContext,
WorkListener workListener) throws WorkException {
}
}
}
}