/*
* Copyright 2009 Google Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.gtugs.web.reporting;
import com.google.appengine.api.datastore.Cursor;
import com.google.appengine.api.labs.taskqueue.Queue;
import com.google.appengine.api.labs.taskqueue.QueueFactory;
import com.google.appengine.api.labs.taskqueue.TaskOptions;
import org.datanucleus.store.appengine.query.JDOCursorHelper;
import org.gtugs.domain.Chapter;
import org.gtugs.repository.JdoStatsDao;
import org.gtugs.repository.PMF;
import org.json.simple.JSONObject;
import org.json.simple.JSONValue;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.servlet.mvc.Controller;
import org.springframework.web.servlet.ModelAndView;
import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import javax.jdo.PersistenceManager;
import javax.jdo.Query;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
/**
* @author jasonacooper@google.com (Jason Cooper)
*/
@org.springframework.stereotype.Controller
public class GetChapterStatsController {
private static final String CURSOR_KEY = "cursor";
@Autowired
private JdoStatsDao statsDao;
@RequestMapping("/offline/getChapterStats")
public ModelAndView handleRequest(HttpServletRequest request,
HttpServletResponse response) throws ServletException, IOException {
Cursor cursor = null;
if (request.getParameter(CURSOR_KEY) != null) {
cursor = Cursor.fromWebSafeString(request.getParameter(CURSOR_KEY));
}
PersistenceManager pm = PMF.get().getPersistenceManager();
Query query = pm.newQuery(Chapter.class);
query.setRange(0, 20);
if (cursor != null) {
Map<String, Object> extensionMap = new HashMap<String, Object>();
extensionMap.put(JDOCursorHelper.CURSOR_EXTENSION, cursor);
query.setExtensions(extensionMap);
}
List<Chapter> chapterResults = (List<Chapter>) query.execute();
List<Chapter> chapters = new ArrayList<Chapter>(chapterResults.size());
for (Chapter result : chapterResults) {
if (result.getStatus().compareToIgnoreCase("inactive") != 0) {
chapters.add(result);
}
}
String cursorString =
JDOCursorHelper.getCursor(chapterResults).toWebSafeString();
TaskOptions options = TaskOptions.Builder
.url("/offline/getChapterStats")
.param(CURSOR_KEY, cursorString);
countChapters(request, chapters, options);
countCountries(request, chapters, options);
if (chapters != null && chapters.size() > 0) {
Queue queue = QueueFactory.getDefaultQueue();
queue.add(options);
}
query.closeAll();
pm.close();
return null;
}
public void setStatsDao(JdoStatsDao statsDao) {
this.statsDao = statsDao;
}
private void countChapters(HttpServletRequest request,
List<Chapter> chapters, TaskOptions options) {
final String KEY = "chapterCount";
int chapterCount = 0;
if (request.getParameter(KEY) != null) {
chapterCount = Integer.parseInt(request.getParameter(KEY));
}
if (chapters == null || chapters.size() == 0) {
statsDao.updateField(JdoStatsDao.Field.NUM_CHAPTERS, chapterCount);
} else {
for (Chapter chapter : chapters) {
chapterCount++;
}
options = options.param(KEY, "" + chapterCount);
}
}
private void countCountries(HttpServletRequest request,
List<Chapter> chapters, TaskOptions options) {
final String KEY = "countryCount";
JSONObject countries = null;
if (request.getParameter(KEY) != null) {
countries = (JSONObject) JSONValue.parse(request.getParameter(KEY));
} else {
countries = new JSONObject();
}
if (chapters == null || chapters.size() == 0) {
statsDao.updateField(JdoStatsDao.Field.NUM_COUNTRIES, countries.size());
} else {
for (Chapter chapter : chapters) {
countries.put(chapter.getCountry(), Boolean.TRUE);
}
options = options.param(KEY, countries.toJSONString());
}
}
}