67 lines
1.2 KiB
Go
67 lines
1.2 KiB
Go
// Package mockserver is the core of this program and it will have the API
|
|
// to record binary data from the UDP server and then be able to mock it
|
|
package mockserver
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"time"
|
|
)
|
|
|
|
// Replay replays a given file <fp> in a UDP server <addr>:<port>
|
|
func Replay(ctx context.Context, address string, port int, loop bool, fp string) error {
|
|
fileInfo, err := os.Stat(fp)
|
|
if err != nil {
|
|
return fmt.Errorf("error stating file: %v", err)
|
|
}
|
|
|
|
bin, err := os.Open(fp)
|
|
if err != nil {
|
|
return fmt.Errorf("error opening file: %v", err)
|
|
}
|
|
|
|
reader := NewGobReader(bin)
|
|
udp, err := NewUDPTransport(address, port)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
ticker := time.NewTicker(time.Second / 60)
|
|
defer ticker.Stop()
|
|
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
return ctx.Err()
|
|
|
|
case <-ticker.C:
|
|
data, err := reader.Next()
|
|
if err == io.EOF {
|
|
if !loop {
|
|
return udp.Close()
|
|
}
|
|
|
|
err = reader.Reset()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
continue
|
|
}
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = udp.Send(data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
percent := int(float64(reader.TotalRead) / float64(fileInfo.Size()) * 100)
|
|
fmt.Printf("\rReplayed: %d%%", percent)
|
|
}
|
|
}
|
|
}
|