summaryrefslogtreecommitdiff
path: root/app/fetch.go
blob: c8b0376944be0b7f735d8cc7ecf69f78b49c0217 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
package main

import (
	"encoding/json"
	"fmt"
	//	"strconv"
	//	"time"
	"log"
)

func run() {
	c := c
	c.AddFunc("@every 10s", healthCheck)
	c.Start()
	jobs := c.Entries()

	for _, i := range jobs {
		fmt.Printf("Job: %v, Schedule: %v; Next %v; Prev %v \n", i.Job, i.Schedule, i.Next, i.Prev)
	}
}

const (
	cache_prefix = "status_"
)

func healthCheck() {
	h := []Host{}
	db := Db
	//db.Where("monitored = ?", true).Find(&h)
	db.Find(&h)

	c := pool.Get()
	defer c.Close()

	for k, v := range h {
		if !h[k].Monitored {
			continue
		}

		if resp, _, err := HttpGet(v.Url); err != nil {
			//fmt.Println(err)

			h[k].Status = "Error"
			h[k].StatusCode = 0
			h[k].Success = false
			h[k].Monitored = false
			h[k].Reason = fmt.Sprintf("%v", err)
			Db.Debug().Save(&h[k])

		} else {
			//		fmt.Println("Health check: Okay! ", resp.Status)

			h[k].Status = resp.Status
			h[k].StatusCode = int64(resp.StatusCode)
			h[k].Success = true
			h[k].Reason = ""
			Db.Debug().Save(&h[k])
		}

		// save struct in redis as serialized json
		// TODO
		/*
			j, err := json.Marshal(h[k])
			if err != nil {
				log.Println("Error: ", err)
				continue
			}
			c.Do("SET", cache_prefix+strconv.Itoa(k), j)
		*/

		//c.Do("SET", cache_prefix+strconv.Itoa(k), j)
		//c.Do("SET", k, j)

		//	fmt.Printf("Id: %v, Url: %v \n", v.Id, v.Url)

	}

	js, err := json.Marshal(h)

	if err != nil {
		log.Fatal(err)
	} else {
		c.Do("SET", cache_prefix+"database", js)
	}
}