org.onecmdb.web.remote.RemoteController.java Source code

Java tutorial

Introduction

Here is the source code for org.onecmdb.web.remote.RemoteController.java

Source

/*
 * Lokomo OneCMDB - An Open Source Software for Configuration
 * Management of Datacenter Resources
 *
 * Copyright (C) 2006 Lokomo Systems AB
 *
 * 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 2 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, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
 * 02110-1301 USA.
 * 
 * Lokomo Systems AB can be contacted via e-mail: info@lokomo.com or via
 * paper mail: Lokomo Systems AB, Svrdvgen 27, SE-182 33
 * Danderyd, Sweden.
 *
 */
package org.onecmdb.web.remote;

import java.io.OutputStream;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.onecmdb.core.IOneCmdbContext;
import org.onecmdb.core.utils.xpath.commands.AuthCommand;
import org.onecmdb.core.utils.xpath.commands.CreateCommand;
import org.onecmdb.core.utils.xpath.commands.DeleteCommand;
import org.onecmdb.core.utils.xpath.commands.ExportCommand;
import org.onecmdb.core.utils.xpath.commands.QueryCommand;
import org.onecmdb.core.utils.xpath.commands.UpdateCommand;
import org.springframework.web.bind.ServletRequestDataBinder;
import org.springframework.web.servlet.ModelAndView;
import org.springframework.web.servlet.mvc.multiaction.MultiActionController;

public class RemoteController extends MultiActionController {

    private IOneCmdbContext onecmdb;
    private Log log = null;

    /**
      * NOTE: Used to satisfy spring only.
      */

    public void setOneCmdb(IOneCmdbContext onecmdb) {
        this.onecmdb = onecmdb;
    }

    public IOneCmdbContext getOneCmdb() {
        return this.onecmdb;
    }

    public void setLog(Log log) {
        this.log = log;
    }

    public Log getLog() {
        if (this.log == null) {
            this.log = LogFactory.getLog(this.getClass());
        }
        return (this.log);
    }

    /**
     * Initilize this controller.
     *
     */
    public void init() {
        getLog().info("RemoteController init()");
    }

    /**
     * Command(s)
     */

    /*
     * Auth Command.
     */
    public ModelAndView authHandler(HttpServletRequest request, HttpServletResponse resp) throws Exception {
        getLog().info("AuthHandler()");
        AuthCommand command = new AuthCommand(this.onecmdb);
        ServletRequestDataBinder binder = new ServletRequestDataBinder(command);
        binder.bind(request);

        try {
            String token = command.getToken();
            resp.setContentLength(token.length());
            resp.setContentType("text/plain");
            resp.getOutputStream().write(token.getBytes());

        } catch (Exception e) {
            resp.setStatus(HttpServletResponse.SC_NOT_ACCEPTABLE, "Authentication Failed!");
        }
        return (null);
    }

    /**
     * Query command
     * 
     * @param request
     * @param resp
     * @return
     * @throws Exception
     */
    public ModelAndView queryHandler(HttpServletRequest request, HttpServletResponse resp) throws Exception {
        long start = System.currentTimeMillis();
        try {
            getLog().info("QueryHandler()");
            QueryCommand command = new QueryCommand(this.onecmdb);
            ServletRequestDataBinder binder = new ServletRequestDataBinder(command);
            binder.bind(request);

            try {
                resp.setContentType(command.getContentType());

                //resp.setContentLength(-1);
                OutputStream out = resp.getOutputStream();
                command.transfer(out);
                out.flush();

            } catch (Throwable e) {
                e.printStackTrace();
                resp.setStatus(HttpServletResponse.SC_BAD_REQUEST, "Error:" + e.getMessage());
            }
            return (null);
        } finally {
            long stop = System.currentTimeMillis();
            System.out.println("Query:" + (stop - start) + "ms");
        }
    }

