mirror of
https://github.com/HeidiSQL/HeidiSQL.git
synced 2025-08-06 18:24:26 +08:00
Move *.pas, *.dfm and *.inc files from root directory to a new "source" subdirectory. Leaving just the readme in the root to give all newbies a very clear and unique starting point.
This commit is contained in:
Ansgar Becker
371
source/mysqlquerythread.pas
Normal file
371
source/mysqlquerythread.pas
Normal file
@ -0,0 +1,371 @@
|
||||
unit MysqlQueryThread;
|
||||
|
||||
interface
|
||||
|
||||
uses
|
||||
Windows, Messages, Forms, Db, Classes, ZConnection, ZDataSet, StdCtrls, SysUtils;
|
||||
|
||||
{$IFDEF EXAMPLE_APP}
|
||||
const
|
||||
WM_MYSQL_THREAD_NOTIFY = WM_USER+100;
|
||||
{$ENDIF}
|
||||
|
||||
type
|
||||
// Exception information
|
||||
TExceptionData = record
|
||||
Msg : String[200];
|
||||
HelpContext : Integer;
|
||||
end;
|
||||
|
||||
// Mysql protocol-relevant connection parameter structure
|
||||
TMysqlConnParams = record
|
||||
Host,
|
||||
Database,
|
||||
Protocol,
|
||||
User,
|
||||
Pass : String;
|
||||
Port : Integer;
|
||||
PrpCompress,
|
||||
PrpTimeout,
|
||||
PrpDbless,
|
||||
PrpClientLocalFiles,
|
||||
PrpClientInteractive : String;
|
||||
end;
|
||||
PMysqlConnParams = ^TMysqlConnParams;
|
||||
|
||||
// Established connection and it's corresponding connection profile.
|
||||
// (The actual connection need not necessarily be open, of course, it could in theory be closed or nil,
|
||||
// but to keep the name short; this beats "TConnectionProfileDataAndConnectionObject", which I guess would be the proper name.
|
||||
TOpenConnProf = record
|
||||
MysqlParams : TMysqlConnParams; // stuff that needs to be shipped over to the mysql driver.
|
||||
DatabaseList : String;
|
||||
DatabaseListSort : Boolean;
|
||||
Description : String;
|
||||
MysqlConn : TZConnection;
|
||||
end;
|
||||
POpenConnProf = ^TOpenConnProf;
|
||||
|
||||
TThreadResult = record
|
||||
ThreadID : Integer;
|
||||
ConnectionID : Cardinal;
|
||||
Action : Integer;
|
||||
Sql : String;
|
||||
Result : Integer;
|
||||
Comment : String;
|
||||
end;
|
||||
|
||||
TMysqlConnectThread = class(TThread)
|
||||
private
|
||||
protected
|
||||
public
|
||||
end;
|
||||
|
||||
TMysqlQueryThread = class(TThread)
|
||||
private
|
||||
FMysqlConn : TZConnection;
|
||||
FConn : TOpenConnProf;
|
||||
FOwner : TObject; // TMysqlQuery object
|
||||
FSql : String;
|
||||
FResult : Integer;
|
||||
FComment : String;
|
||||
FSyncMode : Integer;
|
||||
FNotifyWndHandle : THandle;
|
||||
procedure ReportInit;
|
||||
procedure ReportStart;
|
||||
procedure ReportFinished;
|
||||
procedure ReportFreed;
|
||||
function GetExceptionData(AException : Exception) : TExceptionData;
|
||||
protected
|
||||
procedure Execute; override;
|
||||
procedure SetState (AResult : Integer; AComment : String);
|
||||
procedure SetNotifyWndHandle (Value : THandle);
|
||||
procedure NotifyStatus (AEvent : Integer);
|
||||
procedure NotifyStatusViaEventProc (AEvent : Integer);
|
||||
procedure NotifyStatusViaWinMessage (AEvent : Integer);
|
||||
function AssembleResult () : TThreadResult;
|
||||
function RunDataQuery (ASql : String; var ADataset : TDataset; out AExceptionData : TExceptionData) : Boolean;
|
||||
function RunUpdateQuery (ASql : String; var ADataset : TDataset; out AExceptionData : TExceptionData) : Boolean;
|
||||
function QuerySingleCellAsInteger (ASql : String) : Integer;
|
||||
public
|
||||
constructor Create (AOwner : TObject; AConn : TOpenConnProf; ASql : String; ASyncMode : Integer);
|
||||
destructor Destroy; override;
|
||||
property NotifyWndHandle : THandle read FNotifyWndHandle write SetNotifyWndHandle;
|
||||
end;
|
||||
|
||||
implementation
|
||||
|
||||
uses
|
||||
MysqlQuery, Dialogs, helpers, communication
|
||||
{$IFNDEF EXAMPLE_APP}
|
||||
, Main
|
||||
{$ENDIF}
|
||||
;
|
||||
|
||||
function TMysqlQueryThread.AssembleResult: TThreadResult;
|
||||
begin
|
||||
ZeroMemory (@Result,SizeOf(Result));
|
||||
|
||||
Result.ThreadID := ThreadID;
|
||||
try
|
||||
Result.ConnectionID := FMysqlConn.GetThreadId;
|
||||
except
|
||||
end;
|
||||
Result.Action := 1;
|
||||
Result.Sql := FSql;
|
||||
Result.Result := FResult;
|
||||
Result.Comment := FComment;
|
||||
end;
|
||||
|
||||
constructor TMysqlQueryThread.Create(AOwner : TObject; AConn : TOpenConnProf; ASql : String; ASyncMode : Integer);
|
||||
var
|
||||
mc : TZConnection;
|
||||
begin
|
||||
Inherited Create(True);
|
||||
|
||||
FOwner := AOwner;
|
||||
FConn := AConn;
|
||||
FSyncMode := ASyncMode;
|
||||
mc := TMysqlQuery(FOwner).MysqlConnection;
|
||||
FMysqlConn := mc;
|
||||
FResult := 0;
|
||||
FSql := ASql;
|
||||
|
||||
mc.HostName := AConn.MysqlParams.Host;
|
||||
mc.Database := AConn.MysqlParams.Database;
|
||||
mc.User := AConn.MysqlParams.User;
|
||||
mc.Password := AConn.MysqlParams.Pass;
|
||||
mc.Protocol := AConn.MysqlParams.Protocol;
|
||||
mc.Port := AConn.MysqlParams.Port;
|
||||
|
||||
FreeOnTerminate := True;
|
||||
|
||||
end;
|
||||
|
||||
destructor TMysqlQueryThread.Destroy;
|
||||
begin
|
||||
inherited;
|
||||
end;
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
procedure TMysqlQueryThread.NotifyStatus(AEvent: Integer);
|
||||
var
|
||||
h : THandle;
|
||||
begin
|
||||
|
||||
// trigger query finished event
|
||||
if (FSyncMode=MQM_SYNC) and (AEvent=MQE_FREED) then
|
||||
begin
|
||||
h := OpenEvent (EVENT_ALL_ACCESS,False,PChar(TMysqlQuery(FOwner).EventName));
|
||||
|
||||
if h<>0 then
|
||||
SetEvent (h);
|
||||
end;
|
||||
|
||||
case TMysqlQuery(FOwner).NotificationMode of
|
||||
MQN_EVENTPROC: NotifyStatusViaEventProc(AEvent);
|
||||
MQN_WINMESSAGE: NotifyStatusViaWinMessage(AEvent);
|
||||
end;
|
||||
|
||||
end;
|
||||
|
||||
procedure TMysqlQueryThread.NotifyStatusViaEventProc(AEvent: Integer);
|
||||
begin
|
||||
if FSyncMode=MQM_ASYNC then
|
||||
begin
|
||||
case AEvent of
|
||||
MQE_INITED: Synchronize(ReportInit);
|
||||
MQE_STARTED: Synchronize(ReportStart);
|
||||
MQE_FINISHED: Synchronize(ReportFinished);
|
||||
MQE_FREED: Synchronize(ReportFreed);
|
||||
end;
|
||||
|
||||
end;
|
||||
end;
|
||||
|
||||
procedure TMysqlQueryThread.NotifyStatusViaWinMessage(AEvent: Integer);
|
||||
var
|
||||
qr : TThreadResult;
|
||||
begin
|
||||
qr := AssembleResult();
|
||||
debug(Format('qry: Setting result and posting status %d via WM_MYSQL_THREAD_NOTIFY message', [AEvent]));
|
||||
TMysqlQuery(FOwner).SetThreadResult(qr);
|
||||
PostMessage(FNotifyWndHandle,WM_MYSQL_THREAD_NOTIFY,Integer(FOwner),AEvent);
|
||||
end;
|
||||
|
||||
procedure TMysqlQueryThread.Execute;
|
||||
var
|
||||
q : TZQuery;
|
||||
r : Boolean;
|
||||
e : TExceptionData;
|
||||
begin
|
||||
debug(Format('qry: Thread %d running...', [ThreadID]));
|
||||
NotifyStatus(MQE_INITED);
|
||||
|
||||
try
|
||||
if not FMysqlConn.Connected then
|
||||
FMysqlConn.Connect();
|
||||
except
|
||||
on E: Exception do
|
||||
begin
|
||||
SetState (MQR_CONNECT_FAIL,Format('%s',[E.Message]));
|
||||
end;
|
||||
end;
|
||||
|
||||
if FMysqlConn.Connected then
|
||||
begin
|
||||
NotifyStatus (MQE_STARTED);
|
||||
|
||||
q := nil;
|
||||
if ExpectResultSet(FSql) then
|
||||
begin
|
||||
r := RunDataQuery (FSql,TDataSet(q),e);
|
||||
|
||||
if r then
|
||||
begin
|
||||
if q.State=dsBrowse then
|
||||
begin
|
||||
// WTF?
|
||||
end;
|
||||
end;
|
||||
|
||||
end
|
||||
else
|
||||
r := RunUpdateQuery (FSql,TDataSet(q),e);
|
||||
TMysqlQuery(FOwner).SetMysqlDataset(q);
|
||||
|
||||
if r then
|
||||
SetState (MQR_SUCCESS,'SUCCESS')
|
||||
else
|
||||
SetState (MQR_QUERY_FAIL,e.Msg);
|
||||
|
||||
end;
|
||||
|
||||
NotifyStatus (MQE_FINISHED);
|
||||
NotifyStatus (MQE_FREED);
|
||||
debug(Format('qry: Thread %d suspending.', [ThreadID]));
|
||||
end;
|
||||
|
||||
|
||||
function TMysqlQueryThread.GetExceptionData(
|
||||
AException: Exception): TExceptionData;
|
||||
begin
|
||||
ZeroMemory (@Result,SizeOf(Result));
|
||||
Result.Msg := AException.Message;
|
||||
Result.HelpContext := AException.HelpContext;
|
||||
end;
|
||||
|
||||
function TMysqlQueryThread.QuerySingleCellAsInteger(ASql: String): Integer;
|
||||
var
|
||||
ds : TDataSet;
|
||||
e : TExceptionData;
|
||||
begin
|
||||
Result := 0;
|
||||
|
||||
if RunDataQuery(ASql,ds,e) then
|
||||
begin
|
||||
if ds.Fields.Count > 0 then
|
||||
Result := ds.Fields[0].AsInteger;
|
||||
FreeAndNil (ds);
|
||||
end;
|
||||
end;
|
||||
|
||||
procedure TMysqlQueryThread.ReportStart();
|
||||
var
|
||||
qr : TThreadResult;
|
||||
begin
|
||||
qr := AssembleResult();
|
||||
|
||||
if FOwner <> nil then
|
||||
TMysqlQuery (FOwner).PostNotification(qr,MQE_STARTED);
|
||||
end;
|
||||
|
||||
procedure TMysqlQueryThread.ReportFinished();
|
||||
var
|
||||
qr : TThreadResult;
|
||||
begin
|
||||
qr := AssembleResult();
|
||||
|
||||
if FOwner <> nil then
|
||||
TMysqlQuery (FOwner).PostNotification(qr,MQE_FINISHED);
|
||||
end;
|
||||
|
||||
procedure TMysqlQueryThread.ReportInit();
|
||||
var
|
||||
qr : TThreadResult;
|
||||
begin
|
||||
qr := AssembleResult();
|
||||
|
||||
if FOwner <> nil then
|
||||
TMysqlQuery (FOwner).PostNotification(qr,MQE_INITED);
|
||||
end;
|
||||
|
||||
procedure TMysqlQueryThread.ReportFreed;
|
||||
var
|
||||
qr : TThreadResult;
|
||||
begin
|
||||
qr := AssembleResult();
|
||||
|
||||
if FOwner <> nil then
|
||||
TMysqlQuery (FOwner).PostNotification(qr,MQE_FREED);
|
||||
end;
|
||||
|
||||
function TMysqlQueryThread.RunDataQuery(ASql: String;
|
||||
var ADataset: TDataset; out AExceptionData : TExceptionData): Boolean;
|
||||
var
|
||||
q : TZQuery;
|
||||
begin
|
||||
Result := False;
|
||||
q := TZQuery.Create(nil);
|
||||
q.Connection := FMysqlConn;
|
||||
q.SQL.Text := ASql;
|
||||
ADataset := q;
|
||||
|
||||
try
|
||||
q.Active := True;
|
||||
Result := True;
|
||||
except
|
||||
on E: Exception do
|
||||
begin
|
||||
AExceptionData := GetExceptionData(E);
|
||||
end;
|
||||
// EZSQLException
|
||||
end;
|
||||
end;
|
||||
|
||||
function TMysqlQueryThread.RunUpdateQuery(ASql: String; var ADataset: TDataset; out AExceptionData : TExceptionData): Boolean;
|
||||
var
|
||||
q : TZQuery;
|
||||
begin
|
||||
Result := False;
|
||||
q := TZQuery.Create(nil);
|
||||
q.Connection := FMysqlConn;
|
||||
q.SQL.Text := ASql;
|
||||
ADataSet := q;
|
||||
|
||||
try
|
||||
q.ExecSQL();
|
||||
Result := True;
|
||||
except
|
||||
On E: Exception do
|
||||
AExceptionData := GetExceptionData(E);
|
||||
//MessageDlg( 'SQL Error: '+ CRLF + E.Message, mtError, [mbOK], 0 );
|
||||
end;
|
||||
|
||||
FreeAndNil (q);
|
||||
end;
|
||||
|
||||
procedure TMysqlQueryThread.SetNotifyWndHandle(Value: THandle);
|
||||
begin
|
||||
FNotifyWndHandle := Value;
|
||||
end;
|
||||
|
||||
procedure TMysqlQueryThread.SetState(AResult: Integer; AComment: String);
|
||||
begin
|
||||
FResult := AResult;
|
||||
FComment := AComment;
|
||||
end;
|
||||
|
||||
end.
|
||||
Reference in New Issue
Block a user