Adding proper logging file, cleaning main.go, gitignore update

This commit is contained in:
Sebastien Laithier 2023-05-11 11:53:27 +02:00
parent f3be636af4
commit f6173a09b2
2 changed files with 54 additions and 49 deletions

1
.gitignore vendored
View File

@ -1 +1,2 @@
*.swp *.swp
*.log

View File

@ -1,77 +1,81 @@
// Receive http PURGE request and broadcast it to several Varnish servers.
package main package main
import ( import (
"fmt" "io"
"io" "log"
"log" "net/http"
"net/http" "os"
"os" "strings"
"strings"
) )
const ( const (
// MethodPurge declaration for Varnish. // MethodPurge declaration for Varnish.
MethodPurge = "PURGE" MethodPurge = "PURGE"
) )
var ( var (
varnishList = MakeVarnishList() varnishList = MakeVarnishList()
status = "200 Purged" status = "200 Purged"
) )
// MakeVarnishList reads the list of varnish servers from a file on disk. // MakeVarnishList reads the list of varnish servers from a file on disk.
func MakeVarnishList() []string { func MakeVarnishList() []string {
Data, err := os.ReadFile("./varnish") Data, err := os.ReadFile("./varnish")
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
} }
sliceData := strings.Split(string(Data), ",") sliceData := strings.Split(string(Data), ",")
return sliceData return sliceData
} }
// PurgeHandler handles PURGE request to broadcast it to all varnish instances. // PurgeHandler handles PURGE request to broadcast it to all varnish instances.
func PurgeHandler(w http.ResponseWriter, r *http.Request) { func PurgeHandler(w http.ResponseWriter, r *http.Request) {
url := r.URL.String() url := r.URL.String()
status := SendToVarnish(url) status := SendToVarnish(url)
if status != "200 Purged" { if status != "200 Purged" {
w.WriteHeader(405) w.WriteHeader(405)
} }
io.WriteString(w, status) io.WriteString(w, status)
} }
// SendToVarnish send to all varnish servers define in varnishList the PURGE request. // SendToVarnish send to all varnish servers define in varnishList the PURGE request.
func SendToVarnish(url string) string { func SendToVarnish(url string) string {
status = "200 Purged" status = "200 Purged"
// Take url to ban as argument. // Take url to ban as argument.
// Loop over the list of Varnish servers and send PURGE request to each. // Loop over the list of Varnish servers and send PURGE request to each.
for i := 0; i < len(varnishList); i++ { // Update status variable to check if servers have successfully purge url.
//status = "200 Purged" for i := 0; i < len(varnishList); i++ {
client := &http.Client{} client := &http.Client{}
domain := strings.Trim(varnishList[i], "\r\n") domain := strings.Trim(varnishList[i], "\r\n")
req, err := http.NewRequest(MethodPurge, domain + url, nil) req, err := http.NewRequest(MethodPurge, domain+url, nil)
if err != nil { if err != nil {
log.Fatal("NewRequest: %s", err) log.Fatal("NewRequest: %s", err)
} }
resp, err := client.Do(req) resp, err := client.Do(req)
if err != nil { if err != nil {
log.Fatal("Client.do: %s", err) log.Fatal("Client.do: %s", err)
} }
//fmt.Printf("Resp: %v\n", resp) if resp.StatusCode != 200 {
if resp.StatusCode != 200 { status = "405 Not Allowed"
status = "405 Not Allowed" }
} log.Println("Purge " + domain + url + " : " + status)
fmt.Println("Purge URL" + " " + domain + url + " : " + status) }
} return status
return status
} }
// HealthHandler handles healthcheck requests and return 200. // HealthHandler handles healthcheck requests and return 200.
func HealthHandler(w http.ResponseWriter, r *http.Request) { func HealthHandler(w http.ResponseWriter, _ *http.Request) {
io.WriteString(w, "OK") io.WriteString(w, "OK")
} }
func main() { func main() {
http.HandleFunc("/", PurgeHandler) logFile, err := os.OpenFile("./log/purge.log", os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
http.HandleFunc("/healthcheck", HealthHandler) if err != nil {
log.Fatal(http.ListenAndServe(":6081",nil)) panic(err)
}
log.SetOutput(logFile)
http.HandleFunc("/", PurgeHandler)
http.HandleFunc("/healthcheck", HealthHandler)
log.Fatal(http.ListenAndServe(":6081", nil))
} }