Web Applications with Java/JSP

Listing 6. SaveTaskServlet.java

    public Task save(Task task)
        throws SQLException
    {
        Connection conn = null;
        PreparedStatement ps = null;
        ResultSet rs = null;

        try
        {
            conn = getConnection();

            if(null == task.getId())
            {
                // A new task
                ps = conn.prepareStatement(
                         "INSERT INTO
                              task (owner, date, client_id,
                                    description, duration)
                              VALUES (?,?,?,?,?)",
                         PreparedStatement.RETURN_GENERATED_KEYS);

                ps.setString(1,  task.getOwner());
                ps.setTimestamp(2, new Timestamp(
                                           task.getDate().getTime()));
                ps.setInt(3, task.getClientId());
                ps.setString(4, task.getDescription());
                ps.setInt(5, task.getDuration());

                ps.executeUpdate();

                rs = ps.getGeneratedKeys();

                if(!rs.next())
                    throw new SQLException(
                        "Expected auto-generated key, got none");

                int taskId = rs.getInt(1);

                if(rs.wasNull())
                    throw new SQLException(
                        "Got bogus auto-generated key");

                task = new Task(taskId,
                                task.getOwner(),
                                task.getDate(),
                                task.getClientId(),
                                task.getDescription(),
                                task.getDuration());
            }
            else
            {
                ps = conn.prepareStatement(
                         "UPDATE task SET date=?, client_id=?,
                                          description=?, duration=?
                             WHERE id=? AND owner=?");

                ps.setTimestamp(1, new Timestamp(
                                           task.getDate().getTime()));
                ps.setInt(2, task.getClientId());
                ps.setString(3, task.getDescription());
                ps.setInt(4, task.getDuration());
                ps.setInt(5, task.getId());
                ps.setString(6, task.getOwner());

                ps.executeUpdate();
            }

            return task;
        }
        finally
        {
            close(conn, ps, rs);
        }
    }