proxy.go

package main

import (
    "net"
    "time"
)

// 初始化代理服务
func initProxy() {
    Log.Infof("Proxying %s -> %s\n", Config.Bind, Config.Backend)
    server, err := net.Listen("tcp", Config.Bind)
    if err != nil {
        Log.Fatal(err)
    }
    // 等待的队列长度
    waitQueue := make(chan net.Conn, Config.WaitQueueLen)
    // 最大并发连接
    connPools := make(chan bool, Config.MaxConn)
    for i := 0; i < Config.MaxConn; i++ {
        connPools <- true
    }
    // 等待连接处理
    go waitConn(waitQueue, connPools)
    // 接收连接并抛给管道处理
    for {
        conn, err := server.Accept()
        if err != nil {
            Log.Error(err)
            continue
        }
        Log.Infof("Received connection from %s.\n", conn.RemoteAddr())
        waitQueue <- conn
    }
}

// 连接数控制
func waitConn(waitQueue chan net.Conn, connPools chan bool) {
    for conn := range waitQueue {
        // 接收一个链接,连接池释放一个
        <-connPools
        go func(conn net.Conn) {
            handleConn(conn)
            // 链接处理完毕,增加
            connPools <- true
            Log.Infof("Closed connection from %s.\n", conn.RemoteAddr())
        }(conn)
    }
}

// 处理连接
func handleConn(conn net.Conn) {
    defer conn.Close()
    // 根据链接哈希选择机器
    proxySvr, ok := getBackendSvr(conn)
    if !ok {
        return
    }
    // 链接远程代理服务器
    remote, err := net.Dial("tcp", proxySvr.identify)
    if err != nil {
        Log.Error(err)
        proxySvr.failTimes++
        return
    }
    // 等待双向连接完成
    complete := make(chan bool, 2)
    oneSwitch := make(chan bool, 1)
    otherSwitch := make(chan bool, 1)
    // 将当前客户端链接发送的数据发送给远程被代理的服务器
    go transaction(conn, remote, complete, oneSwitch, otherSwitch)
    // 将远程服务返回的数据返回给客户端
    go transaction(remote, conn, complete, otherSwitch, oneSwitch)
    <-complete
    <-complete
    remote.Close()
}

// 数据交换传输(从from读数据,再写入to)
func transaction(from, to net.Conn, complete, oneSwitch, otherSwitch chan bool) {
    var err error
    var read int
    bytes := make([]byte, 1024)
    for {
        select {
        case <-otherSwitch:
            complete <- true
            return
        default:
            timeOutSec := time.Duration(Config.Timeout) * time.Second
            // 设置超时时间
            from.SetReadDeadline(time.Now().Add(timeOutSec))
            read, err = from.Read(bytes)
            if err != nil {
                complete <- true
                oneSwitch <- true
                return
            }
            // 设置超时时间
            to.SetWriteDeadline(time.Now().Add(timeOutSec))
            _, err = to.Write(bytes[:read])
            if err != nil {
                complete <- true
                oneSwitch <- true
                return
            }
        }
    }
}
				
时间: 2024-11-03 21:53:04

proxy.go的相关文章

Ubuntu Linux下通过代理(proxy)使用git上github.com

github.com,作为程序员的代码仓库,我们经常会用到.但有时候我们不能直接通过网络链接它,只能通过代理. 这里我有一台代理服务器,起初我以为在终端设置了代理环境就行了,其设置为在你的~/.bashrc里增加以下几行: export http_proxy="http://proxy-server:3128/" export https_proxy="http://proxy-server:3128/" export ftp_proxy="http://

利用tinyproxy在Linux上搭建HTTP Proxy Server

之所以需要用到HTTP Proxy Server并不是为了要翻墙,而是为了让没有公网IP地址的内网主机通过有公网IP地址的外网主机访问Internet.举个例子,阿里云ECS在购买时可以不购买公网IP地址,但这种没有公网IP地址的ECS云主机(实例)是没有访问Internet的能力的,也就是说无法在这台实例上下载文件,这在部署应用如部署MySQL时可能遇到无法完成安装问题.解决的办法有两种,一种是在另一台具有公网访问能力的ECS实例上搭建VPN服务,另一种是在另一台具有公网访问能力的ECS实例上

Forward Proxy &amp; Reverse Proxy | 正向代理 和 反向代理

