frieza/main.go at master · dperfly/frieza · GitHub

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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
//go:generate  goreleaser release --snapshot --rm-dist

package main

import (
	"Frieza/jmeter"
	"Frieza/master/input"
	master "Frieza/master/server"
	"Frieza/slave"
	"Frieza/version"
	"context"
	"flag"
	"fmt"
	"log"
	"runtime"
)

func main() {

	var host *string
	var isMaster *bool
	var ver *bool
	host = flag.String("R", "", "master host")
	isMaster = flag.Bool("M", false, "default slave server")
	ver = flag.Bool("v", false, "version")
	flag.Parse()
	fmt.Println(version.Logo)

	if runtime.GOOS == "windows" {
		log.Println("Windows operating system is not currently supported")
		return
	}

	if *ver == true {
		fmt.Println("Frieza-version: ", version.Version)
		return
	}
	log.Println("isMaster", *isMaster)
	jmeterRunner := jmeter.New()
	if *isMaster == false {
		if *host == "" {
			log.Println("[WARNING] please input master host; default localhost")
		} else {
			log.Println("master is : ", fmt.Sprintf("%s:8081", *host))
		}
		slave.StartClient(context.Background(), fmt.Sprintf("%s:8081", *host))
	} else {

		// set jmeter env
		if !jmeter.IsOpenServerSSL() {
			err := jmeter.SetServerSSL()
			if err != nil || !jmeter.IsOpenServerSSL() {
				log.Println(err)
				return
			}
		}
		// set java Server use host
		//if !jmeter.IsSetServerHostname() {
		//	err := jmeter.SetServerHostname()
		//	if err != nil || !jmeter.IsSetServerHostname() {
		//		log.Println(err)
		//		return
		//	}
		//}

		// set java
		if !jmeter.IsFindJavaEnv() {
			err := jmeter.InstallJDK()
			if err != nil || !jmeter.IsFindJavaEnv() {
				log.Println("install jdk failed")
				return
			}
		}

		go master.StartServer(":8081")
		defer jmeterRunner.StopJmeterThread()
		input.CmdInput(jmeterRunner)
	}

}