From 307481edb44a9bfed74ba069f768e3669a9ed4b1 Mon Sep 17 00:00:00 2001
From: Thomas Risberg <trisberg@vmware.com>
Date: Fri, 27 Aug 2010 17:59:27 -0400
Subject: [PATCH] refactoring

---
 .../datastore/document/mongodb/MongoTemplate.java     | 11 ++++-------
 1 file changed, 4 insertions(+), 7 deletions(-)

diff --git a/src/main/java/org/springframework/datastore/document/mongodb/MongoTemplate.java b/src/main/java/org/springframework/datastore/document/mongodb/MongoTemplate.java
index 2a89b993e..7bd096c94 100644
--- a/src/main/java/org/springframework/datastore/document/mongodb/MongoTemplate.java
+++ b/src/main/java/org/springframework/datastore/document/mongodb/MongoTemplate.java
@@ -89,12 +89,10 @@ public class MongoTemplate extends AbstractDocumentStoreTemplate<DB> {
 	
 	public void save(String collectionName, DocumentSource<DBObject> documentSource) {
 		DBObject dbDoc = documentSource.getDocument();		
+		DB db =  getDocumentStoreConnectionFactory().getConnection();
 		WriteResult wr = null;
 		try {
-			wr = getDocumentStoreConnectionFactory()
-					.getConnection()
-					.getCollection(collectionName)
-					.save(dbDoc);
+			wr = db.getCollection(collectionName).save(dbDoc);
 		} catch (MongoException e) {
 			throw new DataRetrievalFailureException(wr.getLastError().getErrorMessage(), e);
 		}
@@ -107,9 +105,8 @@ public class MongoTemplate extends AbstractDocumentStoreTemplate<DB> {
 
 	public <T> List<T> queryForCollection(String collectionName, DocumentMapper<DBObject, T> mapper) {
 		List<T> results = new ArrayList<T>();
-		DBCollection collection = getDocumentStoreConnectionFactory()
-			.getConnection()
-			.getCollection(collectionName);
+		DB db =  getDocumentStoreConnectionFactory().getConnection();
+		DBCollection collection = db.getCollection(collectionName);
 		for (DBObject dbo : collection.find()) {
 			results.add(mapper.mapDocument(dbo));
 		}