package main

import (
	"context"
	"database/sql"
	"flag"
	"log"
	"os"
	"time"

	"pg/src/with"
)

func main() {
	if err := with.Context(run); err != nil {
		panic(err)
	}
}

func run(ctx context.Context) error {
	fs := flag.NewFlagSet(os.Args[0], flag.ContinueOnError)
	c := fs.String("c", "sqlite://", "conn string")
	if err := fs.Parse(os.Args[1:]); err != nil {
		panic(err)
	}

	return with.SQL(ctx, *c, func(db *sql.DB) error {
		go with.Every(ctx, 5*time.Second, func() {
			row := db.QueryRowContext(ctx, `SELECT 1`)
			var n int
			if err := row.Err(); err != nil {
				log.Println("query err:", err)
			} else if err := row.Scan(&n); err != nil {
				log.Println("scan err:", err)
			} else {
				log.Println("ping...")
			}
		})

		<-ctx.Done()
		return nil
	})
}