$ trino --execute 'select * from ndaq_rtat10 limit 5'
$ trino --execute 'select * from zacks_hdm limit 5'

"""
DataLink Trino Connection - Getting Started Example

This file demonstrates how to get started with DataLink's Trino native JDBC connection.
Use this as a reference to implement a solution based on your specific needs.

DataLink supports Trino native JDBC connections, so customers can use their own implementation
using the Trino JDBC driver: https://trino.io/docs/current/client/jdbc.html

Important:
- Review all TODO comments in this file to customize the query and processing logic for your use case.

Troubleshooting:
- If you are having issues with any query, please enable debug logging (set logging level to DEBUG).
- Debug mode will provide you with the query ID.
- Please provide the query ID to the DataLink Client Success team for assistance.
"""

import logging
from trino.dbapi import connect

def _extract_query_id(cur):
    q = getattr(cur, "query_id", None) or getattr(cur, "_query_id", None)
    try:
        if not q and hasattr(cur, "_operation") and isinstance(cur._operation, dict):
            q = cur._operation.get("queryId") or cur._operation.get("query_id")
    except Exception:
        pass
    return q

conn = connect(
    host="data.nasdaq.com",
    port=443,
    user="TODO: your api_key here",
    catalog="main",
    schema="huron",
    http_scheme="https"
)

# Configure logging
logging.basicConfig(
    level=logging.DEBUG,
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
)
logger = logging.getLogger(__name__)

#TODO: replace with your query
sql = ("SELECT 1 LIMIT 10")

try:
    cur = conn.cursor()
    cur.execute(sql)
    # Immediately try to read and print the query id after execute succeeds
    if(logger.isEnabledFor(logging.DEBUG)):
        query_id = _extract_query_id(cur)
        logger.debug(f"--------- Trino query id -------: {query_id}")
    
    # use fetchall()/fetchone()/fetchmany() as needed https://github.com/trinodb/trino-python-client#db-api
    rows = cur.fetchmany(size=1000)
    for row in rows:
        logger.info(f"Row: {row}")
    # TODO: process rows as needed!

except Exception as exc:
    # If execute raised, try to extract/print query id anyway, then re-raise
    query_id = _extract_query_id(cur)
    logger.error(f"Trino query id (after error): {query_id}")
    raise
finally:
    # Safety: if for any reason we haven't printed the id yet, attempt once more
    cur.close()
    logger.info("Closing Trino connection")

$ dotnet run --property DefineConstants=INCLUDE_MAIN

using NReco.PrestoAdo;
using static DataLinkSQL;

class MyApp
{
    static void Main()
    {
        using (PrestoConnection connection = CreateConnection())
        {
            string sql = "...";
            DbDataReader reader = RunSQL(connection, sql);
            DisplayReader(reader);
        }
    }
}

$ dotnet run