1 files changed, 245 insertions, 0 deletions
diff --git a/java/demo/Database/library/LibraryI.java b/java/demo/Database/library/LibraryI.java
new file mode 100644
index 00000000000..7f01f9f87c6
--- /dev/null
+++ b/java/demo/Database/library/LibraryI.java
@@ -0,0 +1,245 @@
+// **********************************************************************
+//
+// Copyright (c) 2003-2008 ZeroC, Inc. All rights reserved.
+//
+// This copy of Ice is licensed to you under the terms described in the
+// ICE_LICENSE file included in this distribution.
+//
+// **********************************************************************
+
+import Demo.*;
+
+//
+// This servant is a default servant. The book identity is retrieved
+// from the Ice.Current object. The session object associated with the
+// library is also retrieved using the Library's identity.
+//
+class LibraryI extends _LibraryDisp
+{
+    public void
+    queryByIsbn(String isbn, BookDescriptionHolder first, BookQueryResultPrxHolder result, Ice.Current current)
+        throws NoResultsException
+    {
+        SessionI session = SessionI.getSession(current.id);
+        if(session == null)
+        {
+            // No associated session.
+            throw new Ice.ObjectNotExistException();
+        }
+        session.reapQueries();
+
+        SQLRequestContext context = SQLRequestContext.getCurrentContext();
+        assert context != null;
+
+        try
+        {
+            java.sql.PreparedStatement stmt = context.prepareStatement("SELECT * FROM books WHERE isbn LIKE ?");
+            stmt.setString(1, "%" + isbn + "%");
+            java.sql.ResultSet rs = stmt.executeQuery();
+            if(!rs.next())
+            {
+                throw new NoResultsException();
+            }
+
+            first.value = BookI.extractDescription(context, rs, current.adapter);
+            if(rs.next())
+            {
+                // The SQLRequestContext is now owned by the query
+                // implementation.
+                context.obtain();
+                BookQueryResultI impl = new BookQueryResultI(_logger, context, rs);
+                result.value = BookQueryResultPrxHelper.uncheckedCast(current.adapter.addWithUUID(impl));
+                session.add(result.value, impl);
+            }
+        }
+        catch(java.sql.SQLException e)
+        {
+            // Log the error, and raise an UnknownException.
+            error(e);
+            Ice.UnknownException ex = new Ice.UnknownException();
+            ex.initCause(e);
+            throw ex;
+        }
+    }
+
+    public void
+    queryByAuthor(String author, BookDescriptionHolder first, BookQueryResultPrxHolder result, Ice.Current current)
+        throws NoResultsException
+    {
+        SessionI session = SessionI.getSession(current.id);
+        if(session == null)
+        {
+            // No associated session.
+            throw new Ice.ObjectNotExistException();
+        }
+        session.reapQueries();
+
+        SQLRequestContext context = SQLRequestContext.getCurrentContext();
+        assert context != null;
+
+        try
+        {
+            // Find each of the authors.
+            java.sql.PreparedStatement stmt = context.prepareStatement("SELECT * FROM authors WHERE name LIKE ?");
+            stmt.setString(1, "%" + author + "%");
+            java.sql.ResultSet rs = stmt.executeQuery();
+            if(!rs.next())
+            {
+                throw new NoResultsException();
+            }
+
+            // Build a query that finds all books by these authors.
+            StringBuffer sb = new StringBuffer("SELECT * FROM books INNER JOIN authors_books ON " +
+                                               "books.id=authors_books.book_id AND (");
+            boolean front = true;
+            do
+            {
+                if(!front)
+                {
+                    sb.append(" OR ");
+                }
+                front = false;
+                sb.append("authors_books.author_id=");
+                sb.append(rs.getInt("id"));
+            }
+            while(rs.next());
+            sb.append(")");
+
+            // Execute the query.
+            stmt = context.prepareStatement(sb.toString());
+            rs = stmt.executeQuery();
+            if(!rs.next())
+            {
+                throw new NoResultsException();
+            }
+
+            first.value = BookI.extractDescription(context, rs, current.adapter);
+            if(rs.next())
+            {
+                // The SQLRequestContext is now owned by the query implementation.
+                context.obtain();
+                BookQueryResultI impl = new BookQueryResultI(_logger, context, rs);
+                result.value = BookQueryResultPrxHelper.uncheckedCast(current.adapter.addWithUUID(impl));
+                session.add(result.value, impl);
+            }
+        }
+        catch(java.sql.SQLException e)
+        {
+            // Log the error, and raise an UnknownException.
+            error(e);
+            Ice.UnknownException ex = new Ice.UnknownException();
+            ex.initCause(e);
+            throw ex;
+        }
+    }
+
+    public BookPrx
+    createBook(String isbn, String title, java.util.List<String> authors, Ice.Current current)
+        throws BookExistsException
+    {
+        SQLRequestContext context = SQLRequestContext.getCurrentContext();
+        assert context != null;
+        try
+        {
+            java.sql.PreparedStatement stmt = context.prepareStatement("SELECT * FROM books WHERE isbn = ?");
+            stmt.setString(1, isbn);
+            java.sql.ResultSet rs = stmt.executeQuery();
+            if(rs.next())
+            {
+                throw new BookExistsException();
+            }
+
+            //
+            // First convert the authors string to an id set.
+            //
+            java.util.List<Integer> authIds = new java.util.LinkedList<Integer>();
+            java.util.Iterator<String> p = authors.iterator();
+            while(p.hasNext())
+            {
+                String author = p.next();
+
+                Integer id;
+                stmt = context.prepareStatement("SELECT * FROM authors WHERE name = ?");
+                stmt.setString(1, author);
+                rs = stmt.executeQuery();
+                if(rs.next())
+                {
+                    // If there is a result, then the database
+                    // already contains this author.
+                    id = rs.getInt(1);
+                    assert !rs.next();
+                }
+                else
+                {
+                    // Otherwise, create a new author record.
+                    stmt = context.prepareStatement("INSERT INTO authors (name) VALUES(?)",
+                                                    java.sql.Statement.RETURN_GENERATED_KEYS);
+                    stmt.setString(1, author);
+                    int count = stmt.executeUpdate();
+                    assert count == 1;
+                    rs = stmt.getGeneratedKeys();
+                    boolean next = rs.next();
+                    assert next;
+                    id = rs.getInt(1);
+                }
+
+                // Add the new id to the list of ids.
+                authIds.add(id);
+            }
+
+            // Create the new book.
+            stmt = context.prepareStatement("INSERT INTO books (isbn, title) VALUES(?, ?)",
+                                            java.sql.Statement.RETURN_GENERATED_KEYS);
+            stmt.setString(1, isbn);
+            stmt.setString(2, title);
+            int count = stmt.executeUpdate();
+            assert count == 1;
+
+            rs = stmt.getGeneratedKeys();
+            boolean next = rs.next();
+            assert next;
+            Integer bookId = rs.getInt(1);
+
+            // Create new authors_books records.
+            java.util.Iterator<Integer> q = authIds.iterator();
+            while(q.hasNext())
+            {
+                stmt = context.prepareStatement("INSERT INTO authors_books (book_id, author_id) VALUES(?, ?)");
+                stmt.setInt(1, bookId);
+                stmt.setInt(2, q.next());
+                count = stmt.executeUpdate();
+                assert count == 1;
+            }
+
+            // Commit the transaction.
+            context.commit();
+
+            return BookPrxHelper.uncheckedCast(current.adapter.createProxy(BookI.createIdentity(bookId)));
+        }
+        catch(java.sql.SQLException e)
+        {
+            // Log the error, and raise an UnknownException.
+            error(e);
+            Ice.UnknownException ex = new Ice.UnknownException();
+            ex.initCause(e);
+            throw ex;
+        }
+    }
+
+    LibraryI(Ice.Logger logger)
+    {
+        _logger = logger;
+    }
+
+    private void
+    error(Exception ex)
+    {
+        java.io.StringWriter sw = new java.io.StringWriter();
+        java.io.PrintWriter pw = new java.io.PrintWriter(sw);
+        ex.printStackTrace(pw);
+        pw.flush();
+        _logger.error("LibraryI: error:\n" + sw.toString());
+    }
+
+    private Ice.Logger _logger;
+}