handle possible panic (#88)

Reviewed-on: https://gitea.com/gitea/act_runner/pulls/88
Reviewed-by: Jason Song <i@wolfogre.com>

poller/poller.go

@@ -57,24 +57,17 @@ func (p *Poller) Wait() {
 	p.routineGroup.Wait()
 }
 
-func (p *Poller) Poll(ctx context.Context) error {
+func (p *Poller) handle(ctx context.Context, l *log.Entry) {
-	l := log.WithField("func", "Poll")
+	defer func() {
-
+		if r := recover(); r != nil {
-	for {
+			l.Errorf("handle task panic: %+v", r)
-		// check worker number
-		p.schedule()
-
-		select {
-		// wait worker ready
-		case <-p.ready:
-		case <-ctx.Done():
-			return nil
 		}
-	LOOP:
+	}()
+
 	for {
 		select {
 		case <-ctx.Done():
-				break LOOP
+			return
 		default:
 			task, err := p.pollTask(ctx)
 			if task == nil || err != nil {
@@ -93,9 +86,25 @@ func (p *Poller) Poll(ctx context.Context) error {
 					l.Errorf("execute task: %v", err.Error())
 				}
 			})
-				break LOOP
+			return
 		}
 	}
+}
+
+func (p *Poller) Poll(ctx context.Context) error {
+	l := log.WithField("func", "Poll")
+
+	for {
+		// check worker number
+		p.schedule()
+
+		select {
+		// wait worker ready
+		case <-p.ready:
+		case <-ctx.Done():
+			return nil
+		}
+		p.handle(ctx, l)
 	}
 }