org.codehaus.cargo.sample.testdata.jdbc.TestServlet.java Source code

Introduction

Here is the source code for org.codehaus.cargo.sample.testdata.jdbc.TestServlet.java

Source

/*
 * ========================================================================
 * 
 * Copyright 2004-2005 Vincent Massol.
 * 
 * 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.codehaus.cargo.sample.testdata.jdbc;

import java.io.IOException;
import java.io.PrintWriter;

import javax.servlet.ServletConfig;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.springframework.transaction.PlatformTransactionManager;
import org.springframework.transaction.TransactionDefinition;
import org.springframework.transaction.TransactionStatus;
import org.springframework.transaction.support.DefaultTransactionDefinition;
import org.springframework.web.context.WebApplicationContext;
import org.springframework.web.context.support.WebApplicationContextUtils;

/**
 * Sample test Servlet used to verify that jdbc datasource is deployed.
 * 
 * @version $Id:$
 */
public class TestServlet extends HttpServlet {
    private WebApplicationContext ctx;

    private PlatformTransactionManager txManager;

    private DerbyDao dao;

    public void doGet(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {

        DefaultTransactionDefinition def = new DefaultTransactionDefinition();

        def.setName("cargotest");
        def.setPropagationBehavior(TransactionDefinition.PROPAGATION_REQUIRED);

        TransactionStatus status = txManager.getTransaction(def);
        try {
            dao.create("Adrian", "Cole");

        } catch (RuntimeException ex) {
            txManager.rollback(status);
            throw ex;
        }
        txManager.commit(status);
        if (dao.selectAll().size() != 1)
            throw new RuntimeException("Commit didn't work");
        PrintWriter out = response.getWriter();
        out.print("all good!");
        out.close();
    }

    public void init(ServletConfig config) throws ServletException {
        super.init(config);
        ctx = WebApplicationContextUtils.getRequiredWebApplicationContext(config.getServletContext());
        dao = (DerbyDao) ctx.getBean("personDao");
        txManager = (PlatformTransactionManager) ctx.getBean("transactionManager");
        dao.createTable();
    }

    public void destroy() {
        dao.dropTable();
        super.destroy();
    }
}