/*須要的表結構以下 CREATE TABLE `test_data` ( `order_id` BIGINT UNSIGNED NOT NULL COMMENT 'Order numbers. Must be unique.', `status` BOOL NOT NULL DEFAULT '0' COMMENT 'Whether data has been used or not. A value of 0 means FALSE.', `date_used` DATETIME NULL COMMENT 'Date/time that the data was used.', UNIQUE ( `order_id` ) ) ENGINE = innodb COMMENT = 'LoadRunner test data'; */ Action() { int rc; int db_connection; // 數據庫鏈接 int query_result; // 查詢結果集 MYSQL_RES char** result_row; // 查詢的數據衕 char *server = "localhost"; char *user = "root"; char *password = "123456"; char *database = "test"; int port = 3306; int unix_socket = NULL; int flags = 0; // 找到libmysql.dll的所在位置. rc = lr_load_dll("C:\\Program Files\\MySQL\\MySQL Server 5.1\\bin\\libmysql.dll"); if (rc != 0) { lr_error_message("Could not load libmysql.dll"); lr_abort(); } // 建立MySQL對象 db_connection = mysql_init(NULL); if (db_connection == NULL) { lr_error_message("Insufficient memory"); lr_abort(); } // 鏈接到MySQL數據庫 rc = mysql_real_connect(db_connection, server, user, password, database, port, unix_socket, flags); if (rc == NULL) { lr_error_message("%s", mysql_error(db_connection)); mysql_close(db_connection); lr_abort(); } // 向數據庫插入數據 // 此處的 {ORDER_ID} 是一個參數,簡單測試時能夠用一個常數代替 lr_save_string (lr_eval_string("INSERT INTO test_data (order_id) VALUES ({ORDER_ID})"),"paramInsertQuery"); rc = mysql_query(db_connection, lr_eval_string("{paramInsertQuery}")); if (rc != 0) { lr_error_message("%s", mysql_error(db_connection)); mysql_close(db_connection); lr_abort(); } // 從數據庫讀取一個數據並顯示 rc = mysql_query(db_connection, "SELECT order_id FROM test_data WHERE status IS FALSE LIMIT 1"); if (rc != 0) { lr_error_message("%s", mysql_error(db_connection)); mysql_close(db_connection); lr_abort(); } query_result = mysql_use_result(db_connection); if (query_result == NULL) { lr_error_message("%s", mysql_error(db_connection)); mysql_free_result(query_result); mysql_close(db_connection); lr_abort(); } // 若是結果集包含多行數據,須要屢次調用 mysql_fetch_row 直到返回NULL result_row = (char **)mysql_fetch_row(query_result); if (result_row == NULL) { lr_error_message("Did not expect the result set to be empty"); mysql_free_result(query_result); mysql_close(db_connection); lr_abort(); } // 保存參數,用於刪除這行數據 lr_save_string(result_row[0], "paramOrderID"); lr_output_message("Order ID is: %s", lr_eval_string("{paramOrderID}")); mysql_free_result(query_result); // 在事務裏更新一行數據,須要用InnoDB引擎 rc = mysql_query(db_connection, "BEGIN"); //啓動事務 if (rc != 0) { lr_error_message("%s", mysql_error(db_connection)); mysql_close(db_connection); lr_abort(); } // 使用 "FOR UPDATE" 鎖住要更新的數據行 rc = mysql_query(db_connection, "SELECT order_id FROM test_data WHERE status IS FALSE LIMIT 1 FOR UPDATE"); if (rc != 0) { lr_error_message("%s", mysql_error(db_connection)); mysql_close(db_connection); lr_abort(); } query_result = mysql_use_result(db_connection); if (query_result == NULL) { lr_error_message("%s", mysql_error(db_connection)); mysql_free_result(query_result); mysql_close(db_connection); lr_abort(); } result_row = (char **)mysql_fetch_row(query_result); if (result_row == NULL) { lr_error_message("沒有查詢到結果"); mysql_free_result(query_result); mysql_close(db_connection); lr_abort(); } lr_save_string(result_row[0], "paramOrderID"); lr_output_message("Order ID is: %s", lr_eval_string("{paramOrderID}")); mysql_free_result(query_result); lr_save_string(lr_eval_string("UPDATE test_data SET status=TRUE, date_used=NOW() WHERE order_id='{paramOrderID}'"),"paramUpdateQuery"); rc = mysql_query(db_connection, lr_eval_string("{paramUpdateQuery}")); if (rc != 0) { lr_error_message("%s", mysql_error(db_connection)); mysql_close(db_connection); lr_abort(); } rc = mysql_query(db_connection, "COMMIT"); // 提交事務 if (rc != 0) { lr_error_message("%s", mysql_error(db_connection)); mysql_close(db_connection); lr_abort(); } // 再次查找數據,應該爲空了,由於前面的事務更新了標誌 rc = mysql_query(db_connection, "SELECT order_id FROM test_data WHERE status IS FALSE LIMIT 1"); if (rc != 0) { lr_error_message("%s", mysql_error(db_connection)); mysql_close(db_connection); lr_abort(); } query_result = mysql_use_result(db_connection); if (query_result == NULL) { lr_error_message("%s", mysql_error(db_connection)); mysql_free_result(query_result); mysql_close(db_connection); lr_abort(); } result_row = (char **)mysql_fetch_row(query_result); if (result_row == NULL) { lr_output_message("Result set is empty as expected"); mysql_free_result(query_result); } else { lr_error_message("Did not expect the result set to contain any rows"); mysql_free_result(query_result); mysql_close(db_connection); lr_abort(); } // 刪除數據 lr_save_string(lr_eval_string("DELETE FROM test_data WHERE order_id = '{paramOrderID}'"),"paramDeleteQuery"); rc = mysql_query(db_connection, lr_eval_string("{paramDeleteQuery}")); if (rc != 0) { lr_error_message("%s", mysql_error(db_connection)); mysql_close(db_connection); lr_abort(); } // 釋放MySQL資源 mysql_close(db_connection); return 0; } /*須要的表結構以下 CREATE TABLE `test_data` ( `order_id` BIGINT UNSIGNED NOT NULL COMMENT 'Order numbers. Must be unique.', `status` BOOL NOT NULL DEFAULT '0' COMMENT 'Whether data has been used or not. A value of 0 means FALSE.', `date_used` DATETIME NULL COMMENT 'Date/time that the data was used.', UNIQUE ( `order_id` ) ) ENGINE = innodb COMMENT = 'LoadRunner test data'; */ Action() { int rc; int db_connection; // 數據庫鏈接 int query_result; // 查詢結果集 MYSQL_RES char** result_row; // 查詢的數據衕 char *server = "localhost"; char *user = "root"; char *password = "123456"; char *database = "test"; int port = 3306; int unix_socket = NULL; int flags = 0; // 找到libmysql.dll的所在位置. rc = lr_load_dll("C:\\Program Files\\MySQL\\MySQL Server 5.1\\bin\\libmysql.dll"); if (rc != 0) { lr_error_message("Could not load libmysql.dll"); lr_abort(); } // 建立MySQL對象 db_connection = mysql_init(NULL); if (db_connection == NULL) { lr_error_message("Insufficient memory"); lr_abort(); } // 鏈接到MySQL數據庫 rc = mysql_real_connect(db_connection, server, user, password, database, port, unix_socket, flags); if (rc == NULL) { lr_error_message("%s", mysql_error(db_connection)); mysql_close(db_connection); lr_abort(); } // 向數據庫插入數據 // 此處的 {ORDER_ID} 是一個參數,簡單測試時能夠用一個常數代替 lr_save_string (lr_eval_string("INSERT INTO test_data (order_id) VALUES ({ORDER_ID})"),"paramInsertQuery"); rc = mysql_query(db_connection, lr_eval_string("{paramInsertQuery}")); if (rc != 0) { lr_error_message("%s", mysql_error(db_connection)); mysql_close(db_connection); lr_abort(); } // 從數據庫讀取一個數據並顯示 rc = mysql_query(db_connection, "SELECT order_id FROM test_data WHERE status IS FALSE LIMIT 1"); if (rc != 0) { lr_error_message("%s", mysql_error(db_connection)); mysql_close(db_connection); lr_abort(); } query_result = mysql_use_result(db_connection); if (query_result == NULL) { lr_error_message("%s", mysql_error(db_connection)); mysql_free_result(query_result); mysql_close(db_connection); lr_abort(); } // 若是結果集包含多行數據,須要屢次調用 mysql_fetch_row 直到返回NULL result_row = (char **)mysql_fetch_row(query_result); if (result_row == NULL) { lr_error_message("Did not expect the result set to be empty"); mysql_free_result(query_result); mysql_close(db_connection); lr_abort(); } // 保存參數,用於刪除這行數據 lr_save_string(result_row[0], "paramOrderID"); lr_output_message("Order ID is: %s", lr_eval_string("{paramOrderID}")); mysql_free_result(query_result); // 在事務裏更新一行數據,須要用InnoDB引擎 rc = mysql_query(db_connection, "BEGIN"); //啓動事務 if (rc != 0) { lr_error_message("%s", mysql_error(db_connection)); mysql_close(db_connection); lr_abort(); } // 使用 "FOR UPDATE" 鎖住要更新的數據行 rc = mysql_query(db_connection, "SELECT order_id FROM test_data WHERE status IS FALSE LIMIT 1 FOR UPDATE"); if (rc != 0) { lr_error_message("%s", mysql_error(db_connection)); mysql_close(db_connection); lr_abort(); } query_result = mysql_use_result(db_connection); if (query_result == NULL) { lr_error_message("%s", mysql_error(db_connection)); mysql_free_result(query_result); mysql_close(db_connection); lr_abort(); } result_row = (char **)mysql_fetch_row(query_result); if (result_row == NULL) { lr_error_message("沒有查詢到結果"); mysql_free_result(query_result); mysql_close(db_connection); lr_abort(); } lr_save_string(result_row[0], "paramOrderID"); lr_output_message("Order ID is: %s", lr_eval_string("{paramOrderID}")); mysql_free_result(query_result); lr_save_string(lr_eval_string("UPDATE test_data SET status=TRUE, date_used=NOW() WHERE order_id='{paramOrderID}'"),"paramUpdateQuery"); rc = mysql_query(db_connection, lr_eval_string("{paramUpdateQuery}")); if (rc != 0) { lr_error_message("%s", mysql_error(db_connection)); mysql_close(db_connection); lr_abort(); } rc = mysql_query(db_connection, "COMMIT"); // 提交事務 if (rc != 0) { lr_error_message("%s", mysql_error(db_connection)); mysql_close(db_connection); lr_abort(); } // 再次查找數據,應該爲空了,由於前面的事務更新了標誌 rc = mysql_query(db_connection, "SELECT order_id FROM test_data WHERE status IS FALSE LIMIT 1"); if (rc != 0) { lr_error_message("%s", mysql_error(db_connection)); mysql_close(db_connection); lr_abort(); } query_result = mysql_use_result(db_connection); if (query_result == NULL) { lr_error_message("%s", mysql_error(db_connection)); mysql_free_result(query_result); mysql_close(db_connection); lr_abort(); } result_row = (char **)mysql_fetch_row(query_result); if (result_row == NULL) { lr_output_message("Result set is empty as expected"); mysql_free_result(query_result); } else { lr_error_message("Did not expect the result set to contain any rows"); mysql_free_result(query_result); mysql_close(db_connection); lr_abort(); } // 刪除數據 lr_save_string(lr_eval_string("DELETE FROM test_data WHERE order_id = '{paramOrderID}'"),"paramDeleteQuery"); rc = mysql_query(db_connection, lr_eval_string("{paramDeleteQuery}")); if (rc != 0) { lr_error_message("%s", mysql_error(db_connection)); mysql_close(db_connection); lr_abort(); } // 釋放MySQL資源 mysql_close(db_connection); return 0; }