*/
package org.iotivity.cloud.accountserver.db;
+import java.text.DateFormat;
+import java.text.SimpleDateFormat;
import java.util.ArrayList;
+import java.util.Date;
import org.bson.Document;
import org.iotivity.cloud.accountserver.Constants;
-import org.iotivity.cloud.util.Logger;
+import org.iotivity.cloud.util.Log;
import com.mongodb.MongoClient;
import com.mongodb.client.MongoCollection;
/**
* API creating MongoClient and initializing MongoDatabase
- *
+ *
* @param dbname
* database name to create MongoDatabase
* @throws Exception
/**
* API creating collection
- *
+ *
* @param tableName
* collection name
*/
/**
* API deleting collection
- *
+ *
* @param tableName
* collection name
*/
/**
* API getting database object
- *
+ *
*/
public MongoDatabase getMongoDatabase() {
return db;
}
- /**
- * API for storing session information of user
- *
- * @param UserSession
- * session information of user
- */
- public void createResource(UserSession userSession) {
-
- Document doc = createDocument(userSession);
+ public void createResource(UserToken userToken) {
+ Document doc = createDocument(userToken);
MongoCollection<Document> collection = db
- .getCollection(Constants.SESSION_TABLE);
-
- if (collection.findOneAndReplace(Filters.and(
- Filters.eq(Constants.USER_ID, doc.get(Constants.USER_ID)),
- Filters.eq(Constants.SESSION_CODE, doc.get(Constants.SESSION_CODE))),
+ .getCollection(Constants.TOKEN_TABLE);
+
+ if (collection.findOneAndReplace(
+ Filters.and(
+ Filters.eq(Constants.KEY_USER_ID,
+ doc.get(Constants.KEY_USER_ID)),
+ Filters.eq(Constants.KEY_ACCESS_TOKEN,
+ doc.get(Constants.KEY_ACCESS_TOKEN)),
+ Filters.eq(Constants.KEY_REFRESH_TOKEN,
+ doc.get(Constants.KEY_REFRESH_TOKEN))),
doc) == null) {
-
collection.insertOne(doc);
}
/**
* API for inserting device information of user
- *
+ *
* @param UserDevice
* device information of user
*/
MongoCollection<Document> collection = db
.getCollection(Constants.DEVICE_TABLE);
- if (collection.findOneAndReplace(Filters.and(
- Filters.eq(Constants.USER_ID, doc.get(Constants.USER_ID)),
- Filters.eq(Constants.DEVICE_ID, doc.get(Constants.DEVICE_ID))), doc) == null) {
+ if (collection.findOneAndReplace(
+ Filters.and(
+ Filters.eq(Constants.KEY_USER_ID,
+ doc.get(Constants.KEY_USER_ID)),
+ Filters.eq(Constants.KEY_DEVICE_ID,
+ doc.get(Constants.KEY_DEVICE_ID))),
+ doc) == null) {
collection.insertOne(doc);
}
/**
* API for getting user identifier corresponding with session code from
* database
- *
+ *
* @param sessionCode
* session code
* @return String - user identifier
*/
- public String getUserId(String sessionCode) {
+ public String getUserIdByAccessToken(String token) {
+
+ String userId = null;
+
+ MongoCollection<Document> collection = db
+ .getCollection(Constants.TOKEN_TABLE);
+
+ MongoCursor<Document> cursor = collection
+ .find(Filters.eq(Constants.KEY_ACCESS_TOKEN, token)).iterator();
+
+ try {
+
+ while (cursor.hasNext()) {
+
+ Document doc = cursor.next();
+ UserToken userToken = convertTokenDocToResource(doc);
+
+ userId = userToken.getUserId();
+ break;
+ }
+
+ } finally {
+
+ cursor.close();
+ }
+
+ return userId;
+ }
+
+ public String getUserIdByRefreshToken(String token) {
String userId = null;
MongoCollection<Document> collection = db
- .getCollection(Constants.SESSION_TABLE);
+ .getCollection(Constants.TOKEN_TABLE);
- MongoCursor<Document> cursor = collection.find(
- Filters.eq(Constants.SESSION_CODE, sessionCode)).iterator();
+ MongoCursor<Document> cursor = collection
+ .find(Filters.eq(Constants.KEY_REFRESH_TOKEN, token))
+ .iterator();
try {
while (cursor.hasNext()) {
Document doc = cursor.next();
- UserSession userSession = convertSessionDocToResource(doc);
+ UserToken userToken = convertTokenDocToResource(doc);
- userId = userSession.getUserId();
+ userId = userToken.getUserId();
break;
}
/**
* API for getting devices corresponding with user identifier from database
- *
+ *
* @param userId
* user identifier
*/
public ArrayList<String> getDevices(String userId) {
- ArrayList<String> deviceList = new ArrayList<String>();
+ ArrayList<String> deviceList = new ArrayList<>();
MongoCollection<Document> collection = db
.getCollection(Constants.DEVICE_TABLE);
- MongoCursor<Document> cursor = collection.find(
- Filters.eq(Constants.USER_ID, userId)).iterator();
+ MongoCursor<Document> cursor = collection
+ .find(Filters.eq(Constants.KEY_USER_ID, userId)).iterator();
try {
return deviceList;
}
+ public Boolean hasAccessToken(String token) {
+
+ Boolean hasAccessToken = false;
+
+ MongoCollection<Document> collection = db
+ .getCollection(Constants.TOKEN_TABLE);
+
+ MongoCursor<Document> cursor = collection
+ .find(Filters.eq(Constants.KEY_ACCESS_TOKEN, token)).iterator();
+
+ if (cursor.hasNext())
+ hasAccessToken = true;
+
+ cursor.close();
+
+ return hasAccessToken;
+ }
+
+ public Boolean hasRefreshToken(String token) {
+
+ Boolean hasRefreshToken = false;
+
+ MongoCollection<Document> collection = db
+ .getCollection(Constants.TOKEN_TABLE);
+
+ MongoCursor<Document> cursor = collection
+ .find(Filters.eq(Constants.KEY_REFRESH_TOKEN, token))
+ .iterator();
+
+ if (cursor.hasNext())
+ hasRefreshToken = true;
+
+ cursor.close();
+
+ return hasRefreshToken;
+ }
+
+ public Boolean updateResource(UserToken oldUserToken,
+ UserToken newUserToken) {
+
+ Boolean updateResource = false;
+ String userId = oldUserToken.getUserId();
+ String oldRefreshToken = oldUserToken.getRefreshToken();
+
+ Document doc = createDocument(newUserToken);
+
+ MongoCollection<Document> collection = db
+ .getCollection(Constants.TOKEN_TABLE);
+
+ // update
+ if (collection.findOneAndReplace(
+ Filters.and(Filters.eq(Constants.KEY_USER_ID, userId), Filters
+ .eq(Constants.KEY_REFRESH_TOKEN, oldRefreshToken)),
+ doc) != null) {
+
+ // collection.insertOne(doc);
+ updateResource = true;
+
+ } else {
+ Log.e("UpdateResource failed!");
+ }
+
+ return updateResource;
+ }
+
+ public String getIssuedTime(String accessToken) {
+
+ MongoCollection<Document> collection = db
+ .getCollection(Constants.TOKEN_TABLE);
+
+ MongoCursor<Document> cursor = collection
+ .find(Filters.eq(Constants.KEY_ACCESS_TOKEN, accessToken))
+ .iterator();
+
+ String issuedTime = null;
+
+ try {
+
+ while (cursor.hasNext()) {
+
+ Document doc = cursor.next();
+ UserToken userToken = convertTokenDocToResource(doc);
+
+ issuedTime = userToken.getIssuedTime();
+ break;
+ }
+
+ } finally {
+
+ cursor.close();
+ }
+
+ return issuedTime;
+
+ }
+
public void printResources() {
ArrayList<UserDevice> dlist = readDeviceResources();
int size = dlist.size();
- Logger.i("*Table: " + Constants.DEVICE_TABLE);
+ Log.i("[" + Constants.DEVICE_TABLE + "]Table");
for (int i = 0; i < size; i++) {
UserDevice item = dlist.get(i);
- Logger.i("[" + i + "]" + item.getUserId() + ", "
+ Log.i("[" + i + "]" + item.getUserId() + ", "
+ item.getDeviceId());
}
- ArrayList<UserSession> slist = readSessionResources();
- size = slist.size();
-
- Logger.i("*Table: " + Constants.SESSION_TABLE);
+ /*
+ * ArrayList<UserSession> slist = readSessionResources(); size =
+ * slist.size();
+ *
+ * Log.i("*Table: " + Constants.SESSION_TABLE);
+ *
+ * for (int i = 0; i < size; i++) {
+ *
+ * UserSession item = slist.get(i);
+ *
+ * Log.i("[" + i + "]" + item.getUserId() + ", " +
+ * item.getSessionCode());
+ *
+ * }
+ */
+
+ ArrayList<UserToken> tlist = readUserTokenResources();
+ size = tlist.size();
+
+ Log.i("[" + Constants.TOKEN_TABLE + "]Table");
for (int i = 0; i < size; i++) {
- UserSession item = slist.get(i);
+ UserToken item = tlist.get(i);
- Logger.i("[" + i + "]" + item.getUserId() + ", "
- + item.getSessionCode());
+ Log.i("[" + i + "]" + item.getUserId() + "/"
+ + item.getAccessToken() + "/" + item.getRefreshToken() + "/"
+ + item.getIssuedTime());
}
}
- private Document createDocument(UserSession userSession) {
+ private Document createDocument(UserToken userToken) {
+
+ String userId = userToken.getUserId();
+ String accessToken = userToken.getAccessToken();
+ String refreshToken = userToken.getRefreshToken();
+
+ DateFormat f = new SimpleDateFormat("yyyyMMddkkmm");
+ Date currentDate = new Date();
- Document doc = new Document(Constants.USER_ID, userSession.getUserId())
- .append(Constants.SESSION_CODE, userSession.getSessionCode());
+ String issuedTime = f.format(currentDate);
+
+ Document doc = new Document(Constants.KEY_USER_ID, userId)
+ .append(Constants.KEY_ACCESS_TOKEN, accessToken)
+ .append(Constants.KEY_REFRESH_TOKEN, refreshToken)
+ .append(Constants.KEY_ISSUED_TIME, issuedTime);
return doc;
}
private Document createDocument(UserDevice userDevice) {
- Document doc = new Document(Constants.USER_ID, userDevice.getUserId())
- .append(Constants.DEVICE_ID, userDevice.getDeviceId());
+ Document doc = new Document(Constants.KEY_USER_ID,
+ userDevice.getUserId()).append(Constants.KEY_DEVICE_ID,
+ userDevice.getDeviceId());
return doc;
}
- private UserSession convertSessionDocToResource(Document doc) {
+ private UserToken convertTokenDocToResource(Document doc) {
+
+ UserToken userToken = new UserToken();
- UserSession userSession = new UserSession();
+ String userId = doc.getString(Constants.KEY_USER_ID);
+ String accessToken = doc.getString(Constants.KEY_ACCESS_TOKEN);
+ String refreshToken = doc.getString(Constants.KEY_REFRESH_TOKEN);
+ String issuedTime = doc.getString(Constants.KEY_ISSUED_TIME);
- userSession.setUserId(doc.getString(Constants.USER_ID));
- userSession.setSessionCode(doc.getString(Constants.SESSION_CODE));
+ // Log.d("issuedTime: " + issuedTime);
+ userToken.setUserToken(userId, accessToken, refreshToken);
+ userToken.setIssuedTime(issuedTime);
- return userSession;
+ return userToken;
}
private UserDevice convertDeviceDocToResource(Document doc) {
UserDevice userDevice = new UserDevice();
- userDevice.setUserId(doc.getString(Constants.USER_ID));
- userDevice.setDeviceId(doc.getString(Constants.DEVICE_ID));
+ userDevice.setUserId(doc.getString(Constants.KEY_USER_ID));
+ userDevice.setDeviceId(doc.getString(Constants.KEY_DEVICE_ID));
return userDevice;
}
- private ArrayList<UserSession> readSessionResources() {
+ private ArrayList<UserToken> readUserTokenResources() {
- ArrayList<UserSession> userSessionList = new ArrayList<UserSession>();
+ ArrayList<UserToken> userTokenList = new ArrayList<>();
MongoCollection<Document> collection = db
- .getCollection(Constants.SESSION_TABLE);
+ .getCollection(Constants.TOKEN_TABLE);
MongoCursor<Document> cursor = collection.find().iterator();
while (cursor.hasNext()) {
Document doc = cursor.next();
- userSessionList.add(convertSessionDocToResource(doc));
+ userTokenList.add(convertTokenDocToResource(doc));
}
cursor.close();
- return userSessionList;
+ return userTokenList;
}
private ArrayList<UserDevice> readDeviceResources() {
- ArrayList<UserDevice> userDeviceList = new ArrayList<UserDevice>();
+ ArrayList<UserDevice> userDeviceList = new ArrayList<>();
MongoCollection<Document> collection = db
.getCollection(Constants.DEVICE_TABLE);
projects / platform / upstream / iotivity.git / blobdiff