基于Go语言实现的简易api网关的示例代码
更新时间:2021年12月8日 12:58 点击:340 作者:董洪臣
浏览器的请求去请求目标地址,然后获得结果它再发送给浏览器。对于Go语言来说,实现转发只需要简单的一行代码即可实现,如下所示:
httputil.NewSingleHostReverseProxy(address)
基于此功能,进行简单包装,实现从远端admin管理中心获取需要转发的路由信息或者可以从本地配置文件中获取,实现动态转发。后续可以根据业务情况,可以实现如下功能:
开发接口,实现动态添加代理规则,进行转发
- 过滤不合法的接口
- 接口限流
- 统一日志记录
- …
代码如下:
package main import ( "encoding/json" "flag" "fmt" "github.com/gin-gonic/gin" "io" "io/ioutil" "log" "net/http" "net/http/httputil" "net/url" "os" "strings" ) type Respond struct { Success bool Status string Data []Proxy } type Proxy struct { Remark string //描述 Prefix string //转发的前缀判断 Upstream string //后端 nginx 地址或者ip地址 RewritePrefix string //重写 } var ( InfoLog *log.Logger ErrorLog *log.Logger proxyMap = make(map[string]Proxy) ) var adminUrl = flag.String("adminUrl", "", "admin的地址") var profile = flag.String("profile", "", "环境") var proxyFile = flag.String("proxyFile", "", "测试环境的数据") //日志初始化 func initLog() { errFile, err := os.OpenFile("errors.log", os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666) infoFile, err := os.OpenFile("info.log", os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666) if err != nil { log.Fatalln("打开日志文件失败:", err) } InfoLog = log.New(io.MultiWriter(os.Stderr, infoFile), "Info:", log.LstdFlags|log.Lmicroseconds|log.Lshortfile) ErrorLog = log.New(io.MultiWriter(os.Stderr, errFile), "Error:", log.LstdFlags|log.Lmicroseconds|log.Lshortfile) } func main() { router := gin.Default() //创建一个router flag.Parse() initLog() if *profile != "" { InfoLog.Printf("加载远端数据: %s ", *adminUrl) initProxyList() } else { InfoLog.Printf("加载本地配置数据: %s", *proxyFile) loadProxyListFromFile() } router.Any("/*action", Forward) //所有请求都会经过Forward函数转发 router.Run(":8000") } func initProxyList() { resp, _ := http.Get(*adminUrl) if resp != nil && resp.StatusCode == 200 { bytes, err := ioutil.ReadAll(resp.Body) defer resp.Body.Close() if err != nil { fmt.Println("ioutil.ReadAll err=", err) return } var respond Respond err = json.Unmarshal(bytes, &respond) if err != nil { fmt.Println("json.Unmarshal err=", err) return } proxyList := respond.Data for _, proxy := range proxyList { //追加 反斜杠,为了动态匹配的时候 防止 /proxy/test /proxy/test1 无法正确转发 proxyMap[proxy.Prefix+"/"] = proxy } } } func Forward(c *gin.Context) { HostReverseProxy(c.Writer, c.Request) } func HostReverseProxy(w http.ResponseWriter, r *http.Request) { if r.RequestURI == "/favicon.ico" { io.WriteString(w, "Request path Error") return } //从内存里面获取转发的url var upstream = "" if value, ok := proxyMap[r.RequestURI]; ok { //如果转发的地址是 / 开头的,需要去掉 if strings.HasSuffix(value.Upstream, "/") { upstream += strings.TrimRight(value.Upstream, "/") } else { upstream += value.Upstream } //如果首位不是/开头,则需要追加 if !strings.HasPrefix(value.RewritePrefix, "/") { upstream += "/" + value.RewritePrefix } else { upstream += value.RewritePrefix } //去掉开头 r.URL.Path = strings.ReplaceAll(r.URL.Path, r.RequestURI, "") } // parse the url remote, err := url.Parse(upstream) InfoLog.Printf("RequestURI %s upstream %s remote %s", r.RequestURI, upstream, remote) if err != nil { panic(err) } r.URL.Host = remote.Host r.URL.Scheme = remote.Scheme r.Header.Set("X-Forwarded-Host", r.Header.Get("Host")) r.Host = remote.Host httputil.NewSingleHostReverseProxy(remote).ServeHTTP(w, r) } func loadProxyListFromFile() { file, err := os.Open(*proxyFile) if err != nil { ErrorLog.Println("err:", err) } var respond Respond // 创建json解码器 decoder := json.NewDecoder(file) err = decoder.Decode(&respond) if err != nil { fmt.Println("LoadProxyListFromFile failed", err.Error()) } proxyList := respond.Data for _, proxy := range proxyList { proxyMap[proxy.Prefix+"/"] = proxy } }
proxy_data.json 格式如下:
{ "success":true, "status": "ok", "data": [ { "remark": "测试环境", "prefix": "/division", "upstream": "http://test.xxxxx.cn/", "rewritePrefix": "/api/division" }, { "remark": "测试环境1", "prefix": "/division1", "upstream": "http://test.xxxx.cn/", "rewritePrefix": "" }, { "remark": "测试环境2", "prefix": "/division3", "upstream": "http://test.xxxxxx.cn/", "rewritePrefix": "/api/division" } ] }
启动脚本
## 加载本地配置文件数据 go run proxy_agent.go -proxyFile ./proxy_data.json ## 启动从配置中心获取数据 go run proxy_agent.go -profile prod -adminUrl http://localhost:3000/proxy/findAll
到此这篇关于基于Go语言实现的简易api网关的示例代码的文章就介绍到这了,更多相关Go api网关 内容请搜索猪先飞以前的文章或继续浏览下面的相关文章希望大家以后多多支持猪先飞!
原文出处:https://blog.csdn.net/dong945221578/article/details/11067948
上一篇: Golang打包配置文件的实现示例
下一篇: 几个小技巧帮你实现Golang永久阻塞
相关文章
- 在程序员中,尤其是go新手,经常听到的一个讨论话题是:如何处理错误,这篇文章主要给大家介绍了关于Go应用中优雅处理Error的一些相关技巧,需要的朋友可以参考下...2021-09-08
Django def clean()函数对表单中的数据进行验证操作
这篇文章主要介绍了Django def clean()函数对表单中的数据进行验证操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-07-09- 这篇文章主要介绍了golang官方嵌入文件到可执行程序,本文通过示例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下...2021-02-20
- 这篇文章主要介绍了go浮点数转字符串保留小数点后N位解决办法,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下...2020-05-11
- 这篇文章主要介绍了Go语言使用读写OPC详解,图文讲解的很清晰,有感兴趣的同学可以学习下...2021-03-05
- 这篇文章主要介绍了Go项目的目录结构,对基础目录做了讲解,对项目开发中的其它目录也一并做了介绍,需要的朋友可以参考下...2020-05-01
- string与[]byte经常需要互相转化,普通转化会发生底层数据的复制,下面这篇文章主要给大家介绍了关于Go中string与[]byte高效互转的相关资料,文中通过示例代码介绍的非常详细,需要的朋友可以参考下...2021-09-20
- Go 语言提供的基础容器,免不了要查询容器中的数据,那么是如何实现遍历的呢?本文将会介绍几种常用容易的遍历及其使用。感兴趣的可以了解一下...2021-06-13
- 这篇文章主要介绍了创建第一个Go语言程序Hello,Go!本文详细的给出项目创建、代码编写的过程,同时讲解了GOPATH、Go install等内容,需要的朋友可以参考下...2020-05-01
- 这篇文章主要介绍了在Django中使用MQTT的方法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2021-05-10
- 这篇文章主要介绍了go语言中的Carbon库时间处理,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下...2021-02-05
- 这篇文章主要介绍了go嵌套匿名结构体的初始化详解,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-12-16
解决导入django_filters不成功问题No module named 'django_filter'
这篇文章主要介绍了解决导入django_filters不成功问题No module named 'django_filter',具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-07-15- 本文主要介绍了Django项目连接MongoDB的三种方法,文中通过示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下...2021-09-27
详解如何使用Docker部署Django+MySQL8开发环境
这篇文章主要介绍了详解如何使用Docker部署Django+MySQL8开发环境,文中通过示例代码以及图文介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面来一起学习学习吧...2020-07-19- 这篇文章主要介绍了Go语言json编码驼峰转下划线、下划线转驼峰的实现,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2020-06-09
- 这篇文章主要介绍了Django 解决由save方法引发的错误,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-05-21
- 这篇文章主要介绍了django前端页面下拉选择框默认值设置方式,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-08-10
- 这篇文章主要介绍了django数据模型中null和blank的区别说明,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-09-03
- 这篇文章主要介绍了Go 自定义package包设置与导入操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2021-05-06