fasthttp LBClient 客户端的使用

fasthttp 不只是一个http server 的包，同时也提供了http client 的功能，以下是关于LBClient
的简单使用，以及一些说明

参考代码

main.go

 
package main

​

import (

    "flag"

    "fmt"

    "log"

    "strings"

​

    _ "net/http/pprof"

​

    "github.com/valyala/fasthttp"

    "github.com/valyala/fasthttp/pprofhandler"

)

​

var (

    add = flag.String("add", "127.0.0.1:8080  127.0.0.1:80", "proxy address")

)

​

type myLog struct {

    lbc *fasthttp.LBClient

}

​

func (l *myLog) Printf(format string, args ...interface{}) {

    log.Printf(format, args...)

}

​

func newLog(add string) *myLog {

    var lbc fasthttp.LBClient

    servers := strings.Split(add, " ")

    for _, addr := range servers {

        if len(addr) > 0 {

            c := &fasthttp.HostClient{

                Addr: addr,

            }

            lbc.Clients = append(lbc.Clients, c)

        }

    }

    lbc.HealthCheck = func(req *fasthttp.Request, resp *fasthttp.Response, err error) bool {

        fmt.Println(string(resp.Header.Header()))

        if err != nil {

            fmt.Println("err info:", err.Error())

            return false

        }

        return true

    }

    return &myLog{

        lbc: &lbc,

    }

}

​

// load blance client

func (l *myLog) lbHandler(ctx *fasthttp.RequestCtx) {

    if strings.Contains(string(ctx.Path()), "/debug/pprof") {

        pprofhandler.PprofHandler(ctx)

        return

    }

    l.lbc.Do(&ctx.Request, &ctx.Response)

}

​

func main() {

    flag.Parse()

    fmt.Println(*add)

    mylog := newLog(*add)

    s := fasthttp.Server{

        Handler:      mylog.lbHandler,

        LogAllErrors: true,

        Logger:       mylog,

    }

    s.ListenAndServe(":8000")

}