如何获取真实IP？

[leiwingqueen]

我们有一个服务在gin框架上跑，依赖gin框架的context，这个方法如何改为使用kratos框架的上下文实现

func Get_client_ip(c *gin.Context) string {
	xForwardedFor := c.GetHeader("X-FORWARDED-FOR")
	ip := strings.TrimSpace(strings.Split(xForwardedFor, ",")[0])
	if ip != "" {
		return ip
	}

	ip = strings.TrimSpace(c.GetHeader("X-Real-Ip"))
	if ip != "" {
		return ip
	}

	if ip, _, err := net.SplitHostPort(strings.TrimSpace(c.Request.RemoteAddr)); err == nil {
		return ip
	}

	return ""
}

[tureo]

是想获取客户端ip写入ctx传递吗？可以参考我自己写的获取requestid并传递的方式：

package requestid

import (
	"context"

	"github.com/go-kratos/kratos/v2/log"
	"github.com/go-kratos/kratos/v2/metadata"
	"github.com/google/uuid"
)

const reqeustIdKey = "requestid"

// 获取请求id，从context获取，如果不存在就生成新的
func RequestId() log.Valuer {
	return func(ctx context.Context) interface{} {
		return GetOrGenerateRequestId(ctx)
	}
}

func GetMdWithRequestId(ctx context.Context) metadata.Metadata {
	var reqeustId string
	md, ok := metadata.FromServerContext(ctx)
	if ok {
		reqeustId = md.Get(reqeustIdKey)
	} else {
		md = metadata.New()
	}
	if reqeustId == "" {
		reqeustId = generateRequestId()
		md.Set(reqeustIdKey, reqeustId)
	}
	return md
}

func GetOrGenerateRequestId(ctx context.Context) string {
	var reqeustId string
	md, ok := metadata.FromServerContext(ctx)
	if ok {
		reqeustId = md.Get(reqeustIdKey)
	} else {
		md = metadata.New()
	}
	if reqeustId == "" {
		reqeustId = generateRequestId()
		md.Set(reqeustIdKey, reqeustId)
	}
	return reqeustId
}

// 生成requestId
func generateRequestId() string {
	return uuid.NewString()
}

// 在context中添加requestId
func ContextWithRequestId(ctx context.Context) context.Context {
	md := GetMdWithRequestId(ctx)
	ctx = metadata.NewServerContext(ctx, md)
	return ctx
}

[leiwingqueen]

thx,bro

[sxlvalue]

Add the following method to the custom service in your project's service package:

// GetIp Obtain the IP address requested by the client
func (s *AuthVerifyService) GetIp(ctx context.Context) string {
	request, isOk := http.RequestFromServerContext(ctx)
	if isOk {
		xForwardedFor := request.Header.Get("X-FORWARDED-FOR")
		ip := strings.TrimSpace(strings.Split(xForwardedFor, ",")[0])
		if ip != "" {
			return ip
		}

		ip = strings.TrimSpace(request.Header.Get("X-Real-Ip"))
		if ip != "" {
			return ip
		}

		if ip, _, err := net.SplitHostPort(strings.TrimSpace(request.RemoteAddr)); err == nil {
			return ip
		}
	}
	return ""
}