Method TestRetryBatchCopyWithHooks

go/logic/migrator_test.go:521–639

()

}

func (suite *MigratorTestSuite) TestRetryBatchCopyWithHooks() {
	ctx := context.Background()

	_, err := suite.db.ExecContext(ctx, "CREATE TABLE test.test_retry_batch (id INT PRIMARY KEY AUTO_INCREMENT, name TEXT)")
	suite.Require().NoError(err)

	const initStride = 1000
	const totalBatches = 3
	for i := 0; i < totalBatches; i++ {
		dataSize := 50 * i
		for j := 0; j < initStride; j++ {
			_, err = suite.db.ExecContext(ctx, fmt.Sprintf("INSERT INTO test.test_retry_batch (name) VALUES ('%s')", strings.Repeat("a", dataSize)))
			suite.Require().NoError(err)
		}
	}

	_, err = suite.db.ExecContext(ctx, fmt.Sprintf("SET GLOBAL max_binlog_cache_size = %d", 1024*8))
	suite.Require().NoError(err)
	defer func() {
		_, err = suite.db.ExecContext(ctx, fmt.Sprintf("SET GLOBAL max_binlog_cache_size = %d", 1024*1024*1024))
		suite.Require().NoError(err)
	}()

	tmpDir, err := os.MkdirTemp("", "gh-ost-hooks")
	suite.Require().NoError(err)
	defer os.RemoveAll(tmpDir)

	hookScript := filepath.Join(tmpDir, "gh-ost-on-batch-copy-retry")
	hookContent := `#!/bin/bash
# Mock hook that reduces chunk size on binlog cache error
ERROR_MSG="$GH_OST_LAST_BATCH_COPY_ERROR"
SOCKET_PATH="/tmp/gh-ost.sock"

if ! [[ "$ERROR_MSG" =~ "max_binlog_cache_size" ]]; then
    echo "Nothing to do for error: $ERROR_MSG"
    exit 0
fi

CHUNK_SIZE=$(echo "chunk-size=?" | nc -U $SOCKET_PATH | tr -d '\n')

MIN_CHUNK_SIZE=10
NEW_CHUNK_SIZE=$(( CHUNK_SIZE * 8 / 10 ))
if [ $NEW_CHUNK_SIZE -lt $MIN_CHUNK_SIZE ]; then
    NEW_CHUNK_SIZE=$MIN_CHUNK_SIZE
fi

if [ $CHUNK_SIZE -eq $NEW_CHUNK_SIZE ]; then
    echo "Chunk size unchanged: $CHUNK_SIZE"
    exit 0
fi

echo "[gh-ost-on-batch-copy-retry]: Changing chunk size from $CHUNK_SIZE to $NEW_CHUNK_SIZE"
echo "chunk-size=$NEW_CHUNK_SIZE" | nc -U $SOCKET_PATH
echo "[gh-ost-on-batch-copy-retry]: Done, exiting..."
`
	err = os.WriteFile(hookScript, []byte(hookContent), 0755)
	suite.Require().NoError(err)