How to use Sum method of example_test Package

1package trace2import (3	"bytes"4	"fmt"5	"runtime"6	"strconv"7	"sync"8)9// 利用代码生成自动注入 Trace 函数10// 要实现向目标代码中的函数 / 方法自动注入 Trace 函数，首先要做的就是将 Trace 函数相关的代码打包到一个 module 中以方便其他 module 导入。11// 下面就先来看看将 Trace 函数放入一个独立的 module 中的步骤。12// 将 Trace 函数放入一个独立的 module 中13// 创建一个名为 instrument_trace 的目录，进入这个目录后，通过 go mod init 命令创建一个名为 github.com/lcy2013/instrument_trace 的 module：14// $mkdir instrument_trace15// $cd instrument_trace16// $go mod init github.com/lcy2013/instrument_trace17// 将最新版的 trace.go 放入到该目录下，将包名改为 trace，并仅保留 Trace 函数、Trace 使用的函数以及包级变量，其他函数一律删除掉。18//19// 作为 trace 包的作者，有义务告诉大家如何使用 trace 包。20// 在 Go 中，通常我们会用一个 example_test.go 文件来编写使用 trace 包的演示代码，21// 为 trace 包提供的 example_test.go 文件。22// 自动注入 Trace 函数23// 现在，在 instrument_trace module 下面增加一个命令行工具，这个工具可以以一个 Go 源文件为单位，自动向这个 Go 源文件中的所有函数注入 Trace 函数。24// 带有可执行文件的 Go 项目布局，在 instrument_trace module 中增加 cmd/instrument 目录，这个工具的 main 包就放在这个目录下，而真正实现自动注入 Trace 函数的代码呢，被我们放在了 instrumenter 目录下。25// $ tree ./instrument_trace26// ./instrument_trace27// ├── Makefile28// ├── cmd/29// │   └── instrument/30// │       └── main.go  # instrument命令行工具的main包31// ├── example_test.go32// ├── go.mod33// ├── go.sum34// ├── instrumenter/    # 自动注入逻辑的相关结构35// │   ├── ast/36// │   │   └── ast.go37// │   └── instrumenter.go38// └── trace.go39// 先看 cmd/instrument/main.go 源码，然后自上而下沿着 main 函数的调用逻辑逐一看一下这个功能的实现。40// 不过，由于 http2curGoroutineID 不是一个导出函数，我们无法直接使用。可以把它复制出来改造一下：41var goroutineSpace = []byte("goroutine ")42// curGoroutineID 改造了两个地方。43// 一个地方是通过直接创建一个 byte 切片赋值给 b，替代原 http2curGoroutineID 函数中从一个 pool 池获取 byte 切片的方式，44// 另外一个是使用 strconv.ParseUint 替代了原先的 http2parseUintBytes。45// 改造后，就可以直接使用 curGoroutineID 函数来获取 Goroutine 的 ID 信息了。46func curGoroutineID() uint64 {47	b := make([]byte, 64)48	b = b[:runtime.Stack(b, false)]49	// Parse the 4707 out of "goroutine 4707 ["50	b = bytes.TrimPrefix(b, goroutineSpace)51	i := bytes.IndexByte(b, ' ')52	if i < 0 {53		panic(fmt.Sprintf("No space found in %q", b))54	}55	b = b[:i]56	n, err := strconv.ParseUint(string(b), 10, 64)57	if err != nil {58		panic(fmt.Sprintf("Failed to parse goroutine ID out of %q: %v", b, err))59	}60	return n61}62var mu sync.Mutex63var m = make(map[uint64]int)64// Trace 使用了一个 map 类型变量 m 来保存每个 Goroutine 当前的缩进信息：m 的 key 为 Goroutine 的 ID，值为缩进的层次。65// 然后，考虑到 Trace 函数可能在并发环境中运行，根据 map 不支持并发写的注意事项，增加了一个 sync.Mutex 实例 mu 用于同步对 m 的写操作。66// 对于一个 Goroutine 来说，每次刚进入一个函数调用，就在输出入口跟踪信息之前，将缩进层次加一，并输出入口跟踪信息，加一后的缩进层次值也保存到 map 中。67// 然后，在函数退出前，取出当前缩进层次值并输出出口跟踪信息，之后再将缩进层次减一后保存到 map 中。68// 除了增加缩进层次信息外，在这一版的 Trace 函数实现中，也把输出出入口跟踪信息的操作提取到了一个独立的函数 printTrace 中，这个函数会根据传入的 Goroutine ID、函数名、箭头类型与缩进层次值，按预定的格式拼接跟踪信息并输出。69func Trace() func() {70	// 通过 runtime.Caller 函数获得当前 Goroutine 的函数调用栈上的信息71	// runtime.Caller 的参数标识的是要获取的是哪一个栈帧的信息。72	// 当参数为 0 时，返回的是 Caller 函数的调用者的函数信息，在这里就是 Trace 函数。73	// 但我们需要的是 Trace 函数的调用者的信息，于是我们传入 1。74	//75	// Caller 函数有四个返回值：76	// 第一个返回值代表的是程序计数（pc）；77	// 第二个和第三个参数代表对应函数所在的源文件名以及所在行数，这里我们暂时不需要；78	// 最后一个参数代表是否能成功获取这些信息，如果获取失败，抛出 panic。79	pc, _, _, ok := runtime.Caller(1)80	if !ok {81		panic("not found caller")82	}83	// 通过 runtime.FuncForPC 函数和程序计数器（PC）得到被跟踪函数的函数名称。84	// runtime.FuncForPC 返回的名称中不仅仅包含函数名，还包含了被跟踪函数所在的包名。85	fn := runtime.FuncForPC(pc)86	// 获取上一个栈针的函数名称87	name := fn.Name()88	// 获取当前goroutine的信息89	gid := curGoroutineID()90	mu.Lock()91	indents := m[gid]92	m[gid] = indents + 193	mu.Unlock()94	printTrace(gid, name, "->", indents+1)95	return func() {96		mu.Lock()97		indent := m[gid]98		m[gid] = indent - 199		mu.Unlock()100		printTrace(gid, name, "<-", indent)101	}102}103// printTrace 格式化缩进输出104func printTrace(id uint64, name, arrow string, indent int) {105	indents := ""106	for i := 0; i < indent; i++ {107		indents = fmt.Sprintf("%s%s", indents, "	")108	}109	fmt.Printf("g[%05d]: %s%s%s\n", id, indents, arrow, name)110}...

