FChartDataSource.java

/*
 *  Copyright 2012 The Solmix Project
 *
 * 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 may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * http://www.gnu.org/licenses/ 
 * or see the FSF site: http://www.fsf.org. 
 */

package com.smartgwt.extensions.fusionchart.server;

import java.util.Map;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.solmix.api.datasource.DSRequest;
import org.solmix.api.datasource.DSResponse;
import org.solmix.api.datasource.DSResponse.Status;
import org.solmix.api.datasource.DataSource;
import org.solmix.api.datasource.DataSourceManager;
import org.solmix.api.exception.SlxException;
import org.solmix.api.jaxb.Tobject;
import org.solmix.api.jaxb.ToperationBinding;
import org.solmix.fmk.SlxContext;
import org.solmix.fmk.util.XMLUtil;
import org.solmix.fmk.velocity.Velocity;

/**
 * 
 * @author solmix.f@gmail.com
 * @version 110035 2013-1-9
 */
@SuppressWarnings({ "unchecked", "rawtypes" })
public  class FChartDataSource
{

    public static final String TEMPLATE = "template";

    public static final String TEMPLATE_FILE = "template-file";

    private static final Logger log = LoggerFactory.getLogger(FChartDataSource.class.getName());

    public DSResponse fetch(DSRequest req, DataSource ds) throws SlxException {
        ToperationBinding bind = ds.getContext().getOperationBinding(req);
        Map<String, Object> critera = req.getContext().getCriteria();
        String characterEncoding = "UTF-8";
        if (critera != null && critera.get("characterEncoding") != null) {
            characterEncoding = critera.get("characterEncoding").toString();
        }
        Tobject object = bind.getConfiguration();
        Map<String, Object> map = XMLUtil.toMap(object);
        DSResponse response;
        Object data = getData(req, ds);
        response = SlxContext.getThreadSystemContext().getExtension(DataSourceManager.class).createDSResponse(req);
        response.setRawData(data);
        Map context = Velocity.getStandardContextMap(req);
        context.putAll(map);
        context.put(Velocity.RESPONSE_DATA, response.getRawData());
        if (map.get(TEMPLATE_FILE) != null) {
            Object template = map.get(TEMPLATE_FILE);
            try {
                String returnValue = Velocity.evaluateTemplateFileAsString(template.toString(), context);
                if (returnValue != null)
                    response.setRawData(returnValue);
            } catch (SlxException e) {
                response.setStatus(Status.STATUS_FAILURE);
                response.setRawData(e.getFullMessage());
            }

        } else if (map.get(TEMPLATE) != null) {
            Object template = map.get(TEMPLATE);
            String returnValue = Velocity.evaluateAsString(template.toString(), context);
            if (returnValue != null)
                response.setRawData(returnValue);
        }
        response.setHandlerName("fchart");
        return response;

    }

    protected Object getData(DSRequest req, DataSource ds) throws SlxException {
        return ds.execute(req).getRawData();
    }

}