Use spring data to access MongoDB.

Author: XDongger

cluster/auth/pom.xml

@@ -8,7 +8,7 @@
 	<parent>
         <groupId>com.thefirstlineofcode.granite</groupId>
         <artifactId>granite-cluster</artifactId>
-        <version>1.0.4-RELEASE</version>
+        <version>1.0.5-RELEASE</version>
         <relativePath>../pom.xml</relativePath>
     </parent>
 

cluster/auth/src/main/java/com/thefirstlineofcode/granite/cluster/auth/AccountManager.java

@@ -1,60 +1,57 @@
 package com.thefirstlineofcode.granite.cluster.auth;
 
-import org.bson.Document;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.data.mongodb.core.MongoTemplate;
+import org.springframework.data.mongodb.core.query.Criteria;
+import org.springframework.data.mongodb.core.query.Query;
 import org.springframework.stereotype.Component;
+import org.springframework.transaction.annotation.Transactional;
 
-import com.mongodb.client.MongoCollection;
-import com.mongodb.client.MongoDatabase;
-import com.mongodb.client.model.Filters;
 import com.thefirstlineofcode.granite.framework.core.auth.Account;
 import com.thefirstlineofcode.granite.framework.core.auth.IAccountManager;
 
 @Component
+@Transactional
 public class AccountManager implements IAccountManager {
+	private static final String FIELD_NAME_NAME = "name";
+	
 	@Autowired
-	private MongoDatabase database;
+	private MongoTemplate mongoTemplate;
 
 	@Override
 	public void add(Account account) {
-		Document doc = new Document().
-				append("name", account.getName()).
-				append("password", account.getPassword());
-		getUsersCollection().insertOne(doc);
+		if (!(account instanceof D_Account))
+			throw new IllegalArgumentException(String.format("Must be type of '%s'", D_Account.class.getName()));
+		
+		if (account.getName() == null || account.getPassword() == null)
+			throw new IllegalArgumentException("Null user name or password.");
+		
+		mongoTemplate.save(account);
 	}
 
 	@Override
 	public void remove(String name) {
-		getUsersCollection().deleteOne(Filters.eq("name", name));
+		mongoTemplate.findAndRemove(
+				new Query().addCriteria(Criteria.where(FIELD_NAME_NAME).is(name)),
+				D_Account.class);
 	}
-
+	
 	@Override
 	public boolean exists(String name) {
-		return getUsersCollection().countDocuments(Filters.eq("name", name)) == 1;
+		return mongoTemplate.count(
+				new Query().addCriteria(Criteria.where(FIELD_NAME_NAME).is(name)),
+				D_Account.class) != 0;
 	}
-
+	
 	@Override
-	public Account get(String name) {
-		MongoCollection<Document> users = getUsersCollection();
-		Document doc = users.find(Filters.eq("name", name)).first();
-		if (doc == null)
-			return null;
-		
-		D_Account account = new D_Account();
-		account.setId(doc.getObjectId("_id").toHexString());
-		account.setName(doc.getString("name"));
-		account.setPassword(doc.getString("password"));
-		
-		return account;
+	public D_Account get(String name) {
+		return mongoTemplate.findOne(
+				new Query().addCriteria(Criteria.where(FIELD_NAME_NAME).is(name)),
+				D_Account.class);
 	}
 
-	private MongoCollection<Document> getUsersCollection() {
-		return database.getCollection("users");
-	}
-
 	@Override
-	public void add(String userName, String password) {
+	public void add(String name, String password) {
 		throw new UnsupportedOperationException();
 	}
-
 }

cluster/auth/src/main/java/com/thefirstlineofcode/granite/cluster/auth/Authenticator.java

@@ -1,36 +1,29 @@
 package com.thefirstlineofcode.granite.cluster.auth;
 
-import org.bson.Document;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
 
-import com.mongodb.client.MongoCollection;
-import com.mongodb.client.MongoDatabase;
-import com.mongodb.client.model.Filters;
+import com.thefirstlineofcode.granite.framework.core.auth.Account;
+import com.thefirstlineofcode.granite.framework.core.auth.IAccountManager;
 import com.thefirstlineofcode.granite.framework.core.auth.IAuthenticator;
 import com.thefirstlineofcode.granite.framework.core.auth.PrincipalNotFoundException;
 
 @Component
 public class Authenticator implements IAuthenticator {
 	@Autowired
-	private MongoDatabase database;
+	private IAccountManager accountManager;
 
 	@Override
 	public Object getCredentials(Object principal) throws PrincipalNotFoundException {
-		MongoCollection<Document> users = getUsersCollection();
-		Document doc = users.find(Filters.eq("name", (String)principal)).projection(
-				new Document("password", 1)).first();
+		Account account = accountManager.get((String)principal);
+		if (account == null)
+			return new PrincipalNotFoundException(String.format("Account name %s wasn't existed.", principal));
 
-		return doc == null ? null : doc.get("password");
+		return account.getPassword();
 	}
 
 	@Override
 	public boolean exists(Object principal) {
-		return getUsersCollection().countDocuments(Filters.eq("name", (String)principal)) == 1;
+		return accountManager.exists((String)principal);
 	}
-	
-	private MongoCollection<Document> getUsersCollection() {
-		return database.getCollection("users");
-	}
-
 }

cluster/auth/src/main/java/com/thefirstlineofcode/granite/cluster/auth/D_Account.java

@@ -1,9 +1,16 @@
 package com.thefirstlineofcode.granite.cluster.auth;
 
+import org.springframework.data.annotation.Id;
+import org.springframework.data.mongodb.core.mapping.Document;
+
 import com.thefirstlineofcode.granite.framework.core.adf.data.IIdProvider;
 import com.thefirstlineofcode.granite.framework.core.auth.Account;
 
+@Document(D_Account.COLLECTION_NAME_ACCOUNTS)
 public class D_Account extends Account implements IIdProvider<String> {
+	public static final String COLLECTION_NAME_ACCOUNTS = "accounts";
+	
+	@Id
 	private String id;
 
 	@Override

cluster/auth/src/main/java/com/thefirstlineofcode/granite/cluster/auth/DataContributor.java

@@ -0,0 +1,15 @@
+package com.thefirstlineofcode.granite.cluster.auth;
+
+import org.pf4j.Extension;
+
+import com.thefirstlineofcode.granite.framework.adf.mongodb.DataContributorAdapter;
+
+@Extension
+public class DataContributor extends DataContributorAdapter {
+	@Override
+	public Class<?>[] getDataObjects() {
+		return new Class<?>[] {
+			D_Account.class
+		};
+	}
+}

cluster/auth/src/main/java/com/thefirstlineofcode/granite/cluster/auth/DbInitializer.java

@@ -11,15 +11,16 @@
 
 @Extension
 public class DbInitializer extends AbstractDbInitializer {
+	private static final String FIELD_NAME_NAME = "name";
 
 	@Override
 	public void initialize(MongoDatabase database) {
-		if (collectionExistsInDb(database, "users"))
+		if (collectionExistsInDb(database, D_Account.COLLECTION_NAME_ACCOUNTS))
 			return;
 
-		database.createCollection("users");
-		MongoCollection<Document> users = database.getCollection("users");
-		users.createIndex(Indexes.ascending("name"), new IndexOptions().unique(true));
+		database.createCollection(D_Account.COLLECTION_NAME_ACCOUNTS);
+		MongoCollection<Document> accounts = database.getCollection(D_Account.COLLECTION_NAME_ACCOUNTS);
+		accounts.createIndex(Indexes.ascending(FIELD_NAME_NAME), new IndexOptions().unique(true));
 	}
 
 }

cluster/auth/src/main/resources/plugin.properties

@@ -1,3 +1,3 @@
 plugin.id=granite-cluster-auth
 plugin.provider=TheFirstLineOfCode
-plugin.version=1.0.4-RELEASE
+plugin.version=1.0.5-RELEASE

cluster/cluster-debug/pom.xml

@@ -8,7 +8,7 @@
 	<parent>
 		<groupId>com.thefirstlineofcode.granite</groupId>
 		<artifactId>granite-cluster</artifactId>
-		<version>1.0.4-RELEASE</version>
+		<version>1.0.5-RELEASE</version>
 		<relativePath>../pom.xml</relativePath>
 	</parent>
 

cluster/dba/pom.xml

@@ -8,7 +8,7 @@
 	<parent>
         <groupId>com.thefirstlineofcode.granite</groupId>
         <artifactId>granite-cluster</artifactId>
-	<version>1.0.4-RELEASE</version>
+	<version>1.0.5-RELEASE</version>
         <relativePath>../pom.xml</relativePath>
     </parent>
 

cluster/dba/src/main/java/com/thefirstlineofcode/granite/cluster/dba/DataObjectFactory.java

@@ -1,17 +1,95 @@
 package com.thefirstlineofcode.granite.cluster.dba;
 
+import java.lang.reflect.InvocationTargetException;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.UUID;
+
+import com.thefirstlineofcode.granite.framework.adf.mongodb.DataObjectMapping;
+import com.thefirstlineofcode.granite.framework.adf.mongodb.IDataContributor;
+import com.thefirstlineofcode.granite.framework.core.adf.IApplicationComponentService;
+import com.thefirstlineofcode.granite.framework.core.adf.IApplicationComponentServiceAware;
 import com.thefirstlineofcode.granite.framework.core.adf.data.IDataObjectFactory;
+import com.thefirstlineofcode.granite.framework.core.adf.data.IIdProvider;
 import com.thefirstlineofcode.granite.framework.core.annotations.AppComponent;
+import com.thefirstlineofcode.granite.framework.core.repository.IInitializable;
 
 @AppComponent("data.object.factory")
-public class DataObjectFactory implements IDataObjectFactory {	
-	@SuppressWarnings("unchecked")
+public class DataObjectFactory implements IInitializable, IApplicationComponentServiceAware, IDataObjectFactory {	
+	private IApplicationComponentService appComponentService;
+	private Map<Class<?>, Class<?>> dataObjectMappings;
+	private volatile boolean inited;
+	
+	public DataObjectFactory() {
+		inited = false;
+		dataObjectMappings = new HashMap<>();
+	}
+	
+	@SuppressWarnings({ "rawtypes", "unchecked" })
 	@Override
 	public <K, V extends K> V create(Class<K> clazz) {
+		if (!inited)
+			init();
+		
 		try {
-			return (V)clazz.getDeclaredConstructor().newInstance();
+			V object = doCreate(clazz);
+			
+			if (object instanceof IIdProvider) {
+				((IIdProvider)object).setId(UUID.randomUUID().toString());
+			}
+			
+			return object;
 		} catch (Exception e) {
 			throw new RuntimeException(String.format("Can't create data object for class '%s'.", clazz.getName()), e);
 		}
 	}
+
+	@SuppressWarnings("unchecked")
+	private <K, V extends K> V doCreate(Class<K> domainType) throws InstantiationException, IllegalAccessException,
+				IllegalArgumentException, InvocationTargetException, NoSuchMethodException, SecurityException {
+		Class<?> dataType = dataObjectMappings.get(domainType);
+		if (dataType == null) {
+			return (V)domainType.getDeclaredConstructor().newInstance();
+		}
+		
+		return (V)dataType.getDeclaredConstructor().newInstance();
+	}
+
+	@Override
+	public void setApplicationComponentService(IApplicationComponentService appComponentService) {
+		this.appComponentService = appComponentService;
+	}
+
+	@Override
+	public void init() {
+		if (inited)
+			return;
+		
+		synchronized (this) {
+			if (inited)
+				return;
+			
+			List<IDataContributor> dataObjectsContributors = appComponentService.getPluginManager().
+					getExtensions(IDataContributor.class);
+			for (IDataContributor dataObjectsContributor : dataObjectsContributors) {
+				DataObjectMapping<?>[] mappings = dataObjectsContributor.getDataObjectMappings();
+				if (mappings == null || mappings.length == 0)
+					continue;
+				
+				for (DataObjectMapping<?> mapping : mappings) {
+					Class<?> domainType = mapping.domainType;
+					if (domainType == null)
+						domainType = mapping.dataType.getSuperclass();
+					
+					if (dataObjectMappings.containsKey(domainType))
+						throw new IllegalArgumentException(String.format("Reduplicate domain data object type: '%s'.", mapping.domainType));
+					
+					dataObjectMappings.put(domainType, mapping.dataType);
+				}
+			}
+			
+			inited = true;
+		}
+	}
 }