1/*2@Time    : 11/21/21 21:143@Author  : nil4@File    : example_test.go5*/6package chaninvoke_test7import (8	"chaninvoke"9	"fmt"10	"testing"11	"time"12)13func TestExample(t *testing.T) {14	s := chaninvoke.StartServer(10)15	done := make(chan struct{})16	go func(){17		s.Register("f0", func(args ...interface{}) interface{}{18			sum := 019			for _, v := range args {20				sum += v.(int)21			}22			return interface{}(sum)23		})24		s.Register("f1", func(args ...int) int {25			product := 126			for _, v := range args {27				product *= v28			}29			return product30		})31		s.Register("f2", func(i, j string) string {32			return i+j33		})34		s.Register("f3", func(t int) int {35			time.Sleep(time.Second * time.Duration(t))36			fmt.Println("f3 end")37			return t38		})39		done <- struct{}{}40		for {41			s.Exec(<- s.CallChan)42		}43	}()44	fmt.Println("register done")45	<-done46	go func() {47		c := chaninvoke.StartClient(10, s)48		fmt.Println(c.Call("f0", 1, 2, 3, 4))49		fmt.Println(c.Call("f1", 1, 2, 3, 4))50		fmt.Println(c.Call("f1", 1))51		fmt.Println(c.Call("f2", "hello", "world"))52		c.AsyncCall("f3", 1, func(t int) {53			fmt.Printf("you finally awake, you sleep %d seconds\n", t)54		})55		c.Cb(<- c.AsyncRetChan)56		done <- struct{}{}57	}()58	<-done59}...

