From 9b28e452954498a49501740b04e869fffb6649a9 Mon Sep 17 00:00:00 2001
From: Henrik Hautakoski <henrik@eossweden.org>
Date: Tue, 7 Feb 2023 12:32:43 +0100
Subject: [PATCH] cmd/antelope-v1-mock-server/main.go: adding `-d` flag for
 random delays.

---
 cmd/antelope-v1-mock-server/main.go | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/cmd/antelope-v1-mock-server/main.go b/cmd/antelope-v1-mock-server/main.go
index 5ec578a..3ecccd5 100644
--- a/cmd/antelope-v1-mock-server/main.go
+++ b/cmd/antelope-v1-mock-server/main.go
@@ -3,6 +3,7 @@ package main
 import (
 	"flag"
 	"fmt"
+	"math/rand"
 	"net/http"
 	"os"
 	"time"
@@ -13,6 +14,7 @@ import (
 var (
 	listen_host = flag.String("h,host", "localhost", "Host to listen on.")
 	listen_port = flag.Int("p", 3333, "Port to listen to.")
+	delay       = flag.Int("d", 0, "Delays responses randomly between 0 and int seconds.")
 )
 
 func getInfo(w http.ResponseWriter, r *http.Request) {
@@ -40,6 +42,11 @@ func getInfo(w http.ResponseWriter, r *http.Request) {
 		ForkDBHeadBlockNum:        100,
 	}
 
+	if *delay > 0 {
+		sleep_for := rand.Intn(*delay)
+		time.Sleep(time.Second * time.Duration(sleep_for))
+	}
+
 	payload, err := leapapi.Json().Marshal(&info)
 	if err != nil {
 		fmt.Fprintln(os.Stderr, err)
@@ -54,6 +61,8 @@ func getInfo(w http.ResponseWriter, r *http.Request) {
 }
 
 func main() {
+	rand.Seed(time.Now().Unix())
+
 	flag.Parse()
 
 	http.HandleFunc("/v1/chain/get_info", getInfo)