Java tutorial
/* * This is free software; you can redistribute it and/or modify it * under the terms of the GNU Lesser General Public License as * published by the Free Software Foundation; either version 2.1 of * the License, or (at your option) any later version. * * This software is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this software; if not, write to the Free * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA * 02110-1301 USA, or see the FSF site: http://www.fsf.org. */ package org.mobicents.servlet.sip.restcomm.dao.mongodb; import static org.mobicents.servlet.sip.restcomm.dao.DaoUtils.readAccountStatus; import static org.mobicents.servlet.sip.restcomm.dao.DaoUtils.readAccountType; import static org.mobicents.servlet.sip.restcomm.dao.DaoUtils.readDateTime; import static org.mobicents.servlet.sip.restcomm.dao.DaoUtils.readSid; import static org.mobicents.servlet.sip.restcomm.dao.DaoUtils.readString; import static org.mobicents.servlet.sip.restcomm.dao.DaoUtils.readUri; import static org.mobicents.servlet.sip.restcomm.dao.DaoUtils.writeAccountStatus; import static org.mobicents.servlet.sip.restcomm.dao.DaoUtils.writeAccountType; import static org.mobicents.servlet.sip.restcomm.dao.DaoUtils.writeDateTime; import static org.mobicents.servlet.sip.restcomm.dao.DaoUtils.writeSid; import static org.mobicents.servlet.sip.restcomm.dao.DaoUtils.writeUri; import java.net.URI; import java.util.ArrayList; import java.util.List; import org.apache.log4j.Logger; import org.joda.time.DateTime; import org.mobicents.servlet.sip.restcomm.Sid; import org.mobicents.servlet.sip.restcomm.annotations.concurrency.ThreadSafe; import org.mobicents.servlet.sip.restcomm.dao.AccountsDao; import org.mobicents.servlet.sip.restcomm.entities.Account; import com.mongodb.BasicDBObject; import com.mongodb.DB; import com.mongodb.DBCollection; import com.mongodb.DBCursor; import com.mongodb.DBObject; import com.mongodb.WriteResult; /** * @author quintana.thomas@gmail.com (Thomas Quintana) */ @ThreadSafe public final class MongoAccountsDao implements AccountsDao { private static final Logger logger = Logger.getLogger(MongoAccountsDao.class); private final DBCollection accounts; public MongoAccountsDao(final DB database) { super(); accounts = database.getCollection("restcomm_accounts"); } @Override public void addAccount(final Account account) { final WriteResult result = accounts.insert(toDbObject(account)); if (!result.getLastError().ok()) { logger.error(result.getLastError().getErrorMessage()); } } @Override public Account getAccount(final Sid sid) { return getAccount(accounts, sid); } private Account getAccount(final DBCollection collection, final Sid sid) { final BasicDBObject query = new BasicDBObject(); query.put("sid", sid.toString()); final DBObject result = collection.findOne(query); if (result != null) { return toAccount(result); } else { return null; } } @Override public List<Account> getAccounts(final Sid sid) { final BasicDBObject query = new BasicDBObject(); query.put("account_sid", sid.toString()); final List<Account> result = new ArrayList<Account>(); final DBCursor results = accounts.find(query); while (results.hasNext()) { result.add(toAccount(results.next())); } return result; } @Override public void removeAccount(final Sid sid) { removeAccount(accounts, sid); } private void removeAccount(final DBCollection collection, final Sid sid) { final BasicDBObject query = new BasicDBObject(); query.put("sid", sid.toString()); final WriteResult result = collection.remove(query); if (!result.getLastError().ok()) { logger.error(result.getLastError().getErrorMessage()); } } @Override public void updateAccount(final Account account) { updateAccount(accounts, account); } private void updateAccount(final DBCollection collection, final Account account) { final BasicDBObject query = new BasicDBObject(); query.put("sid", account.getSid().toString()); final WriteResult result = collection.update(query, toDbObject(account)); if (!result.getLastError().ok()) { logger.error(result.getLastError().getErrorMessage()); } } private Account toAccount(final DBObject object) { final Sid sid = readSid(object.get("sid")); final DateTime dateCreated = readDateTime(object.get("date_created")); final DateTime dateUpdated = readDateTime(object.get("date_updated")); final String emailAddress = readString(object.get("email_address")); final String friendlyName = readString(object.get("friendly_name")); final Sid accountSid = readSid(object.get("account_sid")); final Account.Type type = readAccountType((String) object.get("type")); final Account.Status status = readAccountStatus((String) object.get("status")); final String authToken = readString(object.get("auth_token")); final String role = readString(object.get("role")); final URI uri = readUri(object.get("uri")); return new Account(sid, dateCreated, dateUpdated, emailAddress, friendlyName, accountSid, type, status, authToken, role, uri); } private DBObject toDbObject(final Account account) { final BasicDBObject object = new BasicDBObject(); object.put("sid", writeSid(account.getSid())); object.put("date_created", writeDateTime(account.getDateCreated())); object.put("date_updated", writeDateTime(account.getDateUpdated())); object.put("email_address", account.getEmailAddress()); object.put("friendly_name", account.getFriendlyName()); object.put("account_sid", writeSid(account.getAccountSid())); object.put("type", writeAccountType(account.getType())); object.put("status", writeAccountStatus(account.getStatus())); object.put("auth_token", account.getAuthToken()); object.put("role", account.getRole()); object.put("uri", writeUri(account.getUri())); return object; } }