vendor/github.com/apigee-labs/transicator/test/loadgen/loadgen.go

package main

import (
	"database/sql"
	"flag"
	"fmt"
	_ "github.com/lib/pq"
	"os"
	"strconv"
	"time"
)

var dbURL, trowss, tscopess string
var db *sql.DB
var trows, tscopes int

func main() {

	flag.Parse()
	argsp := flag.Args()
	dbURL = argsp[0]
	trowss = argsp[1]
	tscopess = argsp[2]
	if dbURL == "" || trowss == "" || tscopess == "" {
		fmt.Println("Args missing. loadgen <path> <rows cnt> <scopes cnt>")
		os.Exit(0)
	}
	trows, _ = strconv.Atoi(trowss)
	tscopes, _ = strconv.Atoi(tscopess)

	/*
	 * Give some time before connecting to the DB. DB just got installed.
	 */

	time.Sleep(5 * time.Second)

	db, err := sql.Open("postgres", dbURL)
	if err != nil {
		fmt.Printf("Set env TEST_PG_URL. ")
		fmt.Print("Unable to open %s. Err [%v]\n", dbURL, err)
		os.Exit(1)
	}
	_, err = db.Exec(tableSQL)
	if err != nil {
		fmt.Printf("Unable to create table. Err: [%v]\n", err)
		os.Exit(2)
	}
	for k := 1; k < 6; k++ {
		tname := strconv.FormatInt(int64(k), 10)
		qstr := "insert into table" + tname + " (column1, column2, column3, column4, column5, _change_selector) values ($1, $2, $3, $4, $5, $6)"
		insert, err := db.Prepare(qstr)
		if err != nil || insert == nil {
			fmt.Printf("Unable to Insert table for scope \n")
			os.Exit(3)
		}
		defer insert.Close()

		for i := 0; i < tscopes; i++ {
			s1 := strconv.FormatInt(int64(i), 10)
			for j := 0; j < trows; j++ {
				s2 := strconv.FormatInt(int64(j), 10)
				_, err = insert.Exec(
					"value_"+
						s1+"_"+s2,
					"value",
					"value",
					"value",
					"value",
					"scope_"+s1)
				if err != nil {
					fmt.Printf("Unable to Insert table for pk")
					fmt.Printf(" [%s]\n", "value_"+s1+"_"+s2)
					os.Exit(4)
				}
			}
			fmt.Printf("Inserted %d rows in table %s for %s\n", trows, tname, "scope_"+s1)
		}
	}
	db.Close()

}

var tableSQL = `
  create table table1(
    column1 varchar primary key,
    column2 varchar,
    column3 varchar,
    column4 varchar,
    column5 varchar,
    _change_selector varchar);
  alter table table1 replica identity full;
  create table table2(
    column1 varchar primary key,
    column2 varchar,
    column3 varchar,
    column4 varchar,
    column5 varchar,
    _change_selector varchar);
  alter table table2 replica identity full;
  create table table3(
    column1 varchar primary key,
    column2 varchar,
    column3 varchar,
    column4 varchar,
    column5 varchar,
    _change_selector varchar);
  alter table table3 replica identity full;
  create table table4(
    column1 varchar primary key,
    column2 varchar,
    column3 varchar,
    column4 varchar,
    column5 varchar,
    _change_selector varchar);
  alter table table4 replica identity full;
  create table table5(
    column1 varchar primary key,
    column2 varchar,
    column3 varchar,
    column4 varchar,
    column5 varchar,
    _change_selector varchar);
  alter table table5 replica identity full;
`