You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
1.8 KiB
Go
66 lines
1.8 KiB
Go
package iperf
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/google/uuid"
|
|
"io"
|
|
"time"
|
|
)
|
|
|
|
func NewServer() *Server {
|
|
defaultPort := 5201
|
|
defaultInterval := 1
|
|
s := &Server{
|
|
Port: &defaultPort,
|
|
Interval: &defaultInterval,
|
|
}
|
|
s.Id = uuid.New().String()
|
|
return s
|
|
}
|
|
|
|
type Server struct {
|
|
Id string `json:"id" yaml:"id" xml:"id"`
|
|
OneOff *bool `json:"one_off" yaml:"one_off" xml:"one_off"`
|
|
ExitCode *int `json:"exit_code" yaml:"exit_code" xml:"exit_code"`
|
|
Port *int `json:"port" yaml:"port" xml:"port"`
|
|
Format *rune `json:"format" yaml:"format" xml:"format"`
|
|
Interval *int `json:"interval" yaml:"interval" xml:"interval"`
|
|
JSON *bool `json:"json" yaml:"json" xml:"json"`
|
|
LogFile *string `json:"log_file" yaml:"log_file" xml:"log_file"`
|
|
Running bool `json:"running" yaml:"running" xml:"running"`
|
|
outputStream io.ReadCloser `json:"output_stream" yaml:"output_stream" xml:"output_stream"`
|
|
errorStream io.ReadCloser `json:"error_stream" yaml:"error_stream" xml:"error_stream"`
|
|
cancel context.CancelFunc `json:"cancel" yaml:"cancel" xml:"cancel"`
|
|
}
|
|
|
|
func (s *Server) Start() (err error) {
|
|
var exit chan int
|
|
s.outputStream, s.errorStream, exit, s.cancel, err = ExecuteAsyncWithCancel(fmt.Sprintf("%s -s -J", binaryLocation))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
s.Running = true
|
|
go func() {
|
|
ds := DebugScanner{Silent: true}
|
|
ds.Scan(s.outputStream)
|
|
}()
|
|
go func() {
|
|
ds := DebugScanner{Silent: true}
|
|
ds.Scan(s.errorStream)
|
|
}()
|
|
go func() {
|
|
exitCode := <-exit
|
|
s.ExitCode = &exitCode
|
|
s.Running = false
|
|
}()
|
|
return nil
|
|
}
|
|
|
|
func (s *Server) Stop() {
|
|
if s.Running && s.cancel != nil {
|
|
s.cancel()
|
|
time.Sleep(100 * time.Millisecond)
|
|
}
|
|
}
|