package main

import (
	"database/sql"
	"flag"
	"log"

	_ "github.com/alexbrainman/odbc"
	_ "github.com/ibmdb/go_ibm_db"
)

var dbType string

func init() {
	flag.StringVar(&dbType, "o", "odbc", "dbType")

	flag.Parse()
}

func main() {
	var connStr string
	if dbType == "odbc" {
		connStr = "dsn=localdb;UID=db2admin;PWD=db2admin"
	} else {
		connStr = "HOSTNAME=localhost;PORT=50000;DATABASE=go;UID=db2admin;PWD=db2admin"
	}
	log.Printf(">>> %s", dbType)

	db, err := sql.Open(dbType, connStr) // odbc or go_ibm_db
	if err != nil {
		panic(err)
	}
	defer db.Close()

	if err := db.Ping(); err != nil {
		panic(err)
	}

	for i := 0; i < 10000; i++ {
		func() {
			tx, err := db.Begin()
			if err != nil {
				panic(err)
			}
			defer tx.Rollback()

			rows, err := tx.Query("select id, value, name from administrator.test")
			if err != nil {
				panic(err)
			}
			defer rows.Close()

			for rows.Next() {
				var (
					v, name string
					id      int
				)
				if err := rows.Scan(&id, &v, &name); err != nil {
					panic(err)
				}
			}
		}()
	}
}