用VS.Net编写扩展存储过程（二）

还早，再写一节。大后天准备陪一位女孩以其准男朋友的身份去旅游，顺便把准字给去掉，都准两年了……

三、用扩展存储过程解决第一个问题

虽然这样有些用战斧定点灭老鼠的嫌疑，但不失为一个入门的方法，更重要的是，偶可以借此整理下思路，番茄虽然是好东西，但吃多了也会发胖，太胖了女朋友就不要俺………………你又砸俺，哈哈，上当了吧～～

在Vs.Net中建立一个扩展存储过程项目，命名为xsTrainQuery，点确定之后，在弹出窗口中设置扩展存储过程名为xs_TrainQuery，完成。

现在，Vs.Net自动生成了几个文件，如果你是我这种新手，就先读读Readme.txt，里面对每个文件的作用和加入、运行、删除扩展存储过程xs_TrainQuery的方法有详细解释。

让我们把注意力集中到proc.cpp，这个文件的作用是实现xsTrainQuery.cpp中定义的接口，其内容如下：

#include

// 一些全局常量声明

#ifdef __cplusplus
extern "C" {
#endif
RETCODE __declspec(dllexport) xs_TrainQuery(SRV_PROC *srvproc); //声明
#ifdef __cplusplus
}
RETCODE __declspec(dllexport) xs_TrainQuery(SRV_PROC *srvproc) //实现
{
????????? //......
}

其中，RETCODE xs_TrainQuery(SRV_PROC*)，正是在数据库中运行扩展存储过程xs_TrainQuery后将运行的方法，所以只需要在这一部分中加入适当的代码，就做出了一个扩展存储过程。

SRV_PROC是一个结构，在中定义，代表一个数据库客户端连接的句柄，srv.h中几乎所有的方法都要用到这个句柄，但这个句柄背后究竟有什么数据，我们完全不用关心。（忽然想起写硕士毕业论文时觉得CBitmap类不好用所以自己傻乎乎地超级郁闷地做一个Image类的时候……）

此扩展存储过程调用的格式应该是xs_TrainQuery [起点站名，终点站名，结果 Output]，那么解决第一个问题将分为以下几个步骤：

1、传入两个车站名的值；
2、读入同时含此二车站的所有列车路线；
3、寻找最短的路线；
4、生成结果并传出之。

第一步，让我们解决传值问题。

扩展存储过程中的值传入严格来说包括3步：先获得参数的类型、长度等信息，再设置此参数对应的局部变量的类型和长度，最后读入此参数，即先调用srv_paraminfo，再alloc一个合适的内存空间，最后调用srv_paraminfo将入参值存入此内存空间。

srv_paraminfo的格式为：
int srv_paraminfo(
SRV_PROC *srvproc,
int n,
BYTE *pbType,
ULONG *pcbMaxLen,
ULONG *pcbActualLen,
BYTE *pbData,
BOOL *pfNull )

其中：
n代表是第几个参数，比如xs_TrainQuery '北京', '成都', ''中，'成都'的n值为2，
pbType, pcbMaxLen, pcbActualLen分别代表指向此参数的类型、最大长度和实际传入长度的指针，
pbData代表指向此参数值的指针，
pfNull代表此参数是否为NULL，若是，则运行srv_paraminfo后，*pfNull将被设置为TRUE.
当此参数的信息存在时，srv_paraminfo返回SUCCEED，否则返回FAIL.

我们现在要读入起点和终点站名，就需要如下代码：

{
????????? BOOL bfNull;?????????????????????????????? // 记录入参是否为空(NULL)
????????? PBYTE pbType;?????????????????????????? // 入参的类型
????????? ULONG ulMaxLen = 20;???????????? // 入参的最大长度，令为20字节
????????? ULONG ulActualLen;????????????????? // 入参的实际长度
????????? PBYTE pbStart;????????????????????????? // 起点站名字
????????? PBYTE pbEnd;?????????????????????????? // 终点站名字
????????? // 获得第1个参数，即起点站名的类型、长度等信息
????????? if ( srv_paraminfo(srvproc, 1, pbType, &ulMaxLen, &ulActualLen, NULL, &bfNULL) !=?
SUCCEED )
????????? {
??????????????????? // 一些异常处理代码
????????? }
????????? // 为起点站入参分配空间
????????? pbStart = (PBYTE)::malloc(ulActualLen);
????????? if ( pbStart == NULL)
????????? {
??????????????????? // 一些异常处理代码
????????? }
????????? // 获得第1个参数的值
????????? if ( srv_paraminfo(srvproc, 1, pbType, &ulMaxLen, &ulActualLen, pbStart, &bfNULL)?
!= SUCCEED )
????????? {
??????????????????? // 一些异常处理代码
????????? }
????????? // 重复上面的三步，但n变为2，pbStart变为pbEnd，读入终点站名
}

第二步，从数据库中取得含二车站名的所有路线

