001 /*
002 * JBoss, Home of Professional Open Source.
003 * Copyright 2008, Red Hat Middleware LLC, and individual contributors
004 * as indicated by the @author tags. See the copyright.txt file in the
005 * distribution for a full listing of individual contributors.
006 *
007 * This is free software; you can redistribute it and/or modify it
008 * under the terms of the GNU Lesser General Public License as
009 * published by the Free Software Foundation; either version 2.1 of
010 * the License, or (at your option) any later version.
011 *
012 * This software is distributed in the hope that it will be useful,
013 * but WITHOUT ANY WARRANTY; without even the implied warranty of
014 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
015 * Lesser General Public License for more details.
016 *
017 * You should have received a copy of the GNU Lesser General Public
018 * License along with this software; if not, write to the Free
019 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
020 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
021 */
022 package org.jboss.dna.connector.federation;
023
024 import java.util.concurrent.TimeUnit;
025 import java.util.concurrent.atomic.AtomicReference;
026 import javax.transaction.xa.XAResource;
027 import net.jcip.annotations.ThreadSafe;
028 import org.jboss.dna.graph.ExecutionContext;
029 import org.jboss.dna.graph.cache.CachePolicy;
030 import org.jboss.dna.graph.commands.GraphCommand;
031 import org.jboss.dna.graph.commands.executor.CommandExecutor;
032 import org.jboss.dna.graph.connectors.RepositoryConnection;
033 import org.jboss.dna.graph.connectors.RepositorySourceException;
034 import org.jboss.dna.graph.connectors.RepositorySourceListener;
035
036 /**
037 * @author Randall Hauch
038 */
039 @ThreadSafe
040 public class FederatedRepositoryConnection implements RepositoryConnection {
041
042 protected static final RepositorySourceListener NO_OP_LISTENER = new RepositorySourceListener() {
043 public void notify( String sourceName,
044 Object... events ) {
045 // do nothing
046 }
047 };
048
049 private final FederatedRepository repository;
050 private final String sourceName;
051 private final AtomicReference<RepositorySourceListener> listener;
052
053 protected FederatedRepositoryConnection( FederatedRepository repository,
054 String sourceName ) {
055 assert sourceName != null;
056 assert repository != null;
057 this.sourceName = sourceName;
058 this.repository = repository;
059 this.listener = new AtomicReference<RepositorySourceListener>(NO_OP_LISTENER);
060 this.repository.register(this);
061 }
062
063 /**
064 * @return repository
065 */
066 protected FederatedRepository getRepository() {
067 return this.repository;
068 }
069
070 /**
071 * {@inheritDoc}
072 */
073 public String getSourceName() {
074 return this.sourceName;
075 }
076
077 /**
078 * {@inheritDoc}
079 */
080 public CachePolicy getDefaultCachePolicy() {
081 return this.repository.getConfiguration().getDefaultCachePolicy();
082 }
083
084 /**
085 * {@inheritDoc}
086 */
087 public XAResource getXAResource() {
088 return null;
089 }
090
091 /**
092 * {@inheritDoc}
093 */
094 public void setListener( RepositorySourceListener listener ) {
095 if (listener == null) listener = NO_OP_LISTENER;
096 RepositorySourceListener oldListener = this.listener.getAndSet(listener);
097 this.repository.addListener(listener);
098 if (oldListener != NO_OP_LISTENER) {
099 this.repository.removeListener(oldListener);
100 }
101 }
102
103 /**
104 * {@inheritDoc}
105 */
106 public boolean ping( long time,
107 TimeUnit unit ) {
108 return this.repository.isRunning();
109 }
110
111 /**
112 * {@inheritDoc}
113 */
114 public void execute( ExecutionContext context,
115 GraphCommand... commands ) throws RepositorySourceException {
116 if (!this.repository.isRunning()) {
117 throw new RepositorySourceException(FederationI18n.repositoryHasBeenShutDown.text(this.repository.getName()));
118 }
119 if (commands == null || commands.length == 0) return;
120
121 CommandExecutor executor = this.repository.getExecutor(context, sourceName);
122 try {
123 assert executor != null;
124 for (GraphCommand command : commands) {
125 executor.execute(command);
126 }
127 } finally {
128 executor.close();
129 }
130 }
131
132 /**
133 * {@inheritDoc}
134 */
135 public void close() {
136 try {
137 this.repository.removeListener(this.listener.get());
138 } finally {
139 this.repository.unregister(this);
140 }
141 }
142
143 }