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

	_ "github.com/go-sql-driver/mysql"

	. "github.com/go-jet/jet/v2/mysql"
)

var (
	endpoint = "127.0.0.1:4002"
	username = ""
	password = ""
	database = "public"
)

func main() {
	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
	defer cancel()

	ts := TimestampColumn("ts")
	itemID := IntegerColumn("item_id")
	sampleScore := FloatColumn("sample_score")
	table := NewTable("", "limit_param_repro", "", ts, itemID, sampleScore)

	stmt := SELECT(
		itemID.AS("out_item_id"),
		sampleScore.AS("out_sample_score"),
	).FROM(table).
		LIMIT(1)

	query, args := stmt.Sql()
	fmt.Printf("SQL:\n%s\n", query)
	fmt.Printf("args (%d): %#v\n\n", len(args), args)

	db, err := sql.Open("mysql", dsn())
	if err != nil {
		log.Fatalf("open database: %v", err)
	}
	defer db.Close()

	rows, err := db.QueryContext(ctx, query, args...)
	if err != nil {
		log.Fatalf("run query: %v", err)
	}
	defer rows.Close()

	for rows.Next() {
		var (
			gotItemID      int64
			gotSampleScore float64
		)

		if err := rows.Scan(&gotItemID, &gotSampleScore); err != nil {
			log.Fatalf("scan row: %v", err)
		}

		fmt.Printf("row: item_id=%d sample_score=%f\n", gotItemID, gotSampleScore)
	}

	if err := rows.Err(); err != nil {
		log.Fatalf("read rows: %v", err)
	}
}

func dsn() string {
	return fmt.Sprintf("%s:%s@tcp(%s)/%s?parseTime=true", username, password, endpoint, database)
}