go语言INI配置文件格式，配置反向代理路径映射

最新推荐文章于 2024-02-21 16:32:51 发布

加油呀~

最新推荐文章于 2024-02-21 16:32:51 发布

阅读量489

点赞数

分类专栏： go 文章标签： golang

本文链接：https://blog.csdn.net/weixin_51530404/article/details/129397016

版权

go 专栏收录该内容

10 篇文章 0 订阅

订阅专栏

💌💌💌这篇文章我们会学习go语言设计INI配置文件格式，配置反向代理路径映射，会用到go-ini库。go-ini是 Go 语言中用于操作 ini 文件的第三方库。

首先了解并安装go-ini，这是INI的官方网站： https://ini.unknwon.io/

下载安装：go get github.com/go-ini/ini—— 最低要求安装 Go 语言版本为 1.6。

接下来使用INI配置文件格式，配置反向代理路径映射

一、创建ini文件，以及配置的go文件

这里的ini文件不用非要以.ini结尾，我创建的两个文件名（env，configs.go）。文件目录如下：
在这里插入图片描述

二、编辑 .ini 文件

我们要通过.ini来配置反向代理路径映射——/a,对应路径http://localhost:9091、/b对应的路径http://localhost:9092。所以这里，对应的ini文件（env）的内容就是：

[proxy]

[proxy.a]
path=/a
pass=http://localhost:9091

[proxy.b]
path=/b
pass=http://localhost:9092

这里要介绍.ini文件中的几个概念：

分区/父子分区/键/值
对应上面的env，proxy就是一个分区，a、b也是分区。proxy是a、b的父分区，a、b是proxy的子分区。path，pass是键，“/a”，“http://localhost:9091”，“/b”，“http://localhost:9092”是值。其中path键的值为“/a”。

三、编写 go 文件来操作ini配置文件

可以看下面的代码，configs.go

package util

import (
	"fmt"
	"github.com/go-ini/ini"
	"log"
	"os"
)

var ProxyConfigs map[string]string

type EnvConfig *os.File

func init() {
	ProxyConfigs = make(map[string]string)
	//读取env文件
	EnvConfig, err := ini.Load("env")
	if err != nil {
		log.Println(err)
		return
	}
	//获取固定的proxy
	proxy, _ := EnvConfig.GetSection("proxy") //假设是固定分区，必须得有
	if proxy != nil {
		//通过proxy获取子分区
		secs := proxy.ChildSections()
		for _, sec := range secs {
			path, _ := sec.GetKey("path") //固定key
			pass, _ := sec.GetKey("pass") //国定key
			if path != nil && pass != nil {
				ProxyConfigs[path.Value()] = pass.Value()
			}
		}

	}
	//因为myproxy引用了util包，所以会执行里面的函数
	fmt.Println(ProxyConfigs)
}

这部分代码的只要功能就是读取env文件的内容，将其以map的格式存到ProxyConfigs中，我是在另一个文件中配置路径映射的，代码如下：

//下面使用ini配置路径映射
	for k, v := range util.ProxyConfigs {
		if matched, _ := regexp.MatchString(k, r.URL.Path); matched == true {
			RequestUrl(w, r, v)
			return
		}
	}
	
func RequestUrl(w http.ResponseWriter, r *http.Request, url string) {
	//newr是一个请求
	newreq, _ := http.NewRequest(r.Method, url, r.Body)
	//把浏览器的原始头拷贝到新请求newrequest
	util.CloneHeader(r.Header, &newreq.Header)

	newreq.Header.Set("x-forwarded-for", r.RemoteAddr)

	newresponse, _ := http.DefaultClient.Do(newreq)

	getHead := w.Header()
	//得到头，并拷贝给客户端
	util.CloneHeader(newresponse.Header, &getHead)
	//将响应的状态写入头中
	w.WriteHeader(newresponse.StatusCode)

	defer newresponse.Body.Close()
	//readall返回bytes
	res_cont, _ := ioutil.ReadAll(newresponse.Body)
	w.Write(res_cont) //写入响应给客户端
	return
}