cmd/anelephantinachinashop/main.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

package main

import (
	"flag"
	"fmt"
	"log"

	"codeberg.org/snonux/rampage/internal"
	"codeberg.org/snonux/rampage/internal/config"
)

func main() {
	var conf config.Config
	printVersion := flag.Bool("version", false, "Print version")
	stressCPU := flag.Bool("cpu", false, "Stress CPU")
	stressMemory := flag.Bool("memory", false, "Stress memory")
	stressNetwork := flag.Bool("network", false, "Stress network")
	flag.Parse()

	if *printVersion {
		fmt.Println(internal.Version)
		return
	}

	if *stressCPU {
		conf.CPU = config.CPUStressConfig{}
	}
	if *stressMemory {
		conf.Memory = config.MemoryStressConfig{}
		log.Fatal("Mode not yet implemented")
	}
	// TODO: Also add parallel stressers
	if *stressNetwork {
		conf.Network = config.NetworkStressConfig{}
		log.Fatal("Mode not yet implemented")
	}

	if err := internal.Run(conf); err != nil {
		log.Fatal(err)
	}
}