go语言命令行工具 go语言命令行参数

go语言实现一个简单的简单网关网关=反向代理+负载均衡+各种策略,技术实现也有多种多样,有基于 nginx 使用 lua 的实现,比如 openresty、kong;也有基于 zuul 的通用网关;还有就是 golang 的网关,比如 tyk 。
这篇文章主要是讲如何基于 golang 实现一个简单的网关 。
转自: troy.wang/docs/golang/posts/golang-gateway/
整理:go语言钟文文档:
启动两个后端 web 服务(代码)
这里使用命令行工具进行测试
具体代码
直接使用基础库 httputil 提供的NewSingleHostReverseProxy即可,返回的reverseProxy对象实现了serveHttp方法,因此可以直接作为 handler 。
具体代码
director中定义回调函数,入参为*http.Request,决定如何构造向后端的请求 , 比如 host 是否向后传递,是否进行 url 重写,对于 header 的处理 , 后端 target 的选择等,都可以在这里完成 。
director在这里具体做了:
modifyResponse中定义回调函数,入参为*http.Response,用于修改响应的信息,比如响应的 Body,响应的 Header 等信息 。
最终依旧是返回一个ReverseProxy,然后将这个对象作为 handler 传入即可 。
参考 2.2 中的NewSingleHostReverseProxy,只需要实现一个类似的、支持多 targets 的方法即可,具体实现见后面 。
作为一个网关服务,在上面 2.3 的基础上,需要支持必要的负载均衡策略,比如:
随便 random 一个整数作为索引,然后取对应的地址即可,实现比较简单 。
具体代码
使用curIndex进行累加计数 , 一旦超过 rss 数组的长度,则重置 。
具体代码
轮询带权重,如果使用计数递减的方式 , 如果权重是5,1,1那么后端 rs 依次为a,a,a,a,a,b,c,a,a,a,a…,其中 a 后端会瞬间压力过大;参考 nginx 内部的加权轮询,或者应该称之为平滑加权轮询 , 思路是:
后端真实节点包含三个权重:
操作步骤:
具体代码
一致性 hash 算法,主要是用于分布式 cache 热点/命中问题;这里用于基于某 key 的 hash 值,路由到固定后端,但是只能是基本满足流量绑定,一旦后端目标节点故障,会自动平移到环上最近的那么个节点 。
实现:
具体代码
每一种不同的负载均衡算法,只需要实现添加以及获取的接口即可 。
然后使用工厂方法 , 根据传入的参数,决定使用哪种负载均衡策略 。
具体代码
作为网关,中间件必不可少,这类包括请求响应的模式,一般称作洋葱模式,每一层都是中间件 , 一层层进去,然后一层层出来 。
中间件的实现一般有两种,一种是使用数组,然后配合 index 计数;一种是链式调用 。
具体代码
写命令行应用程序什么不可或缺?Go可以这样处理命令行参数 Go语言内置的flag包实现了命令行参数的解析 , flag包使得开发命令行工具更为简单 。
如果你只是简单的想要获取命令行参数,可以像下面的代码示例一样使用os.Args来获取命令行参数 。
将上面的代码执行go build -o "args_demo"编译之后,执行:
os.Args是一个存储命令行参数的字符串切片,它的第一个元素是执行文件的名称 。
本文介绍了flag包的常用函数和基本用法,更详细的内容请查看官方文档 。
flag包支持的命令行参数类型有bool、int、int64、uint、uint64、float float64、string、duration 。
有以下两种常用的定义命令行flag参数的方法 。
基本格式如下:
flag.Type(flag名, 默认值, 帮助信息)*Type 例如我们要定义姓名、年龄、婚否三个命令行参数,我们可以按如下方式定义:

推荐阅读