server: implement spill disk for cursorFetch result

[YangKeao]

This PR is same with #44730. Some bugs of Github makes me fail to update #44730, so I have to create a new one 😮‍💨 .

What problem does this PR solve?

Issue Number: close #43233

What is changed and how it works?

1. Use the rowContainer to store the rows after execution.
2. Reset the memory/disk tracker when the statement is closed.

TODO:

• Add tests
• Consider the behavior of Reset.
• Consider carefully about the server status. 56 cases of mysql_client_test failed #44672 seems to have shown some bugs about server status.

Check List

Tests

• Unit test
• Integration test
• Manual test (add detailed scripts or steps below)
• No code

I used the following codes to do some basic tests. The preloaded data is 5M rows sysbench ./oltp_insert.lua :

public static void tidbBigSelectTest() throws SQLException, InterruptedException {
    // the following are test codes for MySQL
    ConnectionImpl conn = (ConnectionImpl) DriverManager.getConnection("jdbc:mysql://127.0.0.1:4000/test?useCursorFetch=true&defaultFetchSize=1000", "root", "");
    conn.prepareStatement("set @@tidb_mem_quota_query = 8 << 20\n").execute();
    conn.prepareStatement("begin").execute();

    String sql = "SELECT * FROM test.sbtest1";
    PreparedStatement stmt = conn.prepareStatement(sql, ResultSet.TYPE_FORWARD_ONLY, ResultSet.CONCUR_READ_ONLY);
    stmt.setFetchSize(1000);
    Instant beforeExecuteTime = Clock.systemUTC().instant();
    ResultSet rs = stmt.executeQuery();
    Instant afterExecuteTime = Clock.systemUTC().instant();
    System.out.println("It takes " + (Duration.between(beforeExecuteTime, afterExecuteTime).toString()) + " to EXECUTE QUERY");

    ResultSetMetaData metaData = rs.getMetaData();
    int columnCount = metaData.getColumnCount();
    int batchSize = stmt.getFetchSize();
    int rowCount = 0;
    int totalCount = 0;
    while (rs.next()) {
        rowCount++;
        totalCount++;
        Object[] columns = new Object[columnCount];
        for (int i = 1; i <= columnCount; i++) {
            columns[i - 1] = rs.getObject(i);
        }
        Row row = new Row(columns);
        if (rowCount == batchSize) {
            System.out.println("Fetched " + batchSize + " rows");
            System.out.println("allocated memory is " + (Runtime.getRuntime().totalMemory()-Runtime.getRuntime().freeMemory()));
            System.out.println("TotalCount " + totalCount);
            rowCount = 0;
        }
        if (totalCount == 4999999) {
            System.out.println("here");
        }
    }
    if (rowCount > 0) {
        System.out.println("allocated memory is " + (Runtime.getRuntime().totalMemory()-Runtime.getRuntime().freeMemory()));
        System.out.println("Fetched " + rowCount + " rows");
        System.out.println("TotalCount " + totalCount);
    }

    conn.prepareStatement("commit").execute();
}

Release note

Automatically spill the memory exceeding `tidb_mem_quota_query` into the disk

[YangKeao]

/retest

[ti-chi-bot]

[APPROVALNOTIFIER] This PR is APPROVED

This pull-request has been approved by: hawkingrei, xhebox

The full list of commands accepted by this bot can be found here.

The pull request process is described here

Needs approval from an approver in each of these files:

• OWNERS [hawkingrei,xhebox]

Approvers can indicate their approval by writing /approve in a comment
Approvers can cancel approval by writing /approve cancel in a comment

[ti-chi-bot]

[LGTM Timeline notifier]

Timeline:

• 2023-07-05 05:56:56.336695467 +0000 UTC m=+183448.270328885: ☑️ agreed by hawkingrei.
• 2023-07-06 03:16:39.762632706 +0000 UTC m=+260231.696266127: ☑️ agreed by xhebox.

[ti-chi-bot]

In response to a cherrypick label: new pull request created to branch release-7.1: #45259.

[bb7133]

/run-cherry-picker

[ti-chi-bot]

In response to a cherrypick label: new pull request created to branch release-6.5: #45602.