golang 实时获取测试覆盖率

[myml]

使用未导出字段

package main

import (
	"context"
	"flag"
	"fmt"
	"log"
	"net/http"
	"reflect"
	"sync"
	"testing"
	"unsafe"
)

func TestRunMain(t *testing.T) {
	main()
}

func TestMain(m *testing.M) {
	flag.Set("test.coverprofile", "c.out")

	mux := http.NewServeMux()
	mux.HandleFunc("/coverprofile", func(resp http.ResponseWriter, req *http.Request) {
		log.Printf("coverage %v", testing.Coverage())
	})
	mux.HandleFunc("/shutdown", func(resp http.ResponseWriter, req *http.Request) {
		err := srv.Shutdown(context.TODO())
		if err != nil {
			fmt.Fprint(resp, err)
			resp.WriteHeader(500)
		}
	})
	go func() {
		for {
			code := m.Run()
			log.Println("code", code)

			v := reflect.ValueOf(m).Elem()
			afterOnce := v.FieldByName("afterOnce")
			afterOnce = reflect.NewAt(afterOnce.Type(), unsafe.Pointer(afterOnce.UnsafeAddr())).Elem()
			afterOnce.Set(reflect.ValueOf(sync.Once{}))
		}
	}()
	http.ListenAndServe(":9222", mux)
}

使用未导出函数

package main

import (
	"context"
	"flag"
	"fmt"
	"log"
	"net/http"
	"testing"
	_ "unsafe"
)

//go:linkname coverReport testing.coverReport
func coverReport()

func TestRunMain(t *testing.T) {
	main()
}

func TestMain(m *testing.M) {
	flag.Set("test.coverprofile", "c.out")

	mux := http.NewServeMux()
	mux.HandleFunc("/coverprofile", func(resp http.ResponseWriter, req *http.Request) {
		log.Printf("coverage %v", testing.Coverage())
	})
	mux.HandleFunc("/coverReport", func(resp http.ResponseWriter, req *http.Request) {
		coverReport()
	})
	mux.HandleFunc("/shutdown", func(resp http.ResponseWriter, req *http.Request) {
		err := srv.Shutdown(context.TODO())
		if err != nil {
			fmt.Fprint(resp, err)
			resp.WriteHeader(500)
		}
	})
	go func() {
		m.Run()
	}()
	http.ListenAndServe(":9222", mux)
}