rutina/example/policies.go

// +build ignore

package main

import (
	"context"
	"errors"
	"github.com/neonxp/rutina"
	"log"
	"time"
)

func main() {
	// New instance with builtin context
	r := rutina.New()

	errsChan := make(chan error, 1)

	r = r.With(rutina.WithErrChan(errsChan))

	r.Go(func(ctx context.Context) error {
		<-time.After(1 * time.Second)
		log.Println("Do something 1 second without errors and restart")
		return nil
	}, rutina.RestartIfDone, rutina.ShutdownIfFail)

	r.Go(func(ctx context.Context) error {
		<-time.After(2 * time.Second)
		log.Println("Do something 2 seconds without errors and do nothing")
		return nil
	}, rutina.DoNothingIfDone, rutina.ShutdownIfFail)

	r.Go(func(ctx context.Context) error {
		<-time.After(3 * time.Second)
		log.Println("Do something 3 seconds with error and restart")
		return errors.New("Error #1!")
	}, rutina.RestartIfFail)

	r.Go(func(ctx context.Context) error {
		<-time.After(4 * time.Second)
		log.Println("Do something 4 seconds with error and do nothing")
		return errors.New("Error #2!")
	}, rutina.DoNothingIfFail)

	r.Go(func(ctx context.Context) error {
		<-time.After(10 * time.Second)
		log.Println("Do something 10 seconds with error and close context")
		return errors.New("Successfully shutdown at proper place")
	}, rutina.ShutdownIfFail)

	r.Go(func(ctx context.Context) error {
		for {
			select {
			case <-ctx.Done():
				log.Println("Shutdown chan listener")
				return nil
			case err := <-errsChan:
				log.Printf("Error in chan: %v", err)
			}
		}
	})

	// OS signals subscriber
	r.ListenOsSignals()

	if err := r.Wait(); err != nil {
		log.Fatal(err)
	} else {
		log.Println("Routines stopped but not correct")
	}
}