Project

General

Profile

1 15 jones
/**
2 144 jones
 *        Name: DBWriter.java
3 15 jones
 *     Purpose: A Class that reads in an XML text document and
4
 *              write its contents to a database connection using SAX
5 35 jones
 *   Copyright: 2000 Regents of the University of California and the
6
 *              National Center for Ecological Analysis and Synthesis
7 15 jones
 *     Authors: Matt Jones
8
 *
9
 *     Version: '$Id$'
10
 */
11
12 75 jones
package edu.ucsb.nceas.metacat;
13 51 jones
14 15 jones
import org.xml.sax.*;
15
16
import java.io.*;
17
import java.net.URL;
18
import java.net.MalformedURLException;
19
import java.sql.*;
20
import java.util.Stack;
21
22
// Extensions to the SAX Interfaces for Namespace support.
23
import oracle.xml.parser.v2.XMLDocumentHandler;
24
import oracle.xml.parser.v2.DefaultXMLDocumentHandler;
25
import oracle.xml.parser.v2.NSName;
26
import oracle.xml.parser.v2.SAXAttrList;
27
28
import oracle.xml.parser.v2.SAXParser;
29
30 31 jones
/**
31
 * A Class that reads in an XML text document and
32
 * write its contents to a database connection using SAX
33
 */
34 144 jones
public class DBWriter {
35 15 jones
36 72 bojilova
  static  String 	defaultDB = "jdbc:oracle:thin:@penelope.nceas.ucsb.edu:1526:DEV8";
37 15 jones
  private Connection	conn = null;
38
39 31 jones
  /**
40 144 jones
   * the main routine used to test the DBWriter utility.
41 31 jones
   *
42 144 jones
   * Usage: java DBWriter <filename> <user> <password> [dbstring]
43 31 jones
   *
44
   * @param filename the filename to be loaded into the database
45
   * @param user the username to use for the database connection
46
   * @param password the password to use for the database connection
47
   * @param dbstring the connection info to use for the database connection
48
   */
49 15 jones
  static public void main(String[] args) {
50
51
     if (args.length < 3)
52
     {
53
        System.err.println("Wrong number of arguments!!!");
54 144 jones
        System.err.println("USAGE: java DBWriter " +
55 15 jones
                           "<filename> <user> <password> [dbstring]");
56
        return;
57
     } else {
58
        try {
59
60
          String filename = args[0];
61
          String user     = args[1];
62
          String password = args[2];
63
          String dbstring = null;
64
65
          if (args.length <= 3) {
66
            dbstring = defaultDB;
67
          } else {
68
            dbstring = args[3];
69
          }
70
71 50 jones
          // Open a connection to the database
72
          Connection dbconn = MetaCatUtil.openDBConnection(
73
                "oracle.jdbc.driver.OracleDriver",
74
                dbstring, user, password);
75
76 144 jones
          new DBWriter(filename, dbconn);
77 30 jones
          System.out.println("Document processing finished for: " + filename);
78
79 15 jones
        } catch (Exception e) {
80
          System.err.println("EXCEPTION HANDLING REQUIRED");
81
          System.err.println(e.getMessage());
82
          e.printStackTrace(System.err);
83
        }
84
     }
85
  }
86
87 31 jones
  /**
88
   * construct a new instance of the class to write an XML file to the database
89
   *
90
   * @param filename the filename to be loaded into the database
91 50 jones
   * @param conn the database connection to which to write the XML file
92 31 jones
   */
93 144 jones
  public DBWriter( Reader xml, Connection conn)
94 55 jones
                  throws IOException,
95
                         SQLException,
96
                         ClassNotFoundException
97
  {
98
    this.conn = conn;
99
100 72 bojilova
    try {
101 55 jones
        SAXParser parser = initializeParser(conn);
102 148 bojilova
        conn.setAutoCommit(false);
103 55 jones
        parser.parse(xml);
104 148 bojilova
        conn.commit();
105
        conn.setAutoCommit(true);
106 55 jones
      } catch (SAXParseException e) {
107
        System.err.println(e.getMessage());
108
      } catch (SAXException e) {
109
        System.err.println(e.getMessage());
110 72 bojilova
      } catch (Exception e) {
111
        System.err.println(e.toString());
112 55 jones
      }
113
  }
114
115 144 jones
  public DBWriter( String filename, Connection conn)
116 15 jones
                  throws IOException,
117
                         SQLException,
118
                         ClassNotFoundException
119 55 jones
  {
120 50 jones
     this.conn = conn;
121 55 jones
     FileReader xmlfile = new FileReader(new File(filename).toString());
122 15 jones
     try {
123 55 jones
       SAXParser parser = initializeParser(conn);
124 148 bojilova
       conn.setAutoCommit(false);
125 55 jones
       parser.parse(xmlfile);
126 148 bojilova
       conn.commit();
127
       conn.setAutoCommit(true);
128 55 jones
     } catch (SAXParseException e) {
129
       System.err.println(e.getMessage());
130
     } catch (SAXException e) {
131
       System.err.println(e.getMessage());
132 72 bojilova
     } catch (Exception e) {
133 148 bojilova
       System.err.println("from DBWriter:" + e.toString());
134 55 jones
     }
135
  }
136
137
  private SAXParser initializeParser(Connection conn) {
138
    SAXParser parser = null;
139
    //
140
    // Set up the SAX document handlers for parsing
141
    //
142
    try {
143
      // Use the XMLDocumentHandler interface for namespace support
144
      // instead of org.xml.sax.DocumentHandler
145
      XMLDocumentHandler xmlDocHandler = new DBSAXHandler(conn);
146 72 bojilova
      EntityResolver xmlEntityResolver = new DBEntityResolver(conn);
147
      DTDHandler xmlDTDHandler         = new DBDTDHandler(conn);
148 15 jones
149 55 jones
      // For all the other interface use the default provided by
150
      // Handler base
151
      HandlerBase defHandler = new HandlerBase();
152 15 jones
153 55 jones
      // Get an instance of the parser
154
      parser = new SAXParser();
155 15 jones
156 55 jones
      // Set Handlers in the parser
157
      // Set the DocumentHandler to XMLDocumentHandler
158
      parser.setDocumentHandler(xmlDocHandler);
159 72 bojilova
      parser.setEntityResolver(xmlEntityResolver);
160
      parser.setDTDHandler(xmlDTDHandler);
161 15 jones
162 55 jones
      // Set the other Handler to the defHandler
163
      parser.setErrorHandler(defHandler);
164 15 jones
165 55 jones
    } catch (Exception e) {
166
       System.err.println(e.toString());
167
    }
168 15 jones
169 55 jones
    return parser;
170
  }
171 15 jones
172 55 jones
  /** Utility method to convert a file handle into a URL */
173
  static public URL fileToURL(File file)
174
  {
175 16 jones
     String path = file.getAbsolutePath();
176
     String fSep = System.getProperty("file.separator");
177
     if (fSep != null && fSep.length() == 1)
178
       path = path.replace(fSep.charAt(0), '/');
179
     if (path.length() > 0 && path.charAt(0) != '/')
180
       path = '/' + path;
181
     try {
182
       return new URL("file", null, path);
183
     }
184
     catch (java.net.MalformedURLException e) {
185
       /* According to the spec this could only happen if the file
186
	  protocol were not recognized. */
187
       throw new Error("unexpected MalformedURLException");
188
     }
189 15 jones
  }
190
}