主页 > 知识库 > golang 实现tcp转发代理的方法

golang 实现tcp转发代理的方法

热门标签:地图标注的汽车标 学海导航地图标注 西部云谷一期地图标注 江西转化率高的羿智云外呼系统 广州呼叫中心外呼系统 中国地图标注省会高清 高德地图标注口诀 浙江高速公路地图标注 南通如皋申请开通400电话

我就废话不多说了,直接上代码吧:

package main
 
import (
	"flag"
	"fmt"
	"io"
	"net"
	"os"
	"strings"
	"sync"
)
 
var lock sync.Mutex
var trueList []string
var ip string
var list string
 
func main() {
	flag.StringVar(ip, "l", ":9897", "-l=0.0.0.0:9897 指定服务监听的端口")
	flag.StringVar(list, "d", "127.0.0.1:1789,127.0.0.1:1788", "-d=127.0.0.1:1789,127.0.0.1:1788 指定后端的IP和端口,多个用','隔开")
	flag.Parse()
	trueList = strings.Split(list, ",")
	if len(trueList) = 0 {
		fmt.Println("后端IP和端口不能空,或者无效")
		os.Exit(1)
	}
	server()
}
 
func server() {
	lis, err := net.Listen("tcp", ip)
	if err != nil {
		fmt.Println(err)
		return
	}
	defer lis.Close()
	for {
		conn, err := lis.Accept()
		if err != nil {
			fmt.Println("建立连接错误:%v\n", err)
			continue
		}
		fmt.Println(conn.RemoteAddr(), conn.LocalAddr())
		go handle(conn)
	}
}
 
func handle(sconn net.Conn) {
	defer sconn.Close()
	ip, ok := getIP()
	if !ok {
		return
	}
	dconn, err := net.Dial("tcp", ip)
	if err != nil {
		fmt.Printf("连接%v失败:%v\n", ip, err)
		return
	}
	ExitChan := make(chan bool, 1)
	go func(sconn net.Conn, dconn net.Conn, Exit chan bool) {
		_, err := io.Copy(dconn, sconn)
		fmt.Printf("往%v发送数据失败:%v\n", ip, err)
		ExitChan - true
	}(sconn, dconn, ExitChan)
	go func(sconn net.Conn, dconn net.Conn, Exit chan bool) {
		_, err := io.Copy(sconn, dconn)
		fmt.Printf("从%v接收数据失败:%v\n", ip, err)
		ExitChan - true
	}(sconn, dconn, ExitChan)
	-ExitChan
	dconn.Close()
}
 
func getIP() (string, bool) {
	lock.Lock()
	defer lock.Unlock()
 
	if len(trueList)  1 {
		return "", false
	}
	ip := trueList[0]
	trueList = append(trueList[1:], ip)
	return ip, true
}

以上这篇golang 实现tcp转发代理的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。

您可能感兴趣的文章:
  • 基于go手动写个转发代理服务的代码实现
  • 利用Golang实现TCP连接的双向拷贝详解
  • golang之tcp自动重连实现方法

标签:许昌 保定 德宏 吐鲁番 贵州 东营 曲靖 常州

巨人网络通讯声明:本文标题《golang 实现tcp转发代理的方法》,本文关键词  golang,实现,tcp,转发,代理,;如发现本文内容存在版权问题,烦请提供相关信息告之我们,我们将及时沟通与处理。本站内容系统采集于网络,涉及言论、版权与本站无关。
  • 相关文章
  • 下面列出与本文章《golang 实现tcp转发代理的方法》相关的同类信息!
  • 本页收集关于golang 实现tcp转发代理的方法的相关信息资讯供网民参考!
  • 推荐文章