Skip to content

Instantly share code, notes, and snippets.

@zaz600
Created June 4, 2015 11:56
Show Gist options
  • Save zaz600/7bad4de9df2d3ab9f1f6 to your computer and use it in GitHub Desktop.
Save zaz600/7bad4de9df2d3ab9f1f6 to your computer and use it in GitHub Desktop.
// web_check.go
package main
import (
"flag"
"fmt"
"net/http"
"os"
"time"
)
var (
check_history []string
url string
hist_length int
timeout int
ip string
)
func main() {
if !parse_args() {
return
}
go check_loop()
http.HandleFunc("/", indexHandler)
http.ListenAndServe(":8088", nil)
}
func parse_args() bool {
flag.StringVar(&url, "url", "", "Адрес для проверки. Например, http://golang.org/")
flag.IntVar(&timeout, "t", 5, "Период проверки в минутах. Должен быть больше 2")
flag.IntVar(&hist_length, "l", 30, "Длина истории в браузере")
flag.StringVar(&ip, "i", ":8090", "ip:port для веб-статистики")
flag.Parse()
if url == "" {
fmt.Println("Не задан параметр -url", url)
return false
}
if timeout < 2 {
fmt.Println("Значение -i должно быть больше 2. Задано: ", timeout)
return false
}
if hist_length < 1 {
fmt.Println("Значение -l должно быть больше 1. Задано: ", hist_length)
return false
}
return true
}
func check_loop() {
for {
tm := time.Now().Format("2006-01-02 15:04:05")
// статус, который возвращает check, пока не используем, поэтому ставим _
_, msg := check(url)
log_to_file(tm, msg)
save_history(tm, msg)
fmt.Println(tm, msg)
time.Sleep(time.Duration(timeout) * time.Minute)
}
}
func check(url string) (bool, string) {
// возвращает true - если сервис доступен, false, если нет и текст сообщения
fmt.Println("Проверяем адрес ", url)
resp, err := http.Get(url)
if err != nil {
return false, fmt.Sprintf("Ошибка соединения. %s", err)
}
defer resp.Body.Close()
if resp.StatusCode != 200 {
return false, fmt.Sprintf("Ошибка. http-статус: %s", resp.StatusCode)
}
return true, fmt.Sprintf("Онлайн. http-статус: %d", resp.StatusCode)
}
func log_to_file(tm, s string) {
// Сохраняет сообщения в файл
f, err := os.OpenFile("web_check.log", os.O_RDWR|os.O_APPEND|os.O_CREATE, 0644)
if err != nil {
fmt.Println(tm, err)
return
}
defer f.Close()
if _, err = f.WriteString(fmt.Sprintln(tm, s)); err != nil {
fmt.Println(tm, err)
}
}
func save_history(tm, s string) {
// добавляет запись в массив с историей проверок
check_history = append(check_history, fmt.Sprintf("%s %s", tm, s))
if len(check_history) > hist_length {
check_history = check_history[1:]
}
}
func indexHandler(w http.ResponseWriter, r *http.Request) {
// Выдает историю проверок в браузер
var html = `<html><head><title>Проверка веб-службы</title></head><body><h1>История проверки</h1><div>%s</div></body></html>`
s := ""
for _, h := range check_history {
s += fmt.Sprintf("%s<br/>\n", h)
}
fmt.Fprintf(w, fmt.Sprintf(html, s))
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment