2 changed files with 266 additions and 0 deletions
@ -0,0 +1,129 @@
@@ -0,0 +1,129 @@
|
||||
/* |
||||
* Copyright 2010 the original author or authors. |
||||
* |
||||
* Licensed 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.springframework.data.document.mongodb; |
||||
|
||||
import java.util.List; |
||||
|
||||
import org.apache.commons.logging.Log; |
||||
import org.apache.commons.logging.LogFactory; |
||||
import org.springframework.beans.factory.FactoryBean; |
||||
import org.springframework.beans.factory.InitializingBean; |
||||
import org.springframework.dao.DataAccessException; |
||||
import org.springframework.dao.support.PersistenceExceptionTranslator; |
||||
import org.springframework.util.Assert; |
||||
|
||||
import com.mongodb.Mongo; |
||||
import com.mongodb.MongoOptions; |
||||
import com.mongodb.ServerAddress; |
||||
|
||||
/** |
||||
* Convenient factory for configuring MongoDB. |
||||
* |
||||
* @author Thomas Risberg |
||||
* @author Graeme Rocher |
||||
* |
||||
* @since 1.0 |
||||
*/ |
||||
public class MongoFactoryBean implements FactoryBean<Mongo>, InitializingBean, |
||||
PersistenceExceptionTranslator { |
||||
|
||||
|
||||
/** |
||||
* Logger, available to subclasses. |
||||
*/ |
||||
protected final Log logger = LogFactory.getLog(getClass()); |
||||
|
||||
private Mongo mongo; |
||||
private MongoOptions mongoOptions; |
||||
private String host; |
||||
private Integer port; |
||||
private List<ServerAddress> replicaSetSeeds; |
||||
private List<ServerAddress> replicaPair; |
||||
|
||||
public void setMongoOptions(MongoOptions mongoOptions) { |
||||
this.mongoOptions = mongoOptions; |
||||
} |
||||
|
||||
public void setReplicaSetSeeds(List<ServerAddress> replicaSetSeeds) { |
||||
this.replicaSetSeeds = replicaSetSeeds; |
||||
} |
||||
|
||||
public void setReplicaPair(List<ServerAddress> replicaPair) { |
||||
this.replicaPair = replicaPair; |
||||
} |
||||
|
||||
public void setHost(String host) { |
||||
this.host = host; |
||||
} |
||||
|
||||
public void setPort(int port) { |
||||
this.port = port; |
||||
} |
||||
|
||||
public Mongo getObject() throws Exception { |
||||
Assert.notNull(mongo, "Mongo must not be null"); |
||||
return mongo; |
||||
} |
||||
|
||||
public Class<? extends Mongo> getObjectType() { |
||||
return Mongo.class; |
||||
} |
||||
|
||||
public boolean isSingleton() { |
||||
return false; |
||||
} |
||||
|
||||
public void afterPropertiesSet() throws Exception { |
||||
// apply defaults - convenient when used to configure for tests
|
||||
// in an application context
|
||||
if (mongo == null) { |
||||
|
||||
if (host == null) { |
||||
logger.warn("Property host not specified. Using default configuration"); |
||||
mongo = new Mongo(); |
||||
} |
||||
else { |
||||
ServerAddress defaultOptions = new ServerAddress(); |
||||
if(mongoOptions == null) mongoOptions = new MongoOptions(); |
||||
if(replicaPair != null) { |
||||
if(replicaPair.size() < 2) { |
||||
throw new CannotGetMongoDbConnectionException("A replica pair must have two server entries"); |
||||
} |
||||
mongo = new Mongo(replicaPair.get(0), replicaPair.get(1), mongoOptions); |
||||
} |
||||
else if(replicaSetSeeds != null) { |
||||
mongo = new Mongo(replicaSetSeeds, mongoOptions); |
||||
} |
||||
else { |
||||
String mongoHost = host != null ? host : defaultOptions.getHost(); |
||||
if(port != null) { |
||||
mongo = new Mongo(new ServerAddress(mongoHost, port), mongoOptions); |
||||
} |
||||
else { |
||||
mongo = new Mongo(mongoHost, mongoOptions); |
||||
} |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
public DataAccessException translateExceptionIfPossible(RuntimeException ex) { |
||||
logger.debug("Translating " + ex); |
||||
return MongoDbUtils.translateMongoExceptionIfPossible(ex); |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,137 @@
@@ -0,0 +1,137 @@
|
||||
/* |
||||
* Copyright 2010 the original author or authors. |
||||
* |
||||
* Licensed 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.springframework.data.document.mongodb; |
||||
|
||||
import org.springframework.beans.factory.FactoryBean; |
||||
import org.springframework.beans.factory.InitializingBean; |
||||
|
||||
|
||||
import com.mongodb.MongoOptions; |
||||
|
||||
/** |
||||
* A factory bean for consruction a MongoOptions instance |
||||
* |
||||
* @author Graeme Rocher |
||||
* |
||||
*/ |
||||
public class MongoOptionsFactoryBean implements FactoryBean<MongoOptions>, InitializingBean{ |
||||
|
||||
private static final MongoOptions MONGO_OPTIONS = new MongoOptions(); |
||||
/** |
||||
number of connections allowed per host |
||||
will block if run out |
||||
*/ |
||||
private int connectionsPerHost = MONGO_OPTIONS.connectionsPerHost; |
||||
|
||||
/** |
||||
multiplier for connectionsPerHost for # of threads that can block |
||||
if connectionsPerHost is 10, and threadsAllowedToBlockForConnectionMultiplier is 5, |
||||
then 50 threads can block |
||||
more than that and an exception will be throw |
||||
*/ |
||||
private int threadsAllowedToBlockForConnectionMultiplier = MONGO_OPTIONS.threadsAllowedToBlockForConnectionMultiplier; |
||||
|
||||
/** |
||||
* max wait time of a blocking thread for a connection |
||||
*/ |
||||
private int maxWaitTime = MONGO_OPTIONS.maxWaitTime; |
||||
|
||||
/** |
||||
connect timeout in milliseconds. 0 is default and infinite |
||||
*/ |
||||
private int connectTimeout = MONGO_OPTIONS.connectTimeout; |
||||
|
||||
/** |
||||
socket timeout. 0 is default and infinite |
||||
*/ |
||||
private int socketTimeout = MONGO_OPTIONS.socketTimeout; |
||||
|
||||
/** |
||||
this controls whether or not on a connect, the system retries automatically |
||||
*/ |
||||
private boolean autoConnectRetry = MONGO_OPTIONS.autoConnectRetry; |
||||
|
||||
|
||||
/** |
||||
number of connections allowed per host |
||||
will block if run out |
||||
*/ |
||||
public void setConnectionsPerHost(int connectionsPerHost) { |
||||
this.connectionsPerHost = connectionsPerHost; |
||||
} |
||||
|
||||
/** |
||||
multiplier for connectionsPerHost for # of threads that can block |
||||
if connectionsPerHost is 10, and threadsAllowedToBlockForConnectionMultiplier is 5, |
||||
then 50 threads can block |
||||
more than that and an exception will be throw |
||||
*/ |
||||
public void setThreadsAllowedToBlockForConnectionMultiplier( |
||||
int threadsAllowedToBlockForConnectionMultiplier) { |
||||
this.threadsAllowedToBlockForConnectionMultiplier = threadsAllowedToBlockForConnectionMultiplier; |
||||
} |
||||
|
||||
/** |
||||
* max wait time of a blocking thread for a connection |
||||
*/ |
||||
public void setMaxWaitTime(int maxWaitTime) { |
||||
this.maxWaitTime = maxWaitTime; |
||||
} |
||||
|
||||
/** |
||||
connect timeout in milliseconds. 0 is default and infinite |
||||
*/ |
||||
public void setConnectTimeout(int connectTimeout) { |
||||
this.connectTimeout = connectTimeout; |
||||
} |
||||
|
||||
/** |
||||
socket timeout. 0 is default and infinite |
||||
*/ |
||||
public void setSocketTimeout(int socketTimeout) { |
||||
this.socketTimeout = socketTimeout; |
||||
} |
||||
|
||||
/** |
||||
this controls whether or not on a connect, the system retries automatically |
||||
*/ |
||||
public void setAutoConnectRetry(boolean autoConnectRetry) { |
||||
this.autoConnectRetry = autoConnectRetry; |
||||
} |
||||
|
||||
public void afterPropertiesSet() throws Exception { |
||||
MONGO_OPTIONS.connectionsPerHost = connectionsPerHost; |
||||
MONGO_OPTIONS.threadsAllowedToBlockForConnectionMultiplier = threadsAllowedToBlockForConnectionMultiplier; |
||||
MONGO_OPTIONS.maxWaitTime = maxWaitTime; |
||||
MONGO_OPTIONS.connectTimeout = connectTimeout; |
||||
MONGO_OPTIONS.socketTimeout = socketTimeout; |
||||
MONGO_OPTIONS.autoConnectRetry = autoConnectRetry; |
||||
|
||||
} |
||||
|
||||
public MongoOptions getObject() throws Exception { |
||||
return MONGO_OPTIONS; |
||||
} |
||||
|
||||
public Class<?> getObjectType() { |
||||
return MongoOptions.class; |
||||
} |
||||
|
||||
public boolean isSingleton() { |
||||
return true; |
||||
} |
||||
|
||||
} |
||||
Loading…
Reference in new issue