net.sf.saxon.option.sql.SQLInsert Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of saxon-he Show documentation
Show all versions of saxon-he Show documentation
An OSGi bundle for Saxon-HE
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
// Copyright (c) 2013 Saxonica Limited.
// This Source Code Form is subject to the terms of the Mozilla Public License, v. 2.0.
// If a copy of the MPL was not distributed with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
// This Source Code Form is "Incompatible With Secondary Licenses", as defined by the Mozilla Public License, v. 2.0.
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
package net.sf.saxon.option.sql;
import net.sf.saxon.expr.Expression;
import net.sf.saxon.expr.SimpleExpression;
import net.sf.saxon.expr.XPathContext;
import net.sf.saxon.expr.instruct.Executable;
import net.sf.saxon.om.AxisInfo;
import net.sf.saxon.om.Item;
import net.sf.saxon.om.NodeInfo;
import net.sf.saxon.om.Sequence;
import net.sf.saxon.style.Declaration;
import net.sf.saxon.style.ExtensionInstruction;
import net.sf.saxon.trans.SaxonErrorCode;
import net.sf.saxon.trans.XPathException;
import net.sf.saxon.tree.iter.AxisIterator;
import net.sf.saxon.type.Type;
import net.sf.saxon.value.AtomicValue;
import net.sf.saxon.value.EmptySequence;
import net.sf.saxon.value.ObjectValue;
import net.sf.saxon.value.Whitespace;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
/**
* An sql:insert element in the stylesheet.
*/
public class SQLInsert extends ExtensionInstruction {
Expression connection;
String table;
public void prepareAttributes() throws XPathException {
table = getAttributeList().getValue("", "table");
// TODO: allow table to be an AVT
if (table==null) {
reportAbsence("table");
}
table = SQLConnect.quoteSqlName(table);
String connectAtt = getAttributeList().getValue("", "connection");
if (connectAtt==null) {
reportAbsence("connection");
} else {
connection = makeExpression(connectAtt);
}
}
public void validate(Declaration decl) throws XPathException {
super.validate(decl);
connection = typeCheck("connection", connection);
AxisIterator kids = iterateAxis(AxisInfo.CHILD);
while(true) {
NodeInfo curr = (NodeInfo)kids.next();
if (curr == null) {
break;
}
if (curr instanceof SQLColumn) {
// OK
} else if (curr.getNodeKind() == Type.TEXT && Whitespace.isWhite(curr.getStringValueCS())) {
// OK
} else {
compileError("Only sql:column is allowed as a child of sql:insert", "XTSE0010");
}
}
}
public Expression compile(Executable exec, Declaration decl) throws XPathException {
// Collect names of columns to be added
StringBuffer statement = new StringBuffer(120);
statement.append("INSERT INTO " + table + " (");
AxisIterator kids = iterateAxis(AxisInfo.CHILD);
NodeInfo child;
int cols = 0;
while (true) {
child = (NodeInfo)kids.next();
if (child == null) {
break;
}
if (child instanceof SQLColumn) {
if (cols++ > 0) statement.append(',');
String colname = ((SQLColumn)child).getColumnName();
statement.append(colname);
}
}
statement.append(") VALUES (");
// Add "?" marks for the variable parameters
for(int i=0; i