...5	"github.com/ysmood/got"6	"github.com/ysmood/got/lib/example"7)8func TestSuite(t *testing.T) {9	// Execute each exported methods of SumSuite.10	// Each exported methods on SumSuite is a test case.11	got.Each(t, SumSuite{})12}13type SumSuite struct {14	got.G15}16func (g SumSuite) Sum() {17	g.Eq(example.Sum("1", "1"), "2")18}19func TestSumAdvancedSuite(t *testing.T) {20	// The got.Each can also accept a function to init the g for each test case.21	got.Each(t, func(t *testing.T) SumAdvancedSuite {22		g := got.New(t)23		// Concurrently run each test24		g.Parallel()25		// Timeout for each test26		g.PanicAfter(time.Second)27		return SumAdvancedSuite{g, "1", "2"}28	})29}30type SumAdvancedSuite struct {31	got.G32	a, b string33}34func (g SumAdvancedSuite) Sum() {35	g.Eq(example.Sum(g.a, g.b), "3")36}

1import (2func main() {3	fmt.Println("Hello, playground")4	fmt.Println(example.Sum(1,2))5}6import "testing"7func TestSum(t *testing.T) {8	if Sum(1,2) != 3 {9		t.Error("Expected 3")10	}11}12import "testing"13func TestSum(t *testing.T) {14	if Sum(1,2) != 3 {15		t.Error("Expected 3")16	}17}18func Sum(a int, b int) int {19}20--- PASS: TestSum (0.00s)21--- PASS: TestSum (0.00s)22--- PASS: TestSum (0.00s)23--- PASS: TestSum (0.00s)

1import "fmt"2func main() {3    fmt.Println(x.Sum())4}5import "fmt"6type example_test struct {7}8func (x example_test) Sum() int {9}10func main() {11    fmt.Println(x.Sum())12}13import "fmt"14type example_test struct {15}16func (x *example_test) Sum() int {17}18func main() {19    fmt.Println(x.Sum())20}21import "fmt"22type example_test struct {23}

1func main() {2    var example = example_test.Example{}3    fmt.Println(example.Sum(1, 2))4}5type Example struct{}6func (e Example) Sum(a, b int) int {7}

1import (2func main() {3	fmt.Println(hello.Sum(1, 2))4}5import (6func main() {7	fmt.Println(hello.Sum(1, 2))8	fmt.Println(world.Multiply(2, 3))9}10import (11func main() {12	fmt.Println(hello.Sum(1, 2))13	fmt.Println(world.Multiply(2, 3))14}15import (16func main() {17	fmt.Println(hello.Sum(1, 2))18	fmt.Println(world.Multiply(2, 3))19}

1import (2func main() {3	fmt.Println(example_test.Sum(5, 6))4}5import (6func Sum(a int, b int) int {7}8func TestSum(t *testing.T) {9	if Sum(5, 6) != 11 {10		t.Error("Expected 11, got ", Sum(5, 6))11	}12}13import (14func Sum(a int, b int) int {15}16func TestSum(t *testing.T) {17	if Sum(5, 6) != 11 {18		t.Error("Expected 11, got ", Sum(5, 6))19	}20}21func BenchmarkSum(b *testing.B) {22	for i := 0; i < b.N; i++ {23		Sum(5, 6)24	}25}26import (27func Sum(a int, b int) int {28}29func TestSum(t *testing.T) {30	t.Run("5+6=11", func(t *testing.T) {31		if Sum(5, 6) != 11 {32			t.Error("Expected 11, got ", Sum(5, 6))33		}34	})35	t.Run("0+0=0", func(t *testing.T) {36		if Sum(0, 0) != 0 {37			t.Error("Expected 0, got ", Sum(0, 0))38		}39	})40}41import (42func Sum(a int, b int) int {

1import (2func main() {3	fmt.Println(example.Sum(10, 20))4}5import "testing"6func TestSum(t *testing.T) {7	if Sum(10, 20) != 30 {8		t.Error("Sum(10, 20) should be 30")9	}10}11import "testing"12func TestSum(t *testing.T) {13	if Sum(10, 20) != 30 {14		t.Error("Sum(10, 20) should be 30")15	}16}17func TestSum2(t *testing.T) {18	if Sum(10, 20) != 30 {19		t.Error("Sum(10, 20) should be 30")20	}21}22import "testing"23func TestSum(t *testing.T) {24	if Sum(10, 20) != 30 {25		t.Error("Sum(10, 20) should be 30")26	}27}28func TestSum2(t *testing.T) {29	if Sum(10, 20) != 30 {30		t.Error("Sum(10, 20) should be 30")31	}32}33func TestSum3(t *testing.T) {34	if Sum(10, 20) != 30 {35		t.Error("Sum(10, 20) should be 30")36	}37}38import "testing"39func TestSum(t *testing.T) {40	if Sum(10, 20) != 30 {41		t.Error("Sum(10, 20) should be 30")42	}43}44func TestSum2(t *testing.T) {45	if Sum(10, 20) != 30 {46		t.Error("Sum(10, 20) should be