Skip to content

Instantly share code, notes, and snippets.

@Chen-tao

Chen-tao/go-rpc-bm-client.go

Forked from skoowoo/go-rpc-bm-client.go
Created April 12, 2018 12:00
Show Gist options
  • Save Chen-tao/dba2c744e38e63e74b471ba1f8287c51 to your computer and use it in GitHub Desktop.
Save Chen-tao/dba2c744e38e63e74b471ba1f8287c51 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. @skoo87 skoo87 revised this gist Sep 10, 2013. 2 changed files with 90 additions and 90 deletions.
    128 changes: 64 additions & 64 deletions go-rpc-bm-client.go
    View file
    Original file line number Diff line number Diff line change
    @@ -2,79 +2,79 @@ package main

    import (
    "flag"
    "fmt"
    "log"
    "net/rpc"
    "sync"
    "time"
    "fmt"
    "log"
    "net/rpc"
    "sync"
    "time"
    )

    var (
    Num int
    Conn int
    Addr string
    Num int
    Conn int
    Addr string
    )

    const RS = 100000

    func init() {
    flag.IntVar(&Num, "num", 1, "")
    flag.IntVar(&Conn, "conn", 1, "")
    flag.StringVar(&Addr, "addr", "127.0.0.1", "")
    flag.Parse()
    flag.IntVar(&Num, "num", 1, "")
    flag.IntVar(&Conn, "conn", 1, "")
    flag.StringVar(&Addr, "addr", "127.0.0.1", "")
    flag.Parse()
    }

    func main() {
    clients := make([]*rpc.Client, 0, 10)
    for i := 0; i < Conn; i++ {
    client, err := rpc.Dial("tcp", Addr+":8000")
    if err != nil {
    log.Fatalln(err)
    }
    clients = append(clients, client)
    }
    index := 0

    s1 := "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"

    var wg sync.WaitGroup

    start := time.Now()

    for i := 0; i < Num; i++ {
    wg.Add(1)

    var c *rpc.Client
    if index < len(clients) {
    c = clients[index]
    index++
    } else {
    index = 0
    c = clients[index]
    }

    go func(cli *rpc.Client) {
    defer wg.Done()

    log.Println("goroutine start...")
    st := time.Now()

    for n := 0; n < RS; n++ {
    if err := cli.Call("EchoServer.Echo", &s1, nil); err != nil {
    log.Println(err)
    }
    }

    log.Println(time.Now().Sub(st))
    }(c)
    }
    wg.Wait()

    total := RS * Num
    secs := time.Now().Sub(start) / 1000000000

    fmt.Printf("concurrency: %d\n", Num)
    fmt.Printf("total: %d\n", total)
    fmt.Printf("seconds: %d\n", secs)
    fmt.Printf("qps: %d\n", total/int(secs))
    clients := make([]*rpc.Client, 0, 10)
    for i := 0; i < Conn; i++ {
    client, err := rpc.Dial("tcp", Addr+":8000")
    if err != nil {
    log.Fatalln(err)
    }
    clients = append(clients, client)
    }
    index := 0

    s1 := "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"

    var wg sync.WaitGroup

    start := time.Now()

    for i := 0; i < Num; i++ {
    wg.Add(1)

    var c *rpc.Client
    if index < len(clients) {
    c = clients[index]
    index++
    } else {
    index = 0
    c = clients[index]
    }

    go func(cli *rpc.Client) {
    defer wg.Done()

    log.Println("goroutine start...")
    st := time.Now()

    for n := 0; n < RS; n++ {
    if err := cli.Call("EchoServer.Echo", &s1, nil); err != nil {
    log.Println(err)
    }
    }

    log.Println(time.Now().Sub(st))
    }(c)
    }
    wg.Wait()

    total := RS * Num
    secs := time.Now().Sub(start) / 1000000000

    fmt.Printf("concurrency: %d\n", Num)
    fmt.Printf("total: %d\n", total)
    fmt.Printf("seconds: %d\n", secs)
    fmt.Printf("qps: %d\n", total/int(secs))
    }
    52 changes: 26 additions & 26 deletions go-rpc-bm-server.go
    View file
    Original file line number Diff line number Diff line change
    @@ -2,40 +2,40 @@ package main

    import (
    "log"
    "net"
    "net/rpc"
    "net"
    "net/rpc"
    )

    type EchoServer bool

    func (s *EchoServer) Echo(req *string, res *string) error {
    return nil
    return nil
    }

    func main() {
    echo := new(EchoServer)
    if err := rpc.Register(echo); err != nil {
    log.Println(err)
    return
    }
    echo := new(EchoServer)
    if err := rpc.Register(echo); err != nil {
    log.Println(err)
    return
    }

    var listener *net.TCPListener
    if tcpAddr, err := net.ResolveTCPAddr("tcp", ":8000"); err != nil {
    log.Println(err)
    return
    } else {
    if listener, err = net.ListenTCP("tcp", tcpAddr); err != nil {
    log.Println(err)
    return
    }
    }
    var listener *net.TCPListener
    if tcpAddr, err := net.ResolveTCPAddr("tcp", ":8000"); err != nil {
    log.Println(err)
    return
    } else {
    if listener, err = net.ListenTCP("tcp", tcpAddr); err != nil {
    log.Println(err)
    return
    }
    }

    for {
    conn, err := listener.Accept()
    if err != nil {
    log.Println(err)
    continue
    }
    go rpc.ServeConn(conn)
    }
    for {
    conn, err := listener.Accept()
    if err != nil {
    log.Println(err)
    continue
    }
    go rpc.ServeConn(conn)
    }
    }
  2. @skoo87 skoo87 created this gist Sep 10, 2013.
    80 changes: 80 additions & 0 deletions go-rpc-bm-client.go
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,80 @@
    package main

    import (
    "flag"
    "fmt"
    "log"
    "net/rpc"
    "sync"
    "time"
    )

    var (
    Num int
    Conn int
    Addr string
    )

    const RS = 100000

    func init() {
    flag.IntVar(&Num, "num", 1, "")
    flag.IntVar(&Conn, "conn", 1, "")
    flag.StringVar(&Addr, "addr", "127.0.0.1", "")
    flag.Parse()
    }

    func main() {
    clients := make([]*rpc.Client, 0, 10)
    for i := 0; i < Conn; i++ {
    client, err := rpc.Dial("tcp", Addr+":8000")
    if err != nil {
    log.Fatalln(err)
    }
    clients = append(clients, client)
    }
    index := 0

    s1 := "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"

    var wg sync.WaitGroup

    start := time.Now()

    for i := 0; i < Num; i++ {
    wg.Add(1)

    var c *rpc.Client
    if index < len(clients) {
    c = clients[index]
    index++
    } else {
    index = 0
    c = clients[index]
    }

    go func(cli *rpc.Client) {
    defer wg.Done()

    log.Println("goroutine start...")
    st := time.Now()

    for n := 0; n < RS; n++ {
    if err := cli.Call("EchoServer.Echo", &s1, nil); err != nil {
    log.Println(err)
    }
    }

    log.Println(time.Now().Sub(st))
    }(c)
    }
    wg.Wait()

    total := RS * Num
    secs := time.Now().Sub(start) / 1000000000

    fmt.Printf("concurrency: %d\n", Num)
    fmt.Printf("total: %d\n", total)
    fmt.Printf("seconds: %d\n", secs)
    fmt.Printf("qps: %d\n", total/int(secs))
    }
    41 changes: 41 additions & 0 deletions go-rpc-bm-server.go
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,41 @@
    package main

    import (
    "log"
    "net"
    "net/rpc"
    )

    type EchoServer bool

    func (s *EchoServer) Echo(req *string, res *string) error {
    return nil
    }

    func main() {
    echo := new(EchoServer)
    if err := rpc.Register(echo); err != nil {
    log.Println(err)
    return
    }

    var listener *net.TCPListener
    if tcpAddr, err := net.ResolveTCPAddr("tcp", ":8000"); err != nil {
    log.Println(err)
    return
    } else {
    if listener, err = net.ListenTCP("tcp", tcpAddr); err != nil {
    log.Println(err)
    return
    }
    }

    for {
    conn, err := listener.Accept()
    if err != nil {
    log.Println(err)
    continue
    }
    go rpc.ServeConn(conn)
    }
    }