feat: add panic recover for probe

exporter/templates_test.go

@@ -1,8 +1,8 @@
 package exporter
 
 import (
-	"reflect"
 	"fmt"
+	"reflect"
 
 	"testing"
 )
@@ -45,9 +45,9 @@ func TestInvertIncidents(t *testing.T) {
 func TestBreakText(t *testing.T) {
 	f := templateFuncs["break_text"].(func(string, int) []string)
 
-	tests := []struct{
-		Input string
-		Width int
+	tests := []struct {
+		Input  string
+		Width  int
 		Output []string
 	}{
 		{"hello_world", 20, []string{"hello_world"}},
@@ -69,9 +69,9 @@ func TestBreakText(t *testing.T) {
 func TestAlignCenter(t *testing.T) {
 	f := templateFuncs["align_center"].(func(string, int) string)
 
-	tests := []struct{
-		Input string
-		Width int
+	tests := []struct {
+		Input  string
+		Width  int
 		Output string
 	}{
 		{"foobar", 10, "  foobar"},

main.go

@@ -7,6 +7,7 @@ import (
 	"os"
 	"sync"
 	"sync/atomic"
+	"time"
 
 	"github.com/macrat/ayd/exporter"
 	"github.com/macrat/ayd/probe"
@@ -198,9 +199,19 @@ func RunServer(s *store.Store, tasks []Task) {
 
 		s.AddTarget(t.Probe.Target())
 
-		f := t.Probe.Check
+		p := t.Probe
 		job := func() {
-			s.Append(f()...)
+			defer func() {
+				if err := recover(); err != nil {
+					s.Append(store.Record{
+						CheckedAt: time.Now(),
+						Target:    p.Target(),
+						Status:    store.STATUS_UNKNOWN,
+						Message:   fmt.Sprintf("panic: %s", err),
+					})
+				}
+			}()
+			s.Append(p.Check()...)
 		}
 
 		if t.Schedule.NeedKickWhenStart() {