Broadcast working
This commit is contained in:
parent
8ef58bc9e8
commit
b9dd0db98b
45
main.go
45
main.go
@ -5,25 +5,60 @@ import (
|
||||
"io"
|
||||
"log"
|
||||
"net/http"
|
||||
"os"
|
||||
"strings"
|
||||
)
|
||||
|
||||
const (
|
||||
MethodPurge = "PURGE"
|
||||
)
|
||||
|
||||
var (
|
||||
varnishList = MakeVarnishList()
|
||||
)
|
||||
|
||||
// MakeVarnishList reads the list of varnish servers from a file on disk.
|
||||
func MakeVarnishList() []string {
|
||||
Data, err := os.ReadFile("./varnish")
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
sliceData := strings.Split(string(Data), ",")
|
||||
return sliceData
|
||||
}
|
||||
|
||||
// PurgeHandler handles PURGE request to broadcast it to all varnish instances.
|
||||
func PurgeHandler(w http.ResponseWriter, r *http.Request) {
|
||||
client := &http.Client{}
|
||||
var resp http.Response
|
||||
|
||||
// Retrieve URL sent and send an HTTP PURGE request to all varnish servers.
|
||||
url := r.URL.String()
|
||||
req, err := http.NewRequest(MethodPurge, "http://10.13.32.1:6081" + url, nil)
|
||||
resp, err := client.Do(req)
|
||||
fmt.Printf("Error: %v\n", err)
|
||||
fmt.Printf("Resp: %v\n", resp)
|
||||
SendToVarnish(url)
|
||||
io.WriteString(w, resp.Status)
|
||||
}
|
||||
|
||||
func SendToVarnish(url string) int {
|
||||
var status int
|
||||
var respListStatus []string
|
||||
for i := 0; i < len(varnishList); i++ {
|
||||
client := &http.Client{}
|
||||
domain := strings.Trim(varnishList[i], "\r\n")
|
||||
req, err := http.NewRequest(MethodPurge, domain + url, nil)
|
||||
//fmt.Printf("Request: %v",req)
|
||||
if err != nil {
|
||||
log.Fatal("NewRquest: %s", err)
|
||||
}
|
||||
resp, err := client.Do(req)
|
||||
if err != nil {
|
||||
log.Fatal("Client.do: %s", err)
|
||||
}
|
||||
fmt.Printf("Resp: %v\n", resp)
|
||||
respListStatus = append(respListStatus, resp.Status)
|
||||
}
|
||||
fmt.Println(respListStatus)
|
||||
return status
|
||||
}
|
||||
|
||||
func main() {
|
||||
http.HandleFunc("/", PurgeHandler)
|
||||
log.Fatal(http.ListenAndServe(":6081",nil))
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user