gotask

gotask

task for go

package gotask_test

import (
	"fmt"
	"github.com/pubgo/errors"
	"github.com/pubgo/gotask"
	"github.com/rs/zerolog"
	"io"
	"net/http"
	"testing"
	"time"
)

func TestTasks(t *testing.T) {
	defer errors.Debug()

	zerolog.SetGlobalLevel(zerolog.WarnLevel)

	gotask.TaskRegistry("fn", func(i int) {
		errors.ErrHandle(errors.Try(func() {
			errors.T(i == 29, "90999 error")
		}), func(err *errors.Err) {
			errors.Wrap(err, "wrap")
		})
	})

	var task = gotask.NewTask(10, time.Second+time.Millisecond*10)
	for i := 0; i < 100; i++ {
		task.Do("fn", i)
	}
	task.Wait()
	errors.P(task.Stat())
}

func TestErrLog(t *testing.T) {
	defer errors.Debug()

	gotask.TaskRegistry("fn", func(i int) {
		errors.T(i == 90999, "90999 error")
	})

	var task = gotask.NewTask(500, time.Second+time.Millisecond*10)
	for i := 0; i < 100000; i++ {
		task.Do("fn", i)
	}

	task.Wait()
	errors.P(task.Stat())
}

func parserArticleWithReadability(i int) {
	defer errors.Handle()()

	errChan := make(chan bool)
	go func() {
		time.Sleep(time.Second * 4)
		errChan <- true
	}()

	for {
		select {
		case <-time.After(3 * time.Second):
			errors.Wrap("readbility timeout", "等待 %d", i)
		case <-errChan:
			return
		}
	}

}

func TestW(t *testing.T) {
	defer errors.Debug()

	gotask.TaskRegistry("fn", func(i int) {
		errors.ErrHandle(errors.Try(func() {})(func() {
			parserArticleWithReadability(i)
			fmt.Println("ok", i)
		}), func(err *errors.Err) {
			errors.Wrap(err, "testW")
		})
	})

	var task = gotask.NewTask(10000, time.Second*2)
	for i := 0; i < 1000000; i++ {
		task.Do("fn", i)
	}
	task.Wait()
	errors.P(task.Stat())
}

func isEOF(err error) bool {
	return err == io.EOF || err == io.ErrUnexpectedEOF
}

func TestUrl(t *testing.T) {
	client := &http.Client{Transport: &http.Transport{
		MaxIdleConns:       10,
		IdleConnTimeout:    3 * time.Second,
		DisableCompression: true,
	}}
	client.Timeout = 5 * time.Second

	gotask.TaskRegistry("fn", func(c *http.Client, i int) {
		errors.ErrLog(errors.Retry(3, func() {
			fmt.Println("try: ", i)
			req, err := http.NewRequest(http.MethodGet, "http://baidu.com", nil)
			errors.Panic(err)
			req.Close = true

			resp, err := c.Do(req)
			errors.Panic(err)
			errors.T(resp.StatusCode != http.StatusOK, "状态不正确%d", resp.StatusCode)
		}))
	})

	var task = gotask.NewTask(50, time.Second*2)
	for i := 0; i < 300; i++ {
		fmt.Println(i)
		task.Do("fn", client, i)
	}
	task.Wait()
	errors.P(task.Stat())

}