如何进行Jython数据库插入(JDBC)
假定 Joe 是一名 QA,他需要为功能测试在 SAMPLE 数据库中创建新的雇员。在 Jython 中,Joe 进行少量工作就可以编写他自己的脚本,进行Jython数据库插入。
例如,Joe 可以在 SAMPLE 数据库中通过运行“jython batchinsert.jy joescript.txt”,创建三名新雇员。
清单 6. joescript.txt
createEmployee(firstName='JOHN',midinit='M',lastName='DOE',salary=32888.55,sex='M',workdept='A01', job='DESIGNER') createEmployee(firstName='MANAGER',midinit='M',lastName='DOE',salary=50000,sex='M',workdept='A01', job='MANAGER') createEmployee(firstName='SARA',midinit='M',lastName='DOE',salary=40000,sex='F',workdept='C01', job='DESIGNER')
如果 Joe 想要添加另一名 Employee,就只需用他喜欢的文本编辑器编辑“joescript.txt”。
Jython数据库插入的脚本背后的功能十分简单。Jython 有一个名为 execfile(filename) 的内置函数。如果我们将 createEmployee() 实现为 Jython 函数,用于向 EMPLOYEE 表插入一行记录,那么 execfile('joescript.txt') 将按照 Joe 需要的数目插入雇员。
换言之,我们利用 Jython 的解释器,并可以为 Joe 派生一个像这样的小的测试语言。该解决方案也是可扩展的。如果 Joe 需要在脚本中删除或更新雇员,就只需要添加另一函数。对于 Joe 来说,最酷的事情就是学习曲线很短。他可以立刻编写这样的脚本,而没有意识到正在进行 Jython 函数调用。
清单 7. (batchinsert.jy)createEmployee 函数
from java.lang import * from java.sql import * import sys def createEmployee(**args): global stmt sqlTemplate= "INSERT into EMPLOYEE (EMPNO,FIRSTNAME,MIDINIT,LASTNAME,SALARY,SEX,EDLEVEL, HIREDATE, WORKDEPT,JOB) values ((select RTRIM(CHAR(MAX(INT(EMPNO))+1)) from EMPLOYEE), '%(firstName)s', '%(midinit)s','%(lastName)s', %(salary).2f,'%(sex)s' ,18, CURRENT DATE,'%(workdept)s','%(job)s')" # dictionary based SQL string formatting sql=sqlTemplate % args stmt.addBatch(sql) # load DB2 JDBC type 2 driver Class.forName("COM.ibm.db2.jdbc.app.DB2Driver").newInstance() con = DriverManager.getConnection( 'jdbc:db2:sample', 'vyang','jythonrocks'); con.setAutoCommit(0) stmt = con.createStatement() execfile(sys.argv[1]) stmt.executeBatch() stmt.close() # commit con.commit() con.close()
Jython数据库插入的这部分有许多新内容。
con.setAutoCommit(0) 用于关闭自动提交(auto commit),就像 JDBC 调用中的 con.setAutoCommit(false) 一样。
用于成批插入的 stmt.addBatch(sql) 、 stmt.executeBatch() 能有效地加快批量插入的速度。该技术还在 Neo 的加载仿真器中用于进行大容量的插入。
global stmt 将使 stmt 成为一个共享的全局变量,很适用于该环境中。