致力于测试团队建设和自动化测试开发，欢迎有兴趣者一起研究讨论

VBS写的常用数据库公用函数

上一篇 / 下一篇 2008-06-21 01:25:26 / 个人分类：Automation

以下是我写的常用的数据库公用函数，希望大家能一起探讨和总结，把一些常用的数据库公用函数分享出来。这个是初稿，以后使用中我会继续完善
特别是在做报表自动化测试时，数据库的公用函数使用尤其频繁，包括对存储过程的操作

' DATABASE公用函数
'
'###########################################################################################################
'###########################################################################################################

Dim objConnection    'CONNECTION对象实例
Dim objRecordSet    'RECORDSET对象实例
Dim objCommand '命令对象实例
Dim strConnectionString '连接字符串

' ********************************************************************
' 函数说明：连接数据库;
' 参数说明：(1)strDBType（数据库类型：如ORACEL；DB2；SQL；ACCESS）
'          (2)strDBAlias（数据库别名）
'          (3)strUID（用户名）
'          (4)strPWD（密码）
'          (5)strIP（数据库IP地址：仅SQL SERVER 使用）
'          (6)strLocalHostName（本地主机名：仅SQL SERVER 使用）
'          (7)strDataSource（数据源：仅ACCESS使用；如d:\yysc.mdb）
' 返回结果：无
' 调用方法: ConnectDatabase(strDBType, strDBAlias, strUID, strPWD, strIP, strLocalHostName, strDataSource)
' ********************************************************************
Sub ConnectDatabase(strDBType, strDBAlias, strUID, strPWD, strIP, strLocalHostName, strDataSource)
Set ōbjConnection = CreateObject("ADODB.CONNECTION") '1 - 建立CONNECTION对象的实例

Select Case UCase(Trim(strDBType))
      Case "ORACLE"
         strConnectionString = "Driver={Microsoft ODBC for Oracle};Server=" & strDBAlias & ";Uid="_
            & strUID & ";Pwd=" & strPWD & ";" '2 - 建立连接字符串
         objConnection.Open strConnectionString '3 - 用Open 方法建立与数据库连接
      Case "DB2"
         strConnectionString = "Driver={IBM DB2 ODBC DRIVER};DBALIAS=" & strDBAlias & ";Uid="_
            & strUID & ";Pwd=" & strPWD & ";"
         objConnection.Open strConnectionString
      Case "SQL"
         strConnectionString = "DRIVER=SQL Server; SERVER=" & strIP & "; UID=" & strUID & "; PWD="_
               & strPWD & "; APP=Microsoft Office 2003;WSID=" & strLocalHostName & "; DATABASE=" & strDBAlias & ";"
         objConnection.Open strConnectionString
      Case "ACCESS"
         strConnectionString = "provider=microsoft.jet.oledb.4.0;data source=" & strDataSource &_
            ";Jet OLEDB:Database Password=" & strPWD & ";"
         objConnection.Open strConnectionString
      Case Else
         MsgBox "输入的数据库类型格式有误" & vbCrLf & "支持的数据库类型格式：ORACLE；DB2；SQL；ACCESS；EXCEL"
End Select

If (objConnection.State = 0) Then
      MsgBox "连接数据库失败！"
End If

End Sub

' ********************************************************************
' 函数说明：查询数据库（查询单列）;
' 参数说明：  (1)strSql：SQL语句
'          (2)strFieldName：字段名
'          (3)str_Array_QueryResult：数组名（用来返回单列查询结果）
' 返回结果：  intArrayLength：查询数据库返回的记录行数
'          str_Array_QueryResult：数组名（用来返回单列查询结果）
' 调用方法: intArrayLength = QueryDatabase(strSql, strFieldName, str_Array_QueryResult)
' ********************************************************************
Function QueryDatabase(strSql, strFieldName, str_Array_QueryResult)
Dim intArrayLength       '数组长度
   Dim i

