main.go

package main

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

	"github.com/amir-yaghoobi/floodly/generator"
	"github.com/amir-yaghoobi/floodly/reporter"
	"github.com/amir-yaghoobi/floodly/user"
	"github.com/amir-yaghoobi/floodly/user/repository"

	_ "github.com/herenow/go-crate"
)

type runner struct {
	generatorSrv generator.Generator
	reportSrv    reporter.Reporter
}

func (r *runner) setupWorkers(total int, concurrency int, results chan<- reporter.Result) {
	var wg sync.WaitGroup
	wg.Add(concurrency)

	for i := 0; i < concurrency; i++ {
		go r.generatorSrv.Generate(total/concurrency, results, &wg)
	}

	wg.Wait()
	close(results)
}

func (r *runner) Run(total int, concurrency int) {
	results := make(chan reporter.Result, total)

	go r.setupWorkers(total, concurrency, results)

	r.reportSrv.Analyze(results)
}

func main() {
	dbUrl := flag.String("db", "http://localhost:4200/", "crateDB url")
	total := flag.Int("total", 1000, "total number of inserts")
	concurrency := flag.Int("concurrency", 1000, "number of concurrent workers")
	dropTable := flag.Bool("drop-table", false, "drop database table")

	flag.Parse()

	db, err := sql.Open("crate", *dbUrl)
	if err != nil {
		log.Fatal(err)
	}

	userRepository := repository.NewCrateRepository(db)
	err = userRepository.Migrate(*dropTable)
	if err != nil {
		log.Fatal(err)
	}

	genSrv := user.NewUserGenerator(*userRepository)
	repSrv := reporter.NewStandardReporter(os.Stdout)

	srvRunner := &runner{
		generatorSrv: genSrv,
		reportSrv:    repSrv,
	}

	srvRunner.Run(*total, *concurrency)
}