opensource/HeidiSQL
1
0
Fork 0
mirror of https://github.com/HeidiSQL/HeidiSQL.git synced 2025-08-06 18:24:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

Avoid destructive actions (DROP before CREATE) when user tried to do a 'recreate' action on two sessions connected to the same database. Fixes issue #703.

Browse Source
This commit is contained in:
rosenfield.albert
2008-09-01 17:51:25 +00:00
parent 85075d5be7
commit 020b620c47
1 changed files with 31 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
source/exportsql.pas
View File

@ -488,6 +488,8 @@ var
sourceDb, destDb : WideString;
which : Integer;
tofile,todb,tohost : boolean;
samehost : boolean;
sameuuid : String;
tcount,tablecounter : Integer;
win2export : THandle;
StrProgress : String;
@ -653,6 +655,35 @@ begin
// todo: test this
current_characterset := 'binary';
{***
Check if source and destination session is connected to the same server.
}
samehost := todb;
if tohost then begin
sameuuid := cwin.GetVar('SELECT UUID()');
i := StrToInt(cwin.GetVar('SELECT GET_LOCK(' + esc(sameuuid) + ', 5)'));
if i = 0 then raise Exception.Create('Could not create a server lock.');
query := RemoteExecQuery(
win2export,
'SELECT IS_FREE_LOCK(' + esc(sameuuid) + ')',
'Checking for same server...'
);
i := query.Fields[0].AsInteger;
if i = 0 then samehost := true;
query.Free;
cwin.ExecuteNonQuery('SELECT RELEASE_LOCK(' + esc(sameuuid) + ')');
end;
{***
Avoid destructive actions (DROP before CREATE) on same host and database.
}
if tohost and samehost then begin
if exportdb and (comboDatabase.ItemIndex = DB_DROP_CREATE) then raise Exception.Create('Aborted: selected action "database recreate" on same source/destination host would drop source database.');
if exporttables and (comboTables.ItemIndex = TAB_DROP_CREATE) then begin
if sourceDb = destDb then raise Exception.Create('Aborted: selected action "tables recreate" on same source/destination host and database would drop source tables.');
end;
end;
{***
Some actions which are only needed if we're not in OtherDatabase-mode:
Set character set, create and use database.