Move transport to network/transport
This commit is contained in:
		| @@ -1,138 +0,0 @@ | ||||
| package http | ||||
|  | ||||
| import ( | ||||
| 	"sync" | ||||
| 	"testing" | ||||
|  | ||||
| 	"github.com/micro/go-micro/transport" | ||||
| ) | ||||
|  | ||||
| func call(b *testing.B, c int) { | ||||
| 	b.StopTimer() | ||||
|  | ||||
| 	tr := NewTransport() | ||||
|  | ||||
| 	// server listen | ||||
| 	l, err := tr.Listen("localhost:0") | ||||
| 	if err != nil { | ||||
| 		b.Fatal(err) | ||||
| 	} | ||||
| 	defer l.Close() | ||||
|  | ||||
| 	// socket func | ||||
| 	fn := func(sock transport.Socket) { | ||||
| 		defer sock.Close() | ||||
|  | ||||
| 		for { | ||||
| 			var m transport.Message | ||||
| 			if err := sock.Recv(&m); err != nil { | ||||
| 				return | ||||
| 			} | ||||
|  | ||||
| 			if err := sock.Send(&m); err != nil { | ||||
| 				return | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	done := make(chan bool) | ||||
|  | ||||
| 	// accept connections | ||||
| 	go func() { | ||||
| 		if err := l.Accept(fn); err != nil { | ||||
| 			select { | ||||
| 			case <-done: | ||||
| 			default: | ||||
| 				b.Fatalf("Unexpected accept err: %v", err) | ||||
| 			} | ||||
| 		} | ||||
| 	}() | ||||
|  | ||||
| 	m := transport.Message{ | ||||
| 		Header: map[string]string{ | ||||
| 			"Content-Type": "application/json", | ||||
| 		}, | ||||
| 		Body: []byte(`{"message": "Hello World"}`), | ||||
| 	} | ||||
|  | ||||
| 	// client connection | ||||
| 	client, err := tr.Dial(l.Addr()) | ||||
| 	if err != nil { | ||||
| 		b.Fatalf("Unexpected dial err: %v", err) | ||||
| 	} | ||||
|  | ||||
| 	send := func(c transport.Client) { | ||||
| 		// send message | ||||
| 		if err := c.Send(&m); err != nil { | ||||
| 			b.Fatalf("Unexpected send err: %v", err) | ||||
| 		} | ||||
|  | ||||
| 		var rm transport.Message | ||||
| 		// receive message | ||||
| 		if err := c.Recv(&rm); err != nil { | ||||
| 			b.Fatalf("Unexpected recv err: %v", err) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// warm | ||||
| 	for i := 0; i < 10; i++ { | ||||
| 		send(client) | ||||
| 	} | ||||
|  | ||||
| 	client.Close() | ||||
|  | ||||
| 	ch := make(chan int, c*4) | ||||
|  | ||||
| 	var wg sync.WaitGroup | ||||
| 	wg.Add(c) | ||||
|  | ||||
| 	for i := 0; i < c; i++ { | ||||
| 		go func() { | ||||
| 			cl, err := tr.Dial(l.Addr()) | ||||
| 			if err != nil { | ||||
| 				b.Fatalf("Unexpected dial err: %v", err) | ||||
| 			} | ||||
| 			defer cl.Close() | ||||
|  | ||||
| 			for range ch { | ||||
| 				send(cl) | ||||
| 			} | ||||
|  | ||||
| 			wg.Done() | ||||
| 		}() | ||||
| 	} | ||||
|  | ||||
| 	b.StartTimer() | ||||
|  | ||||
| 	for i := 0; i < b.N; i++ { | ||||
| 		ch <- i | ||||
| 	} | ||||
|  | ||||
| 	b.StopTimer() | ||||
| 	close(ch) | ||||
|  | ||||
| 	wg.Wait() | ||||
|  | ||||
| 	// finish | ||||
| 	close(done) | ||||
| } | ||||
|  | ||||
| func BenchmarkTransport1(b *testing.B) { | ||||
| 	call(b, 1) | ||||
| } | ||||
|  | ||||
| func BenchmarkTransport8(b *testing.B) { | ||||
| 	call(b, 8) | ||||
| } | ||||
|  | ||||
| func BenchmarkTransport16(b *testing.B) { | ||||
| 	call(b, 16) | ||||
| } | ||||
|  | ||||
| func BenchmarkTransport64(b *testing.B) { | ||||
| 	call(b, 64) | ||||
| } | ||||
|  | ||||
| func BenchmarkTransport128(b *testing.B) { | ||||
| 	call(b, 128) | ||||
| } | ||||
		Reference in New Issue
	
	Block a user