@variablesoftware/mock-d1/dist/engine/statementHandlers/handleTruncateTable.js

🎛️🗂️🧠 Mock D1 Database implementation for testing Cloudflare Workers

import { extractTableName } from '../tableUtils/tableNameUtils.js';
import { log } from "@variablesoftware/logface";
import { findTableKey } from '../tableUtils/tableLookup.js';
import { d1Error } from '../errors.js';
import { validateSqlOrThrow } from '../sqlValidation.js';
/**
 * Handles TRUNCATE TABLE <table> statements.
 */
export function handleTruncateTable(sql, db) {
    validateSqlOrThrow(sql);
    log.debug("called", { sql });
    let tableName;
    try {
        tableName = extractTableName(sql, 'TRUNCATE');
    }
    catch {
        throw new Error("Malformed TRUNCATE TABLE statement.");
    }
    const tableKey = findTableKey(db, tableName);
    log.debug("tableKey lookup", { tableName, tableKey });
    if (!tableKey) {
        log.error("Table does not exist in the database", { tableName });
        throw d1Error('TABLE_NOT_FOUND', tableName);
    }
    // Always remove all rows, including schema row, on TRUNCATE
    const tableData = db.get(tableKey);
    if (tableData) {
        tableData.columns = [];
        tableData.rows = [];
    }
    log.info("Table truncated", { tableKey });
    return {
        success: true,
        results: [],
        meta: {
            duration: 0,
            size_after: 0,
            rows_read: 0, rows_written: 0,
            last_row_id: 0, changed_db: true, changes: 0,
        },
    };
}