在扩展存储过程中连接数据库有两种方法，第一种请参见MS Sql Server ODS示例中的xp_dblib，这里只介绍第二种方法，采用SQL开头的一系列ODBC API.

为了使用这些API，首先要在proc.cpp或者stdafx.h中加入#include .

一次数据库连接的过程包括以下几个步骤：
1、初始化ODBC连接，并分配环境句柄；
2、设置环境；
3、根据环境句柄分配数据库连接句柄；
4、调用SQLDriverConnect连接MS SQL Serve数据源；
5、分配和使用语句；
6、调用SQLDisconnect断开连接；
7、依次释放分配的句柄。
直接用例子来说明，现在我们要从数据库中读取含此二车站名的所有路线，可以用下面的代码：
{
??? SQLHENV hEnv = SQL_NULL_ENV;??????????? // 环境句柄
??? SQLHDBC hDbc = SQL_NULL_HDBC;??????? // 连接句柄
??? SQLHSTMT hStmt = SQL_NULL_HSTMT;?? // 语句句柄?
??? SQLCHAR connStr[255] = "Driver={SQL Server};SERVER=localhost;UID=你的用户名;PWD=你的密码;DATABASE=Railway;";
??? // 分配环境句柄
??? SQLAllocHandle(SQL_HANDLE_ENV, NULL, &hEnv);
??? // 设置连接环境，ODBC版本设为3.x
??? SQLSetEnvAttr(hEnv, SQL_ATTR_ODBC_VERSION, (SQLPOINTER)SQL_OV_ODBC3, SQL_IS_INTEGER);
??? // 根据环境分配连接句柄
??? SQLAllocHandle(SQL_HANDLE_DBC, hEnv, &hDbc);
??? // 建立数据库连接
??? SQLDriverConnect(hDbc, NULL, connStr, SQL_NTS, NULL, 255, NULL, SQL_DRIVER_NOPROMPT);
??? // 根据连接分配语句句柄
??? SQLAllocHandle(SQL_HANDLE_STMT, hDbc, &hStmt);

??? // 定义查询语句query，别忘记#include
??? TCHAR query[255] = "select Routes.TrainID, AllStations from Routes, Trains where Routes.TrainID=Trains.TrainID and AllStations like '%";
??? _tcscat(query, (TCHAR*)pbStart);
??? _tcscat(query, "%");
??? _tcscat(query, (TCHAR*)pbEnd);
??? _tcscat(query, "%'");
??? // 现在，query="select Routes…… like '%起点站%终点站%'"，我们运行此查询
??? SQLExecDirect(hStmt, (SQLCHAR*)query, SQL_NTS);
??? // 获得查询所得的行数
??? UINT uiRowsCount;????????????????? //结果集中的行数
??? SQLSetStmtAttr(hStmt, SQL_ATTR_ROWS_FETCHED_PTR), (void*)&uiRowsCount, sizeof(SQLINTEGER));
??? // 初始化结果集数组，包括两个：TrainID和AllStations
??? int *piTrainID = (int*)::malloc(uiRowsCount * sizeof(int));
??? PCHAR *ppcStations = (PCHAR*)::malloc(uiRowsCount * sizeof(PCHAR));

??? // 因为结果集中第一列，即TrainID为整型，长度不变，故可以直接绑定
??? SQLBindCol(hStmt, 1, SQL_INTEGER, (SQLPOINTER)piTrainID, sizeof(int), NULL);

??? // 取出第二列的数据，因为其长度不定，故必须先获得长度
??? SQLINTEGER lColLen;??? // 字符串的列，即AllStations长度
??? UINT nRow = 0;??// 行标

??? while (SQLFetch(hStmt) == SQL_SUCCESS_WITH_INFO)?//取出一行
??? {
?????????// 获得长度
???????? SQLGetData(hStmt, 2, SQL_CHAR, NULL, 0, &lColLen);
???????? // 根据长度为第二列的数据分配空间
???????? ppcStation[nRow] = ::malloc(lColLen);
?????????// 获得数据
???????? SQLGetData(hStmt, 2, SQL_CHAR, ppcStation[nRow], lColLen, &lColLen);
?????????// 行标随行的Fetch递增
?????????++nRow;
??? }
??? // 现在，我们就取出了所有TrainID和AllStations
??? // 关闭数据库连接，释放资源
??? SQLFreeHandle(SQL_HANDLE_STMT, hStmt);
??? SQLDisConnect(hDbc);
??? SQLFreeHandle(SQL_HANDLE_DBC, hDbc);
??? SQLFreeHandle(SQL_HANDLE_ENV, hEnv);
}
顺便说一句，到目前为止给出的两段代码，都是在方法RETCODE xs_TrainQuery(SRV_PROC*)中，加入这两段代码前，请先将项目自动生成的代码清空。

呼，看来这个问题还是分成两节写，真长。要是累坏了现在坐在电脑前面的这位，他准女朋友…………虽然不会在意，他自己还是在意的啊……