func newConnection()

in gitaly-bench.go [31:58]


func newConnection() (*grpc.ClientConn, error) {
	connOpts := make([]grpc.DialOption, len(client.DefaultDialOpts))
	copy(connOpts, client.DefaultDialOpts)

	token := ""

	if *authFileFlag != "" {
		contents, err := ioutil.ReadFile(*authFileFlag)
		if err != nil {
			log.Fatalf("unable to read auth-file: %v", err)
		}
		token = string(contents)
		token = strings.TrimRight(token, "\n\r")
	} else if *authFlag != "" {
		token = *authFlag
	}

	if token != "" {
		connOpts = append(connOpts, grpc.WithPerRPCCredentials(gitalyauth.RPCCredentialsV2(token)))
	}

	conn, err := client.Dial(*hostFlag, connOpts)
	if err != nil {
		return nil, err
	}

	return conn, nil
}