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.mobicents.servlet.restcomm.dao.mybatis; import static org.mobicents.servlet.restcomm.dao.DaoUtils.readDateTime; import static org.mobicents.servlet.restcomm.dao.DaoUtils.readDouble; import static org.mobicents.servlet.restcomm.dao.DaoUtils.readSid; import static org.mobicents.servlet.restcomm.dao.DaoUtils.readString; import static org.mobicents.servlet.restcomm.dao.DaoUtils.readUri; import static org.mobicents.servlet.restcomm.dao.DaoUtils.writeDateTime; import static org.mobicents.servlet.restcomm.dao.DaoUtils.writeSid; import static org.mobicents.servlet.restcomm.dao.DaoUtils.writeUri; import java.net.URI; import java.net.URISyntaxException; import java.util.ArrayList; import java.util.HashMap; import java.util.List; import java.util.Map; import org.apache.ibatis.session.SqlSession; import org.apache.ibatis.session.SqlSessionFactory; import org.joda.time.DateTime; import org.mobicents.servlet.restcomm.amazonS3.S3AccessTool; import org.mobicents.servlet.restcomm.annotations.concurrency.ThreadSafe; import org.mobicents.servlet.restcomm.dao.RecordingsDao; import org.mobicents.servlet.restcomm.entities.Recording; import org.mobicents.servlet.restcomm.entities.Sid; import org.mobicents.servlet.restcomm.util.UriUtils; /** * @author quintana.thomas@gmail.com (Thomas Quintana) */ @ThreadSafe public final class MybatisRecordingsDao implements RecordingsDao { private static final String namespace = "org.mobicents.servlet.sip.restcomm.dao.RecordingsDao."; private final SqlSessionFactory sessions; private S3AccessTool s3AccessTool; private String recordingPath; public MybatisRecordingsDao(final SqlSessionFactory sessions) { super(); this.sessions = sessions; } public MybatisRecordingsDao(final SqlSessionFactory sessions, final S3AccessTool s3AccessTool, final String recordingPath) { super(); this.sessions = sessions; this.s3AccessTool = s3AccessTool; this.recordingPath = recordingPath; } @Override public void addRecording(Recording recording) { if (s3AccessTool != null) { URI s3Uri = s3AccessTool.uploadFile(recordingPath + "/" + recording.getSid().toString() + ".wav"); if (s3Uri != null) { recording = recording.updateFileUri(s3Uri); } } else { recording = recording.updateFileUri(generateLocalFileUri("/restcomm/recordings/" + recording.getSid())); } final SqlSession session = sessions.openSession(); try { session.insert(namespace + "addRecording", toMap(recording)); session.commit(); } finally { session.close(); } } public URI generateLocalFileUri(String recordingRelativeUri) { URI uriToResolve = null; try { //For local stored recordings, add .wav suffix to the URI uriToResolve = new URI(recordingRelativeUri + ".wav"); } catch (URISyntaxException e) { } return UriUtils.resolve(uriToResolve); } @Override public Recording getRecording(final Sid sid) { return getRecording(namespace + "getRecording", sid); } @Override public Recording getRecordingByCall(final Sid callSid) { return getRecording(namespace + "getRecordingByCall", callSid); } private Recording getRecording(final String selector, final Sid sid) { final SqlSession session = sessions.openSession(); try { final Map<String, Object> result = session.selectOne(selector, sid.toString()); if (result != null) { return toRecording(result); } else { return null; } } finally { session.close(); } } @Override public List<Recording> getRecordings(final Sid accountSid) { final SqlSession session = sessions.openSession(); try { final List<Map<String, Object>> results = session.selectList(namespace + "getRecordings", accountSid.toString()); final List<Recording> recordings = new ArrayList<Recording>(); if (results != null && !results.isEmpty()) { for (final Map<String, Object> result : results) { recordings.add(toRecording(result)); } } return recordings; } finally { session.close(); } } @Override public void removeRecording(final Sid sid) { removeRecording(namespace + "removeRecording", sid); } @Override public void removeRecordings(final Sid accountSid) { removeRecording(namespace + "removeRecordings", accountSid); } private void removeRecording(final String selector, final Sid sid) { final SqlSession session = sessions.openSession(); try { session.delete(selector, sid.toString()); session.commit(); } finally { session.close(); } } private Map<String, Object> toMap(final Recording recording) { final Map<String, Object> map = new HashMap<String, Object>(); map.put("sid", writeSid(recording.getSid())); map.put("date_created", writeDateTime(recording.getDateCreated())); map.put("date_updated", writeDateTime(recording.getDateUpdated())); map.put("account_sid", writeSid(recording.getAccountSid())); map.put("call_sid", writeSid(recording.getCallSid())); map.put("duration", recording.getDuration()); map.put("api_version", recording.getApiVersion()); map.put("uri", writeUri(recording.getUri())); map.put("file_uri", writeUri(recording.getFileUri())); return map; } private Recording toRecording(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 Sid accountSid = readSid(map.get("account_sid")); final Sid callSid = readSid(map.get("call_sid")); final Double duration = readDouble(map.get("duration")); final String apiVersion = readString(map.get("api_version")); final URI uri = readUri(map.get("uri")); //For backward compatibility. For old an database that we upgraded to the latest schema, the file_uri will be null so we need //to create the file_uri on the fly String fileUri = (String) map.get("file_uri"); if (fileUri == null || fileUri.isEmpty()) { fileUri = generateLocalFileUri("/restcomm/recordings/" + sid).toString(); } return new Recording(sid, dateCreated, dateUpdated, accountSid, callSid, duration, apiVersion, uri, readUri(fileUri)); } }