fathom/db/seed.go

package db

import (
  "github.com/dannyvankooten/ana/models"
  "log"
  "time"
  "math/rand"
  "fmt"
  "github.com/Pallinder/go-randomdata"
)

var browserNames = []string {
  "Chrome",
  "Chrome",
  "Firefox",
  "Safari",
  "Safari",
  "Internet Explorer",
}

var months = []time.Month {
  time.January,
  time.February,
  time.March,
  time.April,
  time.May,
  time.June,
  time.July,
  time.August,
  time.September,
  time.October,
  time.November,
  time.December,
}

var paths = []string {
  "/",
  "/", // we need this to weigh more.
  "/contact",
  "/about",
  "/checkout",
}

var browserLanguages = []string {
  "en-US",
  "en-US",
  "nl-NL",
  "fr-FR",
  "de-DE",
  "es-ES",
}

var screenResolutions = []string {
  "2560x1440",
  "1920x1080",
  "1920x1080",
  "360x640",
}

func Seed(n int) {

  // prepare statement for inserting data
  stmt, err := Conn.Prepare(`INSERT INTO visits(
    browser_language,
    browser_name,
    browser_version,
    country,
    device_os,
    ip_address,
    path,
    referrer_url,
    screen_resolution,
    timestamp
    ) VALUES( ?, ?, ?, ?, ?, ?, ?, ?, ?, ? )`)
  if err != nil {
      log.Fatal(err)
  }
  defer stmt.Close()

  // insert X random hits
  for i := 0; i < n; i++ {

    // print a dot as progress indicator
    fmt.Print(".")

    // generate random timestamp
    date := randomDateBeforeNow();
    timestamp := fmt.Sprintf("%s %d:%d:%d", date.Format("2006-01-02"), randInt(10, 24), randInt(10, 60), randInt(10, 60))

    visit := models.Visit{
      IpAddress: randomdata.IpV4Address(),
      DeviceOS: "Linux x86_64",
      BrowserName: randSliceElement(browserNames),
      BrowserVersion: "54.0.2840.100",
      BrowserLanguage: randSliceElement(browserLanguages),
      ScreenResolution: randSliceElement(screenResolutions),
      Country: randomdata.Country(randomdata.TwoCharCountry),
      ReferrerUrl: "",
      Timestamp: timestamp,
    }

    // insert between 1-4 pageviews for this visitor
    for j := 0; j < randInt(1, 4); j++ {
      visit.Path = randSliceElement(paths)

      _, err = stmt.Exec(
        visit.BrowserLanguage,
        visit.BrowserName,
        visit.BrowserVersion,
        visit.Country,
        visit.DeviceOS,
        visit.IpAddress,
        visit.Path,
        visit.ReferrerUrl,
        visit.ScreenResolution,
        visit.Timestamp,
      )
      if err != nil {
        log.Fatal(err)
      }
    }

  }
}

func randomDate() time.Time {
  now := time.Now()
  month := months[randInt(0, len(months))]
  t := time.Date(now.Year(), month, randInt(1,31), randInt(0,23), randInt(0,59), randInt(0,59), 0, time.UTC)
  return t
}

func randomDateBeforeNow() time.Time {
  now := time.Now()
  date := randomDate()
  for( date.After(now) ) {
    date = randomDate()
  }

  return date
}

func randSliceElement(slice []string) string {
  return slice[randInt(0, len(slice))]
}

func randInt(min int, max int) int {
    return min + rand.Intn(max-min)
}