oz-lott demo

main.go

package main

import (
	"flag"
	"fmt"
	"math/rand"
	"sort"
	"time"
)

// const
const (
	NumberCount    = 7
	MaximumResults = 50
)

var (
	winners = make(map[[NumberCount]int]int)
)

func main() {
	timesInput := flag.Int("t", 100000, "an int")
	flag.Parse()
	times := *timesInput

	now := time.Now()

	startTime := now.Unix()

	fmt.Printf("Started at %v to run lottery for %v times \n", now, times)
	for i := 0; i < times; i++ {
		winner := run()
		if currentTimes, ok := winners[winner]; ok {
			winners[winner] = currentTimes + 1
		} else {
			winners[winner] = 1
		}
	}

	fmt.Printf("Calculation finished at : %v \n", time.Now())
	// only print out top numbers
	type kv struct {
		Key   [NumberCount]int
		Value int
	}

	var ss []kv
	for k, v := range winners {
		// to reduce the time consuming, if the number only exists once then skip it
		if v > 1 {
			ss = append(ss, kv{k, v})
		}

	}

	sort.Slice(ss, func(i, j int) bool {
		return ss[i].Value > ss[j].Value
	})

	fmt.Printf("Sorting finished at : %v \n", time.Now())

	index := 0
	for _, kv := range ss {
		// only pick the top numbers
		if index > MaximumResults {
			break
		}
		fmt.Printf("%v, %d \n", kv.Key, kv.Value)
		index++
	}
	fmt.Printf("The process takes about: %v seconds", time.Now().Unix()-startTime)
}

func run() [7]int {
	var winNumbers []int
	// generate the randomised numbers
	list := rand.Perm(45)
	// because perm generate [0,45) but we need [1,45]
	for index, item := range list {
		list[index] = item + 1
	}

	rand.Seed(time.Now().UnixNano())

	indexes := make([]int, 0)

	for i := 0; i < NumberCount; i++ {
		pickedIndex := pickIndex(indexes, len(list))
		indexes = append(indexes, pickedIndex)

		//pick random number
		chosen := list[pickedIndex]

		winNumbers = append(winNumbers, chosen)
	}

	sort.Ints(winNumbers)
	var arr [NumberCount]int
	copy(arr[:], winNumbers)

	return arr

}

func pickIndex(indexes []int, length int) int {
	chosen := rand.Intn(length - 1)
	// make sure the index only exist once
	for _, value := range indexes {
		if value == chosen {
			return pickIndex(indexes, length)
		}
	}

	return chosen
}