Java tutorial
/* * TeleStax, Open Source Cloud Communications * Copyright 2011-2014, Telestax Inc and individual contributors * by the @authors tag. * * This program is free software: you can redistribute it and/or modify * under the terms of the GNU Affero General Public License as * published by the Free Software Foundation; either version 3 of * the License, or (at your option) any later version. * * This program 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 Affero General Public License for more details. * * You should have received a copy of the GNU Affero General Public License * along with this program. If not, see <http://www.gnu.org/licenses/> * */ package org.restcomm.connect.dao.mybatis; import org.apache.ibatis.session.SqlSession; import org.apache.ibatis.session.SqlSessionFactory; import org.joda.time.DateTime; import org.restcomm.connect.dao.exceptions.AccountHierarchyDepthCrossed; import org.restcomm.connect.commons.annotations.concurrency.ThreadSafe; import org.restcomm.connect.dao.AccountsDao; import org.restcomm.connect.dao.entities.Account; import org.restcomm.connect.commons.dao.Sid; import java.net.URI; import java.util.ArrayList; import java.util.HashMap; import java.util.List; import java.util.Map; import static org.restcomm.connect.dao.DaoUtils.readAccountStatus; import static org.restcomm.connect.dao.DaoUtils.readAccountType; import static org.restcomm.connect.dao.DaoUtils.readDateTime; import static org.restcomm.connect.dao.DaoUtils.readSid; import static org.restcomm.connect.dao.DaoUtils.readString; import static org.restcomm.connect.dao.DaoUtils.readUri; import static org.restcomm.connect.dao.DaoUtils.writeAccountStatus; import static org.restcomm.connect.dao.DaoUtils.writeAccountType; import static org.restcomm.connect.dao.DaoUtils.writeDateTime; import static org.restcomm.connect.dao.DaoUtils.writeSid; import static org.restcomm.connect.dao.DaoUtils.writeUri; /** * @author quintana.thomas@gmail.com (Thomas Quintana) */ @ThreadSafe public final class MybatisAccountsDao implements AccountsDao { private static final String namespace = "org.mobicents.servlet.sip.restcomm.dao.AccountsDao."; private Integer accountRecursionDepth = 3; // maximum value for recursive account queries private final SqlSessionFactory sessions; public MybatisAccountsDao(final SqlSessionFactory sessions) { super(); this.sessions = sessions; } public void setAccountRecursionDepth(Integer accountRecursionDepth) { this.accountRecursionDepth = accountRecursionDepth; } @Override public void addAccount(final Account account) { final SqlSession session = sessions.openSession(); try { session.insert(namespace + "addAccount", toMap(account)); session.commit(); } finally { session.close(); } } @Override public Account getAccount(final Sid sid) { return getAccount(namespace + "getAccount", sid.toString()); } @Override public Account getAccount(final String name) { Account account = null; account = getAccount(namespace + "getAccountByFriendlyName", name); if (account == null) { account = getAccount(namespace + "getAccountByEmail", name); } if (account == null) { account = getAccount(namespace + "getAccount", name); } return account; } @Override public Account getAccountToAuthenticate(final String name) { Account account = null; account = getAccount(namespace + "getAccountByEmail", name); if (account == null) { account = getAccount(namespace + "getAccount", name); } return account; } private Account getAccount(final String selector, final Object parameters) { final SqlSession session = sessions.openSession(); try { final Map<String, Object> result = session.selectOne(selector, parameters); if (result != null) { return toAccount(result); } else { return null; } } finally { session.close(); } } @Override public List<Account> getChildAccounts(final Sid parentSid) { final SqlSession session = sessions.openSession(); try { final List<Map<String, Object>> results = session.selectList(namespace + "getChildAccounts", parentSid.toString()); final List<Account> accounts = new ArrayList<Account>(); if (results != null && !results.isEmpty()) { for (final Map<String, Object> result : results) { accounts.add(toAccount(result)); } } return accounts; } finally { session.close(); } } @Override public void removeAccount(final Sid sid) { removeAccount(namespace + "removeAccount", sid); } private void removeAccount(final String selector, final Sid sid) { final SqlSession session = sessions.openSession(); try { session.delete(selector, sid.toString()); session.commit(); } finally { session.close(); } } @Override public void updateAccount(final Account account) { updateAccount(namespace + "updateAccount", account); } @Override public List<String> getSubAccountSidsRecursive(Sid parentAccountSid) { List<String> parentList = new ArrayList<String>(); parentList.add(parentAccountSid.toString()); List<String> allChildren = new ArrayList<String>(); int depth = 1; List<String> childrenList = getSubAccountsSids(parentList); while (childrenList != null && !childrenList.isEmpty() && depth <= accountRecursionDepth) { allChildren.addAll(childrenList); childrenList = getSubAccountsSids(childrenList); // retrieve children's children depth++; } return allChildren; } @Override public List<String> getAccountLineage(Sid accountSid) throws AccountHierarchyDepthCrossed { if (accountSid == null) return null; List<String> ancestorList = new ArrayList<String>(); Sid sid = accountSid; Account account = getAccount(sid); if (account == null) throw new IllegalArgumentException( "Wrong accountSid is given to search for ancestor on it. This account does not even exist"); int depth = 1; // already having one-level of accounts while (true) { Sid parentSid = account.getParentSid(); if (parentSid != null) { depth++; if (depth > accountRecursionDepth) throw new AccountHierarchyDepthCrossed(); ancestorList.add(parentSid.toString()); Account parentAccount = getAccount(parentSid); if (parentAccount == null) throw new IllegalStateException("Parent account " + parentSid.toString() + " does not exist although its child does " + account.getSid().toString()); account = parentAccount; } else break; } return ancestorList; } @Override public List<String> getAccountLineage(Account account) throws AccountHierarchyDepthCrossed { if (account == null) return null; List<String> lineage = new ArrayList<String>(); Sid parentSid = account.getParentSid(); if (parentSid != null) { lineage.add(parentSid.toString()); lineage.addAll(getAccountLineage(parentSid)); } return lineage; } private List<String> getSubAccountsSids(List<String> parentAccountSidList) { final SqlSession session = sessions.openSession(); try { final List<String> results = session.selectList(namespace + "getSubAccountSids", parentAccountSidList); return results; } finally { session.close(); } } private void updateAccount(final String selector, final Account account) { final SqlSession session = sessions.openSession(); try { session.update(selector, toMap(account)); session.commit(); } finally { session.close(); } } private Account toAccount(final Map<String, Object> map) { final Sid sid = readSid(map.get("sid")); final DateTime dateCreated = readDateTime(map.get("date_created")); final DateTime dateUpdated = readDateTime(map.get("date_updated")); final String emailAddress = readString(map.get("email_address")); final String friendlyName = readString(map.get("friendly_name")); final Sid parentSid = readSid(map.get("parent_sid")); final Account.Type type = readAccountType(map.get("type")); final Account.Status status = readAccountStatus(map.get("status")); final String authToken = readString(map.get("auth_token")); final String role = readString(map.get("role")); final URI uri = readUri(map.get("uri")); return new Account(sid, dateCreated, dateUpdated, emailAddress, friendlyName, parentSid, type, status, authToken, role, uri); } private Map<String, Object> toMap(final Account account) { final Map<String, Object> map = new HashMap<String, Object>(); map.put("sid", writeSid(account.getSid())); map.put("date_created", writeDateTime(account.getDateCreated())); map.put("date_updated", writeDateTime(account.getDateUpdated())); map.put("email_address", account.getEmailAddress()); map.put("friendly_name", account.getFriendlyName()); map.put("parent_sid", writeSid(account.getParentSid())); map.put("type", writeAccountType(account.getType())); map.put("status", writeAccountStatus(account.getStatus())); map.put("auth_token", account.getAuthToken()); map.put("role", account.getRole()); map.put("uri", writeUri(account.getUri())); return map; } }