package main

import (
	"context"
	"database/sql"
	"fmt"
	"log"
	"testing"
	"time"

	_ "github.com/apache/arrow-adbc/go/adbc/sqldriver/flightsql"
)

var dsnTemplate = "uri=grpc://%s;username=%s;password=%s"

const (
	FLIGHT_SQL_USER = "gizmosql_username"
	FLIGHT_SQL_PASS = "password123"
	FLIGHT_SQL_HOST = "localhost:31337"
)

// This example uses Go's native SQL driver
func TestFlightSQL(t *testing.T) {
	dsn := fmt.Sprintf(dsnTemplate,
		FLIGHT_SQL_HOST,
		FLIGHT_SQL_USER,
		FLIGHT_SQL_PASS,
	)

	db, err := sql.Open("flightsql", dsn)
	if err != nil {
		log.Fatalf("opening a connection failed: %v", err)
	}

	defer db.Close()

	timestamp := time.Now()

	// Table schema
	// CREATE TABLE IF NOT EXISTS flight_test.events (id INTEGER, event_time TIMESTAMP)
	result, err := db.ExecContext(context.Background(), `INSERT INTO flight_test.events (id, event_time) VALUES (1, $1)`, timestamp)
	if err != nil {
		log.Fatal(err)
	}

	log.Print(result)
}