databricks-cli/dist/commands/print-row.js

A CLI tool for quick Databricks tasks

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.printRow = printRow;
const connection_1 = require("../utils/connection");
const interactive_1 = require("../utils/interactive");
async function printRow(options) {
    try {
        let database = options.database;
        if (!database) {
            database = await (0, interactive_1.selectDatabase)();
        }
        let table = options.table;
        if (!table) {
            table = await (0, interactive_1.selectTable)(database);
        }
        const count = options.count || 10;
        const session = await (0, connection_1.getSharedSession)();
        const query = `SELECT * FROM ${database}.${table} LIMIT ${count}`;
        const queryOperation = await session.executeStatement(query);
        const result = await queryOperation.fetchAll();
        if (result.length === 0) {
            console.log("No rows found in the table.");
            return;
        }
        console.table(result);
    }
    catch (error) {
        console.error("Error printing rows:", error instanceof Error ? error.message : error);
        process.exit(1);
    }
    finally {
        await (0, connection_1.closeConnection)();
    }
}