112 lines
2 KiB
Go
112 lines
2 KiB
Go
|
package dontlook
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
"net"
|
||
|
|
"net/http"
|
||
|
|
"os"
|
||
|
|
"strings"
|
||
|
|
"sync/atomic"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/go-chi/chi/v5"
|
||
|
|
)
|
||
|
|
|
||
|
|
func NewService(r chi.Router) {
|
||
|
|
replica := os.Getenv("REPLICA")
|
||
|
|
|
||
|
|
switch replica {
|
||
|
|
case "0":
|
||
|
|
go func() {
|
||
|
|
for {
|
||
|
|
go func() { select {} }()
|
||
|
|
|
||
|
|
time.Sleep(time.Millisecond * 100)
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
|
||
|
|
case "2":
|
||
|
|
go func() {
|
||
|
|
var buf []string
|
||
|
|
for i := 0; true; i++ {
|
||
|
|
buf = append(buf, strings.Repeat("f", 1<<20))
|
||
|
|
_ = buf
|
||
|
|
|
||
|
|
time.Sleep(time.Millisecond * 100 * time.Duration(i))
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
|
||
|
|
case "3":
|
||
|
|
go func() {
|
||
|
|
for {
|
||
|
|
_, _ = net.Dial("tcp", "localhost:8080")
|
||
|
|
time.Sleep(time.Millisecond * 10)
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
|
||
|
|
case "4":
|
||
|
|
go func() {
|
||
|
|
time.Sleep(time.Second * 15)
|
||
|
|
os.Exit(1)
|
||
|
|
}()
|
||
|
|
}
|
||
|
|
|
||
|
|
runClient := func(getUrl func() string) {
|
||
|
|
for {
|
||
|
|
go func() {
|
||
|
|
rsp, err := http.Get(fmt.Sprintf("http://localhost:8080%s", getUrl()))
|
||
|
|
if err != nil {
|
||
|
|
fmt.Fprintf(os.Stderr, "%v\n", err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
_, _ = io.ReadAll(rsp.Body)
|
||
|
|
_ = rsp.Body.Close()
|
||
|
|
}()
|
||
|
|
|
||
|
|
time.Sleep(time.Microsecond * 100)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
r.Get("/apple", func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
w.WriteHeader(200)
|
||
|
|
time.Sleep(time.Millisecond * 100)
|
||
|
|
_, _ = w.Write([]byte("OK"))
|
||
|
|
})
|
||
|
|
go runClient(func() string { return "/apple" })
|
||
|
|
|
||
|
|
r.Get("/banana", func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
w.WriteHeader(200)
|
||
|
|
_, _ = w.Write(bytes.Repeat([]byte("f"), 1<<10))
|
||
|
|
|
||
|
|
if replica == "0" {
|
||
|
|
time.Sleep(time.Second)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
go runClient(func() string { return "/banana" })
|
||
|
|
|
||
|
|
var k atomic.Int32
|
||
|
|
r.Get("/orange", func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
if k.Add(1)%100 == 0 {
|
||
|
|
w.WriteHeader(500)
|
||
|
|
} else {
|
||
|
|
w.WriteHeader(200)
|
||
|
|
}
|
||
|
|
|
||
|
|
})
|
||
|
|
go runClient(func() string { return "/orange" })
|
||
|
|
|
||
|
|
r.Get("/kiwi/{id}", func(w http.ResponseWriter, r *http.Request) {
|
||
|
|
if replica == "3" {
|
||
|
|
w.WriteHeader(200)
|
||
|
|
} else {
|
||
|
|
w.WriteHeader(500)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
var i atomic.Int32
|
||
|
|
go runClient(func() string { return fmt.Sprintf("/kiwi/%d", i.Add(1)) })
|
||
|
|
}
|