Adding proper logging file, cleaning main.go, gitignore update
This commit is contained in:
parent
f3be636af4
commit
f6173a09b2
1
.gitignore
vendored
1
.gitignore
vendored
@ -1 +1,2 @@
|
|||||||
*.swp
|
*.swp
|
||||||
|
*.log
|
||||||
|
|||||||
102
app/main.go
102
app/main.go
@ -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))
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user