package org.activityinfo.server.command.handler;
/*
* #%L
* ActivityInfo Server
* %%
* Copyright (C) 2009 - 2013 UNICEF
* %%
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public
* License along with this program. If not, see
* <http://www.gnu.org/licenses/gpl-3.0.html>.
* #L%
*/
import com.bedatadriven.rebar.sql.client.SqlResultCallback;
import com.bedatadriven.rebar.sql.client.SqlResultSet;
import com.bedatadriven.rebar.sql.client.SqlResultSetRow;
import com.bedatadriven.rebar.sql.client.SqlTransaction;
import com.bedatadriven.rebar.sql.client.query.SqlQuery;
import com.google.appengine.api.memcache.Expiration;
import com.google.appengine.api.memcache.MemcacheService;
import com.google.appengine.api.memcache.MemcacheServiceFactory;
import com.google.common.base.Preconditions;
import com.google.common.collect.Lists;
import com.google.gwt.user.client.rpc.AsyncCallback;
import com.google.inject.Inject;
import org.activityinfo.legacy.shared.command.GetReportModel;
import org.activityinfo.legacy.shared.exception.UnexpectedCommandException;
import org.activityinfo.legacy.shared.impl.CommandHandlerAsync;
import org.activityinfo.legacy.shared.impl.ExecutionContext;
import org.activityinfo.legacy.shared.impl.Tables;
import org.activityinfo.legacy.shared.model.ReportDTO;
import org.activityinfo.legacy.shared.model.ReportMetadataDTO;
import org.activityinfo.legacy.shared.reports.model.EmailDelivery;
import org.activityinfo.legacy.shared.reports.model.Report;
import org.activityinfo.server.database.hibernate.entity.ReportDefinition;
import org.activityinfo.server.report.ReportParserJaxb;
import javax.persistence.EntityManager;
import javax.xml.bind.JAXBException;
import java.util.List;
import java.util.logging.Level;
import java.util.logging.Logger;
public class GetReportModelHandler implements CommandHandlerAsync<GetReportModel, ReportDTO> {
private static final Logger LOGGER = Logger.getLogger(GetReportModelHandler.class.getName());
private final MemcacheService memcacheService = MemcacheServiceFactory.getMemcacheService();
private final EntityManager em;
@Inject
public GetReportModelHandler(final EntityManager em) {
this.em = em;
}
@Override
public void execute(final GetReportModel cmd,
final ExecutionContext context,
final AsyncCallback<ReportDTO> callback) {
LOGGER.finest("Loading model for report id = " + cmd.getReportId());
Preconditions.checkNotNull(cmd.getReportId());
ReportDTO cachedReport = (ReportDTO) memcacheService.get(cmd);
if (cachedReport != null) {
callback.onSuccess(cachedReport);
return;
}
// always load report
ReportDefinition entity = em.find(ReportDefinition.class, cmd.getReportId());
Report report = parseReport(entity)
.setId(cmd.getReportId());
ReportDTO reportDTO = new ReportDTO(report);
if (cmd.isLoadMetadata()) {
loadMetadataAndCallback(cmd, context, reportDTO, callback);
} else {
// report object without metadata
memcache(cmd, reportDTO);
callback.onSuccess(reportDTO);
}
}
private void memcache(GetReportModel cmd, ReportDTO report) {
try {
memcacheService.put(cmd, report, Expiration.byDeltaSeconds(60), MemcacheService.SetPolicy.ADD_ONLY_IF_NOT_PRESENT);
} catch (Exception e) {
LOGGER.log(Level.WARNING, "Exception putting report model to memcache", e);
}
}
private Report parseReport(ReportDefinition entity) {
try {
LOGGER.finest("Starting to parse xml (size = " + entity.getXml().length() + ")");
return ReportParserJaxb.parseXml(entity.getXml());
} catch (JAXBException e) {
throw new UnexpectedCommandException(e);
}
}
private void loadMetadataAndCallback(final GetReportModel cmd,
final ExecutionContext context,
final ReportDTO reportDTO,
final AsyncCallback<ReportDTO> callback) {
final int userId = context.getUser().getId();
SqlQuery mySubscriptions = SqlQuery.selectAll().from("reportsubscription").where("userId").equalTo(userId);
SqlQuery myDatabases = SqlQuery.selectSingle("d.databaseid")
.from("userdatabase", "d")
.leftJoin(SqlQuery.selectAll()
.from(Tables.USER_PERMISSION, "UserPermission")
.where("UserPermission.UserId")
.equalTo(userId), "p")
.on("p.DatabaseId = d.DatabaseId")
.where("d.ownerUserId")
.equalTo(userId)
.or("p.AllowView")
.equalTo(1);
SqlQuery.select()
.appendColumn("r.reportTemplateId", "reportId")
.appendColumn("r.title", "title")
.appendColumn("r.ownerUserId", "ownerUserId")
.appendColumn("o.name", "ownerName")
.appendColumn("s.dashboard", "dashboard")
.appendColumn("s.emaildelivery", "emaildelivery")
.appendColumn("s.emailday", "emailday")
.appendColumn(SqlQuery.selectSingle("max(defaultDashboard)")
.from("reportvisibility", "v")
.where("v.databaseId")
.in(myDatabases)
.whereTrue("v.reportid = r.reportTemplateId"), "defaultDashboard")
.from("reporttemplate", "r")
.leftJoin("userlogin o")
.on("o.userid = r.ownerUserId")
.leftJoin(mySubscriptions, "s")
.on("r.reportTemplateId = s.reportId")
.where("r.ownerUserId")
.equalTo(userId)
.where("r.reportTemplateId")
.equalTo(cmd.getReportId())
.execute(context.getTransaction(), new SqlResultCallback() {
@Override
public void onSuccess(final SqlTransaction tx, final SqlResultSet results) {
List<ReportMetadataDTO> dtos = Lists.newArrayList();
for (SqlResultSetRow row : results.getRows()) {
ReportMetadataDTO dto = new ReportMetadataDTO();
dto.setId(row.getInt("reportId"));
dto.setAmOwner(row.getInt("ownerUserId") == userId);
dto.setOwnerName(row.getString("ownerName"));
dto.setTitle(row.getString("title"));
dto.setEditAllowed(dto.getAmOwner());
if (!row.isNull("emaildelivery")) {
dto.setEmailDelivery(EmailDelivery.valueOf(row.getString("emaildelivery")));
}
if (row.isNull("emailday")) {
dto.setDay(1);
} else {
dto.setDay(row.getInt("emailday"));
}
if (row.isNull("dashboard")) {
// inherited from database-wide visibility
dto.setDashboard(!row.isNull("defaultDashboard") && row.getBoolean("defaultDashboard"));
} else {
dto.setDashboard(row.getBoolean("dashboard"));
}
dtos.add(dto);
}
if (dtos.size() == 0) {
ReportMetadataDTO dummy = new ReportMetadataDTO();
dummy.setId(cmd.getReportId());
dtos.add(dummy);
}
// there should be only one result
reportDTO.setReportMetadataDTO(dtos.get(0));
// exit handler with both the report and metadata objects
// filled
memcache(cmd, reportDTO);
callback.onSuccess(reportDTO);
}
});
}
}