Golang TcpProxy和Nodejs TcpProxy

自己平时的工作基本都在php和nodejs之间徘徊,但是目前面对python和java的猛烈攻击呢,其实内心有一种隐隐的痛"PHP是世界上最好的语言","nodejs在cpu密集时服务彻底瘫痪"。。。

看了半个月python真实发现,其实它太像php语言了,所以基本不用怎么理解就会了。golang看了1个多月了真的得多写多看源代码才能收获,别看才30几个关键字但是内容真的很多,golang的性能是真的高可以大大缩减服务器开销,举个例子web服务中php需要100台机器,那么golang可能只需要10台甚至更少!

最近在研究mysql proxy,其实mysql本身是支持代理的,但是想自己尝试下这样就会很灵活:

• 灵活slb mysql负载均衡
• 读写直接通过proxy直接进行判断
• 提前预警或拒绝危险性sql
• ...太多太多太多好处

以下是golang mysql proxy的代码

package main
 
import (
 "net"
 "fmt"
 "time"
)
 
const (
 MYSQL_ADDRESS = "mysq-host"
 MYSQL_PORT = "3306"
)
 
func main() {
 listener, err := net.Listen("tcp", ":1234")
 if err != nil {
 fmt.Println("tcp", err.Error())
 return
 } else {
 fmt.Println("tcp", listener.Addr(), "success")
 }
 
 for {
 user, err := listener.Accept()
 if err != nil {
 fmt.Println("accept error: ", err.Error())
 continue
 }
 go proxyRequest(user)
 }
}
 
 
//打理用户请求
func proxyRequest(user net.Conn) {
 
 fmt.Println(user.LocalAddr())
 bytes := make([]byte, 10240)
 
 conn, err := net.Dial("tcp", MYSQL_ADDRESS + ":" + MYSQL_PORT)
 
 if err != nil {
 fmt.Println(conn.RemoteAddr(), "error:", err.Error())
 conn.Close()
 return
 }
 
 ch := make(chan bool, 1)
 go proxyResponse(user, conn, ch)
 
 for {
 n, err := user.Read(bytes)
 if err != nil {
 break
 }
 fmt.Println(string(bytes[:n]))
 conn.Write(bytes[:n])
 }
 
 defer close(ch)
 
 select {
 case <-ch:
 conn.Close()
 user.Close()
 fmt.Println("proxy over")
 case <-time.After(time.Second * 60):
 fmt.Println("proxy timeout")
 }
 
}
 
//代理服务的返回给用户
func proxyResponse(user net.Conn, service net.Conn, ch chan bool) {
 bytes := make([]byte, 10240)
 
 for {
 n, err := service.Read(bytes)
 if err != nil {
 break
 }
 user.Write(bytes[:n])
 }
 
 ch <- true
}

以下是nodejs简单的proxy

var net = require('net');
var model = require('../../models/proxy');
var trace = require('../../libs/trace');
//代理表
var proxys = [];
//tcp server
var server = null;
//proxy information
var information = {count: 0, success: 0, error: 0};
/**
 * 启动服务
 * @param info array
 * @param callback function
 */
exports.start = (info, callback) => {
 model.getProxyListFromServer(info.id, (err, result)=> {
 if (err) {
 callback(err, err);
 } else {
 proxys = result;
 initServer(info.port, info.to_host, info.to_port);
 callback(null);
 }
 });
};
/**
 * 停止服务.
 * @return bool
 */
exports.stop = function () {
 if (server && server.listening) {
 server.close();
 server = null;
 return true;
 }
 return false;
};
/**
 * 获取信息
 * @return object
 */
exports.getInfo = () => {
 return information;
};
/**
 * 初始化tcp proxy server.
 * @param port
 * @param toHost
 * @param toPort
 */
function initServer(port, toHost, toPort) {
 server = net.createServer((client)=> {
 information.count++;
 client.on('end', () => {
 connect.end();
 });
 var connect = net.createConnection({host: toHost, port: toPort}, (err)=> {
 });
 connect.on('error', (err)=> {
 information.error++;
 });
 connect.on('end', ()=> {
 information.success++;
 });
 client.pipe(connect);
 connect.pipe(client);
 // client.on('data', function (data) {
 // var buf = Buffer.from(data);
 // console.log('data: ' + buf.toString());
 // });
 });
 server.listen(port, (err) => {
 if (err) {
 trace.log('proxy server error', err);
 process.exit();
 }
 trace.log('proxy server started', port);
 });
}

本文来自:Segmentfault

感谢作者:gofounder

查看原文:Golang TcpProxy和Nodejs TcpProxy

• 编辑
• 预览