【QTP专题】连接数据库

1. 获取数据库连接串

   在本地新建一个.txt文件，修改扩展名名*.udl；双击*.udl文件，打开数据库链接属性，定位到"提供程序"选显卡，选中如sqlserver的连接  Microsoft  for OLE DB Provider for sql server,点击下一步；输入数据源，数据库用户名以及密码，选择数据库，点击测试连接，测试成功后，点击"确定"。如图所示

   然后用UE或记事本打开*.udl文件，sqlserver的连接串已经生成了，如下：

Provider=SQLOLEDB.1;Password=trsadmin;Persist Security Info=True;User ID=sa;Initial Catalog=test;Data Source=192.***.***.***

  

　　2.连接数据库

Dim objConnection 'Connection对象实例 

Dim strConnString '连接字符串 

strConnString="Provider=SQLOLEDB.1;Password=trsadmin;Persist Security Info=True;User ID=sa;Initial Catalog=test;Data Source=192.***.***.***"



Set objConnection = CreateObject("ADODB.CONNECTION")  '建立Connection对象的实例 

objConnection.Open strConnString  '用Open方法建立与数据库连接
objConnection.state=0  '表示数据库连接失败

  

　　3.关闭数据库连接

    objRecordSet.Close 

    objConnection.Close 

    Set objRecordSet = Nothing 

    Set objConnection = Nothing

 

　　4.执行查询 

Dim objRecordSet 'RecordSet对象实例

Dim sqlStr 'sql查询语句



Set objRecordSet = CreateObject("ADODB.Recordset") '建立Recordset对象的实例

objRecordSet.Open sqlStr,objConnection '执行查询



If (objRecordSet.BOF or objRecordSet.EOF) Then  '查询不存在任何记录

  executeSQL = NULL '返回NULL

Else '存在查询记录

  '使用 GetRows 方法将记录从 Recordset 复制到二维数组中。第一个下标标识字段，第二个则标识记录号。

  executeSQL = objRecordSet.GetRows() '返回查询结果的二维数组

End If

  

以下代码将数据库各个常用功能封装成函数，已测试通过，仅供参考：

Option Explicit

'公共对象

Dim objConnection    'Connection对象实例 

Dim strConnString    '连接字符串 

Dim objRecordSet    'RecordSet对象实例  





'==================================================================

' 函数功能：连接到数据库

' 参数说明：

'        1.strDBType（数据库类型：支持SQL SERVER、ORACEL、MYSQL、DB2、SYBASE） 

'        2.strDBName（数据库名） 

'        3.strUID（用户名） 

'        4.strPWD（密码） 

'        5.strIP（服务器名或IP地址） 

' 返回结果：无 

' 调用方法：ConnectDB(strDBType, strDBName, strUID, strPWD) 

'==================================================================

Sub connectDB(strDBType, strDBName, strUID, strPWD,strIP) 

    Select Case UCase(Trim(strDBType)) '不同的数据库类型连接数据库的字符串不同

         Case "SQL SERVER" 

            strConnString="Provider=SQLOLEDB; Data Source="& strIP &"; Initial Catalog="& strDBName &"; User ID="& strUID &"; Password="&strPWD        

        Case "ORACLE" 

            strConnString="Provider=OraOLEDB.Oracle; Data Source="& strDBName &"; User ID="& strUID &"; Password="& strPWD  

        Case "MYSQL" 

            strConnString="Driver={MySQL ODBC 5.1 Driver}; Server="& strIP &"; Database="& strDBName &"; User="&strUID&"; Password="& strPWD 

　　     Case Else 

            Reporter.ReportEvent micFail,"数据库类型错误" ,"目前支持的数据库类型有：SQL SERVER、ORACEL、MYSQL、DB2、SYBASE" 

    End Select 

    Set objConnection = CreateObject("ADODB.CONNECTION")  '建立Connection对象的实例 

    objConnection.Open strConnString  '用Open方法建立与数据库连接 

End Sub 





'==================================================================

' 函数功能：关闭数据库连接

' 参数说明：无 

' 返回结果：无 

' 调用方法：closeDB() 

'==================================================================

Sub closeDB() 

    objRecordSet.Close 

    objConnection.Close 

    Set objRecordSet = Nothing 

    Set objConnection = Nothing 

End Sub 



'==================================================================

' 函数功能：执行SQL语句

' 参数说明：sqlStr（SQL语句 ）

' 返回结果：返回一个存放查询结果的二维数组

' 调用方法：results = executeSQL(sqlStr)

'==================================================================

Function executeSQL (sqlStr)

    Call connectDB(gDBType,gDBName,gUID,gPWD,gIP)' 连接到数据库，函数的实参在配置文件中定义

    Set objRecordSet = CreateObject("ADODB.Recordset") '建立Recordset对象的实例 

    objRecordSet.Open sqlStr,objConnection

    If (objRecordSet.BOF or objRecordSet.EOF) Then  '查询不存在任何记录

        Reporter.ReportEvent micDone,"查询不存在任何记录","数据库中查询不存在任何记录！"

        executeSQL = NULL '返回NULL

    Else '存在查询记录

        '使用 GetRows 方法将记录从 Recordset 复制到二维数组中。第一个下标标识字段，第二个则标识记录号。

        executeSQL = objRecordSet.GetRows() '返回查询结果的二维数组

    End If

    Call closeDB() '关闭数据库连接

End Function

=======================================================================================================

 SQL执行结果处理：

======================================================================================================

1.检查查询结果记录数

Dim num,sql
sql = "select COUNT(ID) from TableName where .....

results = executeSQL(sql)
num=cint(results(0,0)) 'cint函数将查询结果转化为Integer类型,num=0表示查询结果为空，num=1表示查询结果有且唯一，num>1表示查询结果有多条

以下为使用案例：

'检查用户是否存在，如果存在则返回true，不存在则返回false

'参数为用户名

Function chkUserExistDB(userName)

 Dim sql,results

 sql = "select COUNT(USERNAME) from WCMUSER where USERNAME ='"& userName &"'"

 results = executeSQL(sql)

 If cint(results(0,0))=1 Then '存在用户userName

  chkUserExistDB = true

  Reporter.ReportEvent micDone, "用户存在","系统中存在用户["& userName &"] " 

 Else '不存在用户userName

  chkUserExistDB = false

  Reporter.ReportEvent micDone, "用户不存在","系统中不存在用户["& userName &"] " 

 End If

End Function