Add test for issue 87

jit/jit_test.go

@@ -310,6 +310,35 @@ func TestJitEmbeddedStruct(t *testing.T) {
 	}
 }
 
+func TestSchedule(t *testing.T) {
+	conf := baseConfig
+	data := testData{
+		files: []string{"./testdata/test_schedule/test.go"},
+		pkg:   "testdata/test_schedule",
+	}
+	testNames := []string{"BuildGoFiles", "BuildGoPackage", "BuildGoText"}
+
+	for _, testName := range testNames {
+		t.Run(testName, func(t *testing.T) {
+			module, symbols := buildLoadable(t, conf, testName, data)
+
+			test := symbols["Test"].(func())
+			for i := 0; i < 100; i++ {
+				fmt.Println("Test ", i)
+				test()
+			}
+
+			err := module.Unload()
+			if err != nil {
+				t.Fatal(err)
+			}
+			if err != nil {
+				t.Fatal(err)
+			}
+		})
+	}
+}
+
 func TestJitCGoCall(t *testing.T) {
 	if os.Getenv("CGO_ENABLED") == "0" {
 		t.Skip("CGo disabled")

jit/testdata/test_schedule/test.go

@@ -0,0 +1,82 @@
+package test_schedule
+
+import (
+	"fmt"
+	"runtime"
+	"sync"
+	"time"
+)
+
+func Test() {
+	var ch = make(chan bool, 20000)
+	var begin = make(chan bool)
+	var closeCh = make(chan bool)
+
+	w := sync.WaitGroup{}
+
+	w.Add(1)
+	go func() {
+		defer logPanic()
+		runtime.LockOSThread()
+		<-begin
+		fmt.Println("begin")
+		tm := time.Now()
+		for i := 0; i < 10000000; i++ {
+			<-ch
+		}
+		fmt.Println(time.Now().Sub(tm))
+		close(closeCh)
+		w.Done()
+	}()
+
+	for i := 0; i < 50000; i++ {
+		w.Add(1)
+		go func() {
+			defer logPanic()
+			var count int
+			load := 100
+			for {
+				count++
+				re := fmt.Sprint("haha")
+				if re == "haha" && count >= load {
+					count = 0
+					runtime.Gosched()
+				}
+				select {
+				case <-closeCh:
+					w.Done()
+					return
+				default:
+				}
+			}
+		}()
+	}
+
+	for i := 0; i < 20; i++ {
+		w.Add(1)
+		go func() {
+			defer logPanic()
+			for {
+				select {
+				case <-closeCh:
+					w.Done()
+					return
+				case ch <- true:
+				}
+			}
+		}()
+	}
+
+	fmt.Println("all start")
+	begin <- true
+
+	w.Wait()
+}
+
+func logPanic() {
+	if r := recover(); r != nil {
+		trace := make([]byte, 1024)
+		count := runtime.Stack(trace, false)
+		fmt.Printf("Stack of %d bytes: %s\n", count, trace[:count])
+	}
+}