main.go

package main

import (
	"fmt"
	"mysql-proxy/lib"
	"net"
	"runtime"
	"strconv"
	"sync"
)

func main() {

	mysqlUrl := ":3306"
	listen := ":3366"
	connCount := runtime.NumCPU()

	server, _ := net.Listen("tcp", listen)
	connList := make([]lib.ProxyConn, connCount)
	for i := 0; i < connCount; i++ {
		proxy := lib.ProxyConn{ID: i}
		connList[i] = proxy
	}

	wg := sync.WaitGroup{}
	wg.Add(1)

	for i := 0; i < connCount; i++ {

		go func(proxy lib.ProxyConn) {

			proxy.NewClientConn(server)
			proxy.NewMysqlConn(mysqlUrl)

			fmt.Println("Connection " + strconv.Itoa(proxy.ID) + " established")

			err := proxy.Handshake()
			if err != nil {
				proxy.Close()
			}

			go proxy.PipeMysql2Client()
			go proxy.PipeClient2Mysql()

			for {
				if !proxy.IsClientClose() {
					continue
				}
				proxy.NewClientConn(server)
				fmt.Println("Connection " + strconv.Itoa(proxy.ID) + " working")
				err := proxy.FakeHandshake()
				if err != nil {
					proxy.CloseClient()
				}
				go proxy.PipeClient2Mysql()
			}

		}(connList[i])

	}

	fmt.Println("Start at " + listen + " , with max " + strconv.Itoa(connCount) + " connection")
	wg.Wait()
}