HeidiSQL/source/runsqlfile.pas

unit runsqlfile;

interface

uses
  Windows, SysUtils, Classes, Controls, Forms, Dialogs, ComCtrls, StdCtrls;

type
  TRunSQLFileForm = class(TForm)
    btnClose: TButton;
    lblFilenameValue: TLabel;
    lblFilenameName: TLabel;
    lblQueryName: TLabel;
    lblPositionName: TLabel;
    lblPositionValue: TLabel;
    prbarRun: TProgressBar;
    memoQueryValue: TMemo;
    lblQueryCountName: TLabel;
    lblQueryCountValue: TLabel;
    lblTimeName: TLabel;
    lblTimeValue: TLabel;
    lblAffectedRowsName: TLabel;
    lblAffectedRowsValue: TLabel;
    procedure FormActivate(Sender: TObject);
    procedure FormCreate(Sender: TObject);
  private
    { Private declarations }
    SQLFileName : String;
    Running : Boolean;
  public
    { Public declarations }
  end;

  function RunSQLFileWindow(AOwner: TComponent; SQLFileName: String): Boolean;

implementation

uses
  helpers,
  main;

{$R *.dfm}


{**
  Create form on demand
  @param TComponent Owner of form (should be calling form)
  @return Boolean Form closed using modalresult mrOK
}
function RunSQLFileWindow(AOwner: TComponent; SQLFileName: String): Boolean;
var
  f : TRunSQLFileForm;
begin
  f := TRunSQLFileForm.Create(AOwner);
  f.SQLFileName := SQLFileName;
  Result := (f.ShowModal=mrOK);
  FreeAndNil(f);
end;


{**
  Startup - run the sql file
}
procedure TRunSQLFileForm.FormActivate(Sender: TObject);
var
  Stream              : TFileStream;
  FileCharset         : TFileCharset;
  lines               : String;
  filesize,
  querycount,
  rowsaffected        : Int64;
  starttime           : Cardinal;
  SQL                 : TStringList;
  i                   : Integer;
  lines_remaining     : String;
begin
  if Running then
    abort;
  Running := True;

  // Initialize various variables
  lblFilenameValue.Caption := mince( SQLFileName, 35 );
  lblTimeValue.Caption := FormatTimeNumber( 0 );
  memoQueryValue.Lines.Clear;
  lines := '';
  querycount := 0;
  rowsaffected := 0;
  lines_remaining := '';
  starttime := GetTickCount;

  Screen.Cursor := crHourGlass;

  try
    // Start file operations
    filesize := _GetFileSize( SQLFileName );

    OpenTextfile(SQLFileName, Stream, FileCharset );
    lblPositionValue.Caption := FormatNumber( Stream.Position ) + ' / ' + FormatNumber( filesize );
    Repaint;

    while Stream.Position < Stream.Size do
    begin
      // Read lines from SQL file until buffer reaches a limit of some MB
      // This strategy performs vastly better than looping through each line
      lines := ReadTextfileChunk(Stream, FileCharset, 5*SIZE_MB);

      // Display position in file
      lblPositionValue.Caption := FormatByteNumber( Stream.Position ) + ' / ' + FormatByteNumber( filesize );

      // Time
      lblTimeValue.Caption := FormatTimeNumber( (GetTickCount - starttime) DIV 1000 );

      // Step progressbar's position
      prbarRun.Position := Trunc( prbarRun.Max / filesize * Stream.Position );

      Repaint;

      // Split buffer into single queries
      SQL := parseSQL( lines_remaining + lines );
      lines := '';
      lines_remaining := '';

      // Execute detected queries
      for i := 0 to SQL.Count - 1 do
      begin
        // Last line has to be processed in next loop if end of file is not reached
        if (i = SQL.Count-1) and (Stream.Position < Stream.Size) then
        begin
          lines_remaining := SQL[i];
          break;
        end;

        // Count queries
        querycount := querycount + 1;
        lblQueryCountValue.Caption := FormatNumber( querycount );

        // Display part of query
        memoQueryValue.Text := sstr( SQL[i], 100 );

        // Time
        lblTimeValue.Caption := FormatTimeNumber( (GetTickCount - starttime) DIV 1000 );

        // Execute single query and display affected rows
        Mainform.Connection.Query(SQL[i]);
        rowsaffected := rowsaffected + Mainform.Connection.RowsAffected;
        lblAffectedRowsValue.Caption := FormatNumber( rowsaffected );

        Repaint;
      end;
      SQL.Free;

    end;
    Stream.Free;
  except
    on E: Exception do
    begin
      MessageDLG( 'Error while reading file ' + SQLFileName + ':' + CRLF + CRLF + E.Message, mtError, [mbOK], 0);
      Mainform.AddOrRemoveFromQueryLoadHistory( SQLFileName, false );
      Mainform.FillPopupQueryLoad;
    end;
  end;
  Screen.Cursor := crDefault;
  btnClose.Enabled := True;
end;


procedure TRunSQLFileForm.FormCreate(Sender: TObject);
begin
  InheritFont(Font);
end;

end.