    /**
     * Export command
     * 
     * @param request
     * @param resp
     * @return
     * @throws Exception
     */
    public ModelAndView exportHandler(HttpServletRequest request, HttpServletResponse resp) throws Exception {
        long start = System.currentTimeMillis();
        try {
            getLog().info("ExportHandler()");
            ExportCommand command = new ExportCommand(this.onecmdb);
            ServletRequestDataBinder binder = new ServletRequestDataBinder(command);
            binder.bind(request);

            try {
                resp.setContentType(command.getContentType());

                //resp.setContentLength(-1);
                OutputStream out = resp.getOutputStream();
                command.transfer(out);
                out.flush();

            } catch (Throwable e) {
                e.printStackTrace();
                resp.setStatus(HttpServletResponse.SC_BAD_REQUEST, "Error:" + e.getMessage());
            }
            return (null);
        } finally {
            long stop = System.currentTimeMillis();
            System.out.println("Query:" + (stop - start) + "ms");
        }
    }

    /**
     * Update command
     * 
     * @param request
     * @param resp
     * @return
     * @throws Exception
     */
    public ModelAndView updateHandler(HttpServletRequest request, HttpServletResponse resp) throws Exception {
        long start = System.currentTimeMillis();
        try {
            getLog().info("UpdateHandler()");
            UpdateCommand command = new UpdateCommand(this.onecmdb);
            ServletRequestDataBinder binder = new ServletRequestDataBinder(command);
            binder.bind(request);

            try {
                resp.setContentType(command.getContentType());

                //resp.setContentLength(-1);
                OutputStream out = resp.getOutputStream();
                command.transfer(out);
                out.flush();

            } catch (Throwable e) {
                e.printStackTrace();
                resp.setStatus(HttpServletResponse.SC_BAD_REQUEST, "Error:" + e.getMessage());
            }
            return (null);
        } finally {
            long stop = System.currentTimeMillis();
            System.out.println("Update:" + (stop - start) + "ms");
        }
    }

    /**
     * Create command
     * 
     * @param request
     * @param resp
     * @return
     * @throws Exception
     */
    public ModelAndView createHandler(HttpServletRequest request, HttpServletResponse resp) throws Exception {
        long start = System.currentTimeMillis();
        try {
            getLog().info("CreateHandler()");
            CreateCommand command = new CreateCommand(this.onecmdb);
            ServletRequestDataBinder binder = new ServletRequestDataBinder(command);
            binder.bind(request);

            try {
                resp.setContentType(command.getContentType());

                //resp.setContentLength(-1);
                OutputStream out = resp.getOutputStream();
                command.transfer(out);
                out.flush();

            } catch (Throwable e) {
                e.printStackTrace();
                resp.setStatus(HttpServletResponse.SC_BAD_REQUEST, "Error:" + e.getMessage());
            }
            return (null);
        } finally {
            long stop = System.currentTimeMillis();
            System.out.println("Update:" + (stop - start) + "ms");
        }
    }

    /**
     * Delete command
     * 
     * @param request
     * @param resp
     * @return
     * @throws Exception
     */
    public ModelAndView deleteHandler(HttpServletRequest request, HttpServletResponse resp) throws Exception {
        long start = System.currentTimeMillis();
        try {
            getLog().info("DeleteHandler()");
            DeleteCommand command = new DeleteCommand(this.onecmdb);
            ServletRequestDataBinder binder = new ServletRequestDataBinder(command);
            binder.bind(request);

            try {
                resp.setContentType(command.getContentType());

                //resp.setContentLength(-1);
                OutputStream out = resp.getOutputStream();
                command.transfer(out);
                out.flush();

            } catch (Throwable e) {
                e.printStackTrace();
                resp.setStatus(HttpServletResponse.SC_BAD_REQUEST, "Error:" + e.getMessage());
            }
            return (null);
        } finally {
            long stop = System.currentTimeMillis();
            System.out.println("Update:" + (stop - start) + "ms");
        }
    }

}