add RealIP to Request interface

pkg/filters/proxies/loadbalance.go

@@ -275,18 +275,9 @@ func (lbp *WeightedRandomLoadBalancePolicy) ChooseServer(req protocols.Request,
 type IPHashLoadBalancePolicy struct {
 }
 
-type realIPer interface {
- RealIP() string
-}
-
 // ChooseServer chooses a server by ip hash.
 func (lbp *IPHashLoadBalancePolicy) ChooseServer(req protocols.Request, sg *ServerGroup) *Server {
- ri, ok := req.(realIPer)
- if !ok {
- panic("IPHashLoadBalancePolicy only support request with RealIP()")
- }
-
- ip := ri.RealIP()
+ ip := req.RealIP()
  hash := fnv.New32()
  hash.Write([]byte(ip))
  return sg.Servers[hash.Sum32()%uint32(len(sg.Servers))]

pkg/filters/proxies/requestmatch.go

@@ -114,12 +114,7 @@ type ipHashMatcher struct {
 
 // Match implements protocols.Matcher.
 func (iphm ipHashMatcher) Match(req protocols.Request) bool {
- ri, ok := req.(realIPer)
- if !ok {
- panic("IPHashLoadBalancePolicy only support request with RealIP()")
- }
-
- ip := ri.RealIP()
+ ip := req.RealIP()
  hash := fnv.New32()
  hash.Write([]byte(ip))
  return hash.Sum32()%1000 < iphm.permill

pkg/protocols/mqttprot/request.go

@@ -140,6 +140,11 @@ func (r *Request) Header() protocols.Header {
  return nil
 }
 
+// RealIP returns the real IP of the request.
+func (r *Request) RealIP() string {
+ panic("not implemented")
+}
+
 // SetPayload set the payload of the request to payload.
 func (r *Request) SetPayload(payload interface{}) {
  p, ok := payload.([]byte)

pkg/protocols/protocols.go

@@ -39,6 +39,9 @@ type Request interface {
  // Header returns the header of the request.
  Header() Header
 
+ // RealIP returns the real IP of the request.
+ RealIP() string
+
  // IsStream returns whether the payload is a stream, which cannot be
  // read for more than once.
  IsStream() bool