mirror of
https://github.com/HeidiSQL/HeidiSQL.git
synced 2025-08-06 18:24:26 +08:00
359 lines
12 KiB
ObjectPascal
359 lines
12 KiB
ObjectPascal
unit tabletools;
|
|
|
|
|
|
// -------------------------------------
|
|
// Table-diagnostics
|
|
// -------------------------------------
|
|
|
|
|
|
interface
|
|
|
|
uses
|
|
Windows, SysUtils, Classes, Controls, Forms, StdCtrls, ComCtrls, Buttons,
|
|
WideStrings, WideStrUtils, VirtualTrees, ExtCtrls, Db, Contnrs, Graphics;
|
|
|
|
type
|
|
TfrmTableTools = class(TForm)
|
|
btnClose: TButton;
|
|
pnlTop: TPanel;
|
|
TreeObjects: TVirtualStringTree;
|
|
spltHorizontally: TSplitter;
|
|
pnlRight: TPanel;
|
|
ResultGrid: TVirtualStringTree;
|
|
lblResults: TLabel;
|
|
PageControlTools: TPageControl;
|
|
Maintenance: TTabSheet;
|
|
comboOperation: TComboBox;
|
|
lblOperation: TLabel;
|
|
chkQuick: TCheckBox;
|
|
chkFast: TCheckBox;
|
|
chkMedium: TCheckBox;
|
|
chkExtended: TCheckBox;
|
|
chkChanged: TCheckBox;
|
|
btnExecuteMaintenance: TButton;
|
|
chkUseFrm: TCheckBox;
|
|
lblOptions: TLabel;
|
|
btnHelp: TButton;
|
|
procedure FormDestroy(Sender: TObject);
|
|
procedure FormCreate(Sender: TObject);
|
|
procedure FormShow(Sender: TObject);
|
|
procedure btnHelpClick(Sender: TObject);
|
|
procedure TreeObjectsGetText(Sender: TBaseVirtualTree; Node: PVirtualNode; Column: TColumnIndex;
|
|
TextType: TVSTTextType; var CellText: WideString);
|
|
procedure TreeObjectsInitNode(Sender: TBaseVirtualTree; ParentNode, Node: PVirtualNode;
|
|
var InitialStates: TVirtualNodeInitStates);
|
|
procedure TreeObjectsGetImageIndex(Sender: TBaseVirtualTree; Node: PVirtualNode; Kind: TVTImageKind;
|
|
Column: TColumnIndex; var Ghosted: Boolean; var ImageIndex: Integer);
|
|
procedure TreeObjectsInitChildren(Sender: TBaseVirtualTree; Node: PVirtualNode; var ChildCount: Cardinal);
|
|
procedure comboOperationChange(Sender: TObject);
|
|
procedure ExecuteOperation(Sender: TObject);
|
|
procedure ResultGridInitNode(Sender: TBaseVirtualTree; ParentNode, Node: PVirtualNode;
|
|
var InitialStates: TVirtualNodeInitStates);
|
|
procedure ResultGridGetNodeDataSize(Sender: TBaseVirtualTree; var NodeDataSize: Integer);
|
|
procedure ResultGridGetText(Sender: TBaseVirtualTree; Node: PVirtualNode; Column: TColumnIndex;
|
|
TextType: TVSTTextType; var CellText: WideString);
|
|
procedure TreeObjectsChecked(Sender: TBaseVirtualTree; Node: PVirtualNode);
|
|
procedure MaintenanceOptionClick(Sender: TObject);
|
|
procedure ResultGridHeaderClick(Sender: TVTHeader; HitInfo: TVTHeaderHitInfo);
|
|
procedure ResultGridCompareNodes(Sender: TBaseVirtualTree; Node1, Node2: PVirtualNode;
|
|
Column: TColumnIndex; var Result: Integer);
|
|
procedure ResultGridPaintText(Sender: TBaseVirtualTree; const TargetCanvas: TCanvas; Node: PVirtualNode;
|
|
Column: TColumnIndex; TextType: TVSTTextType);
|
|
private
|
|
{ Private declarations }
|
|
FResults: TObjectList;
|
|
procedure ValidateControls(Sender: TObject);
|
|
procedure ProcessTableNode(Sender: TObject; Node: PVirtualNode);
|
|
procedure AddResults(SQL: WideString);
|
|
public
|
|
{ Public declarations }
|
|
end;
|
|
|
|
|
|
implementation
|
|
|
|
uses main, helpers;
|
|
|
|
{$R *.DFM}
|
|
|
|
|
|
procedure TfrmTableTools.FormCreate(Sender: TObject);
|
|
begin
|
|
// Restore GUI setup
|
|
Width := GetRegValue(REGNAME_TOOLSWINWIDTH, Width);
|
|
Height := GetRegValue(REGNAME_TOOLSWINHEIGHT, Height);
|
|
TreeObjects.Width := GetRegValue(REGNAME_TOOLSTREEWIDTH, TreeObjects.Width);
|
|
SetWindowSizeGrip( Self.Handle, True );
|
|
InheritFont(Font);
|
|
FixVT(TreeObjects);
|
|
FixVT(ResultGrid);
|
|
TreeObjects.RootNodeCount := Mainform.DBtree.RootNodeCount;
|
|
FResults := TObjectList.Create;
|
|
end;
|
|
|
|
|
|
procedure TfrmTableTools.FormDestroy(Sender: TObject);
|
|
begin
|
|
// Save GUI setup
|
|
OpenRegistry;
|
|
MainReg.WriteInteger( REGNAME_TOOLSWINWIDTH, Width );
|
|
MainReg.WriteInteger( REGNAME_TOOLSWINHEIGHT, Height );
|
|
MainReg.WriteInteger( REGNAME_TOOLSTREEWIDTH, TreeObjects.Width);
|
|
end;
|
|
|
|
|
|
procedure TfrmTableTools.FormShow(Sender: TObject);
|
|
begin
|
|
// When this form is displayed the second time, databases may be deleted or filtered
|
|
treeObjects.ReinitChildren(treeObjects.GetFirst, False);
|
|
// CHECKSUM available since MySQL 4.1.1
|
|
if Mainform.mysql_version < 40101 then
|
|
comboOperation.Items[comboOperation.Items.IndexOf('Checksum')] := 'Checksum ('+STR_NOTSUPPORTED+')';
|
|
comboOperation.OnChange(Sender);
|
|
end;
|
|
|
|
|
|
procedure TfrmTableTools.comboOperationChange(Sender: TObject);
|
|
var
|
|
op: String;
|
|
begin
|
|
// Only enable available options
|
|
op := LowerCase(comboOperation.Text);
|
|
chkQuick.Enabled := (op = 'check') or (op = 'checksum') or (op = 'repair');
|
|
chkFast.Enabled := op = 'check';
|
|
chkMedium.Enabled := op = 'check';
|
|
chkExtended.Enabled := (op = 'check') or (op = 'checksum') or (op = 'repair');
|
|
chkChanged.Enabled := op = 'changed';
|
|
chkUseFrm.Enabled := op = 'repair';
|
|
ValidateControls(Sender);
|
|
end;
|
|
|
|
|
|
procedure TfrmTableTools.ValidateControls(Sender: TObject);
|
|
begin
|
|
btnExecuteMaintenance.Enabled := (Pos(STR_NOTSUPPORTED, comboOperation.Text) = 0) and
|
|
(TreeObjects.CheckedCount > 0);
|
|
// CHECKSUM's options are mutually exclusive
|
|
if comboOperation.Text = 'Checksum' then begin
|
|
if (Sender = chkExtended) and chkExtended.Checked then chkQuick.Checked := False
|
|
else if chkQuick.Checked then chkExtended.Checked := False;
|
|
end;
|
|
end;
|
|
|
|
|
|
procedure TfrmTableTools.TreeObjectsChecked(Sender: TBaseVirtualTree; Node: PVirtualNode);
|
|
begin
|
|
ValidateControls(Sender);
|
|
end;
|
|
|
|
|
|
procedure TfrmTableTools.TreeObjectsGetImageIndex(Sender: TBaseVirtualTree; Node: PVirtualNode;
|
|
Kind: TVTImageKind; Column: TColumnIndex; var Ghosted: Boolean; var ImageIndex: Integer);
|
|
begin
|
|
Mainform.DBtreeGetImageIndex(Sender, Node, Kind, Column, Ghosted, ImageIndex);
|
|
end;
|
|
|
|
|
|
procedure TfrmTableTools.TreeObjectsGetText(Sender: TBaseVirtualTree; Node: PVirtualNode;
|
|
Column: TColumnIndex; TextType: TVSTTextType; var CellText: WideString);
|
|
begin
|
|
Mainform.DBtreeGetText(Sender, Node, Column, TextType, CellText);
|
|
end;
|
|
|
|
|
|
procedure TfrmTableTools.TreeObjectsInitChildren(Sender: TBaseVirtualTree; Node: PVirtualNode;
|
|
var ChildCount: Cardinal);
|
|
begin
|
|
Mainform.DBtreeInitChildren(Sender, Node, ChildCount);
|
|
end;
|
|
|
|
|
|
procedure TfrmTableTools.TreeObjectsInitNode(Sender: TBaseVirtualTree; ParentNode, Node: PVirtualNode;
|
|
var InitialStates: TVirtualNodeInitStates);
|
|
var
|
|
ds: TDataset;
|
|
begin
|
|
// Attach a checkbox to all nodes
|
|
Mainform.DBtreeInitNode(Sender, ParentNode, Node, InitialStates);
|
|
Node.CheckType := ctTriStateCheckBox;
|
|
Node.CheckState := csUncheckedNormal;
|
|
case Sender.GetNodeLevel(Node) of
|
|
1: begin
|
|
// Preselect active database
|
|
if Mainform.Databases[Node.Index] = Mainform.ActiveDatabase then begin
|
|
Node.CheckState := csCheckedNormal;
|
|
TreeObjects.ReinitChildren(Node, False);
|
|
end;
|
|
ValidateControls(Sender);
|
|
end;
|
|
2: begin
|
|
// Disable stored routines
|
|
ds := Mainform.FetchDbTableList(Mainform.Databases[ParentNode.Index]);
|
|
ds.RecNo := Node.Index+1;
|
|
if not (GetDBObjectType(ds.Fields) in [lntTable, lntCrashedTable, lntView]) then
|
|
Node.CheckType := ctNone
|
|
else if Node.Parent.CheckState in [csCheckedNormal, csCheckedPressed] then
|
|
Node.CheckState := csCheckedNormal;
|
|
end;
|
|
end;
|
|
end;
|
|
|
|
|
|
procedure TfrmTableTools.btnHelpClick(Sender: TObject);
|
|
begin
|
|
Mainform.CallSQLHelpWithKeyword(UpperCase(comboOperation.Text) + ' TABLE');
|
|
end;
|
|
|
|
|
|
procedure TfrmTableTools.MaintenanceOptionClick(Sender: TObject);
|
|
begin
|
|
ValidateControls(Sender);
|
|
end;
|
|
|
|
procedure TfrmTableTools.ExecuteOperation(Sender: TObject);
|
|
var
|
|
DBNode, TableNode: PVirtualNode;
|
|
begin
|
|
Screen.Cursor := crHourGlass;
|
|
ResultGrid.Clear;
|
|
FResults.Clear;
|
|
TreeObjects.SetFocus;
|
|
DBNode := TreeObjects.GetFirstChild(TreeObjects.GetFirst);
|
|
while Assigned(DBNode) do begin
|
|
if not (DBNode.CheckState in [csUncheckedNormal, csUncheckedPressed]) then begin
|
|
TableNode := TreeObjects.GetFirstChild(DBNode);
|
|
while Assigned(TableNode) do begin
|
|
ProcessTableNode(Sender, TableNode);
|
|
TableNode := TreeObjects.GetNextSibling(TableNode);
|
|
end;
|
|
end;
|
|
DBNode := TreeObjects.GetNextSibling(DBNode);
|
|
end;
|
|
Screen.Cursor := crDefault;
|
|
end;
|
|
|
|
|
|
procedure TfrmTableTools.ProcessTableNode(Sender: TObject; Node: PVirtualNode);
|
|
var
|
|
SQL: WideString;
|
|
begin
|
|
// Prepare SQL for one table node
|
|
if (csCheckedNormal in [Node.CheckState, Node.Parent.CheckState]) and
|
|
(Node.CheckType <> ctNone) then begin
|
|
SQL := UpperCase(comboOperation.Text) + ' TABLE ' +
|
|
Mainform.mask(TreeObjects.Text[Node.Parent, 0])+'.'+Mainform.mask(TreeObjects.Text[Node, 0]);
|
|
if chkQuick.Enabled and chkQuick.Checked then SQL := SQL + ' QUICK';
|
|
if chkFast.Enabled and chkFast.Checked then SQL := SQL + ' FAST';
|
|
if chkMedium.Enabled and chkMedium.Checked then SQL := SQL + ' MEDIUM';
|
|
if chkExtended.Enabled and chkExtended.Checked then SQL := SQL + ' EXTENDED';
|
|
if chkChanged.Enabled and chkChanged.Checked then SQL := SQL + ' CHANGED';
|
|
if chkUseFrm.Enabled and chkUseFrm.Checked then SQL := SQL + ' USE_FRM';
|
|
AddResults(SQL);
|
|
end;
|
|
end;
|
|
|
|
|
|
procedure TfrmTableTools.AddResults(SQL: WideString);
|
|
var
|
|
i: Integer;
|
|
Col: TVirtualTreeColumn;
|
|
Row: TWideStringlist;
|
|
ds: TDataset;
|
|
begin
|
|
// Execute query and append results into grid
|
|
ds := Mainform.GetResults(SQL);
|
|
if ds = nil then
|
|
Exit;
|
|
|
|
// Add missing columns
|
|
for i:=ResultGrid.Header.Columns.Count to ds.FieldCount-1 do begin
|
|
Col := ResultGrid.Header.Columns.Add;
|
|
Col.Width := 100;
|
|
end;
|
|
// Remove superfluous columns
|
|
for i:=ResultGrid.Header.Columns.Count-1 downto ds.FieldCount do
|
|
ResultGrid.Header.Columns[i].Free;
|
|
// Set column header names
|
|
for i:=0 to ds.FieldCount-1 do begin
|
|
Col := ResultGrid.Header.Columns[i];
|
|
Col.Text := ds.Fields[i].FieldName;
|
|
if ds.Fields[i].DataType in [ftSmallint, ftInteger, ftWord, ftLargeint, ftFloat] then
|
|
Col.Alignment := taRightJustify;
|
|
end;
|
|
ds.First;
|
|
while not ds.Eof do begin
|
|
Row := TWideStringlist.Create;
|
|
for i:=0 to ds.FieldCount-1 do begin
|
|
Row.Add(ds.Fields[i].AsString);
|
|
end;
|
|
FResults.Add(Row);
|
|
ds.Next;
|
|
end;
|
|
|
|
ResultGrid.RootNodeCount := FResults.Count;
|
|
ResultGrid.FocusedNode := ResultGrid.GetLast;
|
|
ResultGrid.Selected[ResultGrid.FocusedNode] := True;
|
|
ResultGrid.Repaint;
|
|
end;
|
|
|
|
|
|
procedure TfrmTableTools.ResultGridCompareNodes(Sender: TBaseVirtualTree; Node1, Node2: PVirtualNode;
|
|
Column: TColumnIndex; var Result: Integer);
|
|
begin
|
|
Mainform.vstCompareNodes(Sender, Node1, Node2, Column, Result);
|
|
end;
|
|
|
|
procedure TfrmTableTools.ResultGridGetNodeDataSize(Sender: TBaseVirtualTree; var NodeDataSize: Integer);
|
|
begin
|
|
NodeDataSize := SizeOf(TWideStringList);
|
|
end;
|
|
|
|
|
|
procedure TfrmTableTools.ResultGridInitNode(Sender: TBaseVirtualTree; ParentNode, Node: PVirtualNode;
|
|
var InitialStates: TVirtualNodeInitStates);
|
|
var
|
|
Data: ^TWideStringList;
|
|
begin
|
|
// Bind string list to node
|
|
Data := Sender.GetNodeData(Node);
|
|
Data^ := FResults[Node.Index] as TWideStringList;
|
|
end;
|
|
|
|
|
|
procedure TfrmTableTools.ResultGridPaintText(Sender: TBaseVirtualTree; const TargetCanvas: TCanvas;
|
|
Node: PVirtualNode; Column: TColumnIndex; TextType: TVSTTextType);
|
|
var
|
|
VT: TVirtualStringTree;
|
|
Msg: WideString;
|
|
begin
|
|
// Red text color for errors, purple for nodes
|
|
VT := Sender as TVirtualStringTree;
|
|
if VT.Header.Columns.Count >= 3 then begin
|
|
Msg := LowerCase(VT.Text[Node, 2]);
|
|
if Msg = 'note' then
|
|
TargetCanvas.Font.Color := clPurple
|
|
else if Msg = 'error' then
|
|
TargetCanvas.Font.Color := clRed;
|
|
end;
|
|
end;
|
|
|
|
procedure TfrmTableTools.ResultGridGetText(Sender: TBaseVirtualTree; Node: PVirtualNode; Column: TColumnIndex;
|
|
TextType: TVSTTextType; var CellText: WideString);
|
|
var
|
|
Data: ^TWideStringList;
|
|
begin
|
|
if Column > NoColumn then begin
|
|
Data := Sender.GetNodeData(Node);
|
|
CellText := Data^[Column];
|
|
end;
|
|
end;
|
|
|
|
|
|
procedure TfrmTableTools.ResultGridHeaderClick(Sender: TVTHeader; HitInfo: TVTHeaderHitInfo);
|
|
begin
|
|
// Header column clicked to sort
|
|
Mainform.vstHeaderClick(Sender, HitInfo);
|
|
end;
|
|
|
|
end.
|