i = 0
str_Array_QueryResult = Array() '重新初始化数组为一个空数组

Set ōbjRecordSet = CreateObject("ADODB.RECORDSET") '4 - 建立RECORDSET对象实例
Set ōbjCommand = CreateObject("ADODB.COMMAND")             '5 - 建立COMMAND对象实例
objCommand.ActiveConnection = objConnection
objCommand.CommandText = strSql
objRecordSet.CursorLocation = 3
objRecordSet.Open objCommand                         '6 - 执行SQL语句,将结果保存在RECORDSET对象实例中

intArrayLength = objRecordSet.RecordCount    '将查询结果的行数作为数组的长度

If intArrayLength > 0 Then
ReDim str_Array_QueryResult(intArrayLength-1)

Do While NOT objRecordSet.EOF         '将数据库查询的列值赋值给数组
      str_Array_QueryResult(i) = objRecordSet(strFieldName)
'Debug.WriteLine str_Array_QueryResult(i)
objRecordSet.MoveNext
i = i + 1
Loop
' Else
'ReDim str_Array_QueryResult(0)
'str_Array_QueryResult(0) = ""
End If

QueryDatabase = intArrayLength
End Function

' ********************************************************************
' 函数说明：更新数据库;包括INSERT、DELETE 和 UPDATE操作
' 参数说明：(1)strSql：SQL语句
' 返回结果：无
' 调用方法: UpdateDatabase(strSql)
' ********************************************************************
Sub UpdateDatabase(strSql)
Dim objCommand
Dim objField

Set ōbjCommand = CreateObject("ADODB.COMMAND")
Set ōbjRecordSet = CreateObject("ADODB.RECORDSET")
objCommand.CommandText = strSql
objCommand.ActiveConnection = objConnection
Set ōbjRecordSet = objCommand.Execute

' Do Until objRecordSet.EOF

' For Each objField In objRecordSet.Fields
' Debug.Write objField.Name & ": " & objField.Value & " "
' Next

' objRecordSet.MoveNext
' Debug.WriteLine
' Loop

Set ōbjCommand = Nothing
Set ōbjRecordSet = Nothing

End Sub

' ********************************************************************
' 函数说明：返回符合查询结果的列的长度
' 参数说明：(1)strSql：SQL语句
' 返回结果：返回符合查询结果的列的长度
' 调用方法: MaxLength = GetLenOfField(strSql)
' ********************************************************************
Function GetLenOfField(strSql)
'如果SQL语句为空，则默认返回的列长度为0，结束函数;否则返回列的实际长度
If strSql = "" Then
      GetLenOfField  = 0
Exit Function
Else
      Set ōbjRecordSet = CreateObject("ADODB.RECORDSET") '4 - 建立RECORDSET对象实例
      Set ōbjCommand = CreateObject("ADODB.COMMAND")             '5 - 建立COMMAND对象实例
      objCommand.ActiveConnection = objConnection
      objCommand.CommandText = strSql
objRecordSet.CursorLocation = 3
objRecordSet.Open objCommand                               '6 - 执行SQL语句,将结果保存在RECORDSET对象实例中

      GetLenOfField = objRecordSet.RecordCount          '返回符合查询结果的列的长度

Set ōbjCommand = Nothing
Set ōbjRecordSet = Nothing
End If
End Function

' ********************************************************************
' 函数说明：关闭数据库连接;
' 参数说明：无
' 返回结果：无
' 调用方法: CloseDatabase()
' ********************************************************************
Sub CloseDatabase()
objRecordSet.Close
objConnection.Close

Set ōbjCommand = Nothing
Set ōbjRecordSet = Nothing
Set ōbjConnection = Nothing
End Sub

用户菜单

我的栏目

标题搜索

日历

我的存档

数据统计

访问量: 91044
日志数: 79
图片数: 1
建立时间: 2008-05-18
更新时间: 2009-06-04