对请求和响应内容不做修改的转发的服务器,被称为代理服务器.代理服务器分为两种类型:正向代理 和 反向代理. 正向代理:面向互联网,从更广范围获取信息的代理. 反向代理:面向内部,一般用于某企业的网站的前端的代理.反向代理能承担负载均衡,身份认证,内容缓存的任务.这些功能在反向代理上面实现会显得很自然. 正向代理: 如果使用过 vpn 或者 shadowsocks 等FQ工具访问 Google,那么就是在使用正向代理服务器. 下面的图例解释了正向代理的使用.正向代理服务器在互联网中扮演用户的角色,

代理模式(Proxy)

代理模式的作用是:为其他对象提供一种代理以控制对这个对象的访问. 在某些情况下,一个客户不想或者不能直接引用另一个对象,而代理对象可以在客户端和目标对象之间起到中介的作用. 代理模式一般涉及到的角色有 抽象角色:声明真实对象和代理对象的共工接口 代理角色:代理对象角色内部含有对真实对象的引用,从而可以操作真实对象,同时代理对象提供与真实对象相同的接口以便在任何时刻都能代替真实对象.同时,代理对象可以在执行真实对象操作时,附加其他的操作,相当于对真实对象进行封装. 真实角色:代理角色所代表的真实对

Net Core下通过Proxy 模式

Net Core下通过Proxy 模式 NET Core下的WCF客户端也是开源的,这次发布.NET Core 2.0,同时也发布了 WCF for .NET Core 2.0.0, 本文介绍在.NET Core下如何通过Proxy 消费WCF服务. 我们现在直接可以在 standard 2.0下调用wcf服务了,不过 Microsoft WCF Web Service Reference Provider 目前是beta阶段,要使用这个插件,需要安装一个Visual Studio插件,下载地址

两条proxy switchy的配置规则

原文发表于:2010-12-07 转载至cu于:2012-07-21 今天下午使用chrome的扩展程序proxy switchy做代理,配置切换规则时写了三条规则如下:*twitter.com/**youtube.com/**facebook.com/*使用时,twitter正常访问,有youtube和facebook的站点能访问,但页面不能正常显示.接下来当然是google了,还真有强人遇到这样的问题并解决了.造成上面所描述问题的原因是proxy switchy的配置规则有问题,youtub

Android源代码之DeskClock (三) Proxy/Delegate Application 框架应用

一.概述 当项目有加壳子,插件化或热修复等需求的时候,能够使用Proxy/Delegate Application框架的方式,在正常的模式中,一个程序一般仅仅有一个Application入口,而Proxy/Delegate模式中须要有两个Application,原程序的Application改为Delegate Application,再新加一个Proxy Application,由Proxy Application 提供一系列的个性化定制,再将所有的context和context相关的引用所有

使用mysql proxy对数据库进行读写分离

服务器安排如下: 192.168.100.128 主 192.168.100.129 从 192.168.100.130 mysql-proxy 1.在100.130中下载安装mysql-proxy tar -zxvf mysql-proxy-0.8.5-linux-el6-x86-64bit.tar.gz cp mysql-proxy-0.8.5-linux-el6-x86-64bit /usr/local/mysql-proxy 2.配置环境变量 LUA_PATH="/usr/local/m

Charles Proxy代理使用简要说明

1.去官网下载免费试用版: http://www.charlesproxy.com/ (需要机器有Java运行时)或下载破解注册版:http://charles.iiilab.com/,安装后开启默认代理端口为 本机IP:8888 , 比如: 192.168.43.240:8888 2.手机端http代理设置 iPhone代理设置: 设置 -> 无线局域网 -> 当前连接的wifi 最右侧详细信息按钮 -> 最下面的HTTP代理 -> 手动 -> 填入IP,端口 Androi

proxy switchysharp

Proxy SwitchySharp是Google Chrome浏览器上的一个代理管理扩展程序,目前最好用的代理程序.Proxy SwitchySharp插件是针对谷歌浏览器而开发的一款代理设置插件,基于"Proxy Switchy!"和 "SwitchyPlus"开发,能够帮助用户轻松快捷的管理和切换多个代理设置,用户可以自定义切换规则,自由畅游互联网的海洋. Proxy SwitchySharp软件截图 这款插件界面简洁大方,使用操作简单,支持直接连接.自动切换