67 lines
1.0 KiB
Go
67 lines
1.0 KiB
Go
package rpc
|
|
|
|
import (
|
|
"jiacrontab/pkg/proto"
|
|
"log"
|
|
"net/http"
|
|
_ "net/http/pprof"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
type Logic struct {
|
|
}
|
|
|
|
func (l *Logic) Ping(args *proto.EmptyArgs, reply *proto.EmptyReply) error {
|
|
return nil
|
|
}
|
|
|
|
func (p *Logic) Say(args string, reply *string) error {
|
|
|
|
*reply = "hello boy"
|
|
time.Sleep(100 * time.Second)
|
|
return nil
|
|
}
|
|
func TestCall(t *testing.T) {
|
|
done := make(chan struct{})
|
|
go func() {
|
|
|
|
done <- struct{}{}
|
|
|
|
log.Println("start server")
|
|
err := listen(":6478", &Logic{})
|
|
if err != nil {
|
|
t.Fatal("server error:", err)
|
|
}
|
|
}()
|
|
<-done
|
|
time.Sleep(5 * time.Second)
|
|
// 等待server启动
|
|
var wg sync.WaitGroup
|
|
for i := 0; i < 100; i++ {
|
|
wg.Add(1)
|
|
go func(i int) {
|
|
|
|
defer wg.Done()
|
|
var ret string
|
|
// var args string
|
|
err := Call(":6478", "Logic.Say", "", &ret)
|
|
if err != nil {
|
|
log.Println(i, "error:", err)
|
|
}
|
|
t.Log(i, ret)
|
|
}(i)
|
|
|
|
}
|
|
|
|
go func() {
|
|
t.Log("listen :6060")
|
|
t.Log(http.ListenAndServe(":6060", nil))
|
|
}()
|
|
|
|
wg.Wait()
|
|
log.Println("end")
|
|
time.Sleep(2 * time.Minute)
|
|
}
|