Apply .gitignore rules

.svn/pristine/db/db1dfeaa1c83992313b5e816175c43d5072d5678.svn-base

@@ -0,0 +1,56 @@
+package webServer
+
+import (
+	"fmt"
+	"net/http"
+)
+
+var (
+	// 处理方法集合
+	// Key:API名
+	handlerData = make(map[string]*ApiHandler)
+
+	// 文档方法
+	remarkFunc func(w http.ResponseWriter, r *http.Request)
+)
+
+// RegisterRemarkFunc
+// 	@description: 注册文档api方法
+// parameter:
+//	@_remarkFunc: _remarkFunc
+// return:
+func RegisterRemarkFunc(_remarkFunc func(w http.ResponseWriter, r *http.Request)) {
+	remarkFunc = _remarkFunc
+}
+
+// RegisterHandleFunc
+// 	@description: 注册处理方法
+// parameter:
+//	@apiName: API名称
+//	@callback: 回调方法
+//	@paramNames: 参数名称集合
+// return:
+func RegisterHandleFunc(apiName string, callback HandleFunc, paramNames ...string) {
+	apiFullName := fmt.Sprintf("/API/%s", apiName)
+
+	if _, exist := handlerData[apiFullName]; exist {
+		panic(fmt.Errorf("重复注册处理函数:%s", apiFullName))
+	}
+
+	handlerData[apiFullName] = newApiHandler(apiFullName, callback, paramNames...)
+}
+
+// GetHandleFunc
+// 	@description: 获取请求方法
+// parameter:
+//	@apiFullName: 方法名称
+// return:
+//	@*ApiHandler: 请求方法
+//	@bool: 是否存在
+func GetHandleFunc(apiFullName string) (*ApiHandler, bool) {
+	if requestFuncObj, exists := handlerData[apiFullName]; exists {
+		return requestFuncObj, exists
+	}
+
+	return nil, false
+}

.svn/pristine/db/dbd36c0b2ae73296a480cbab67c3ac8aea40cd6b.svn-base

@@ -0,0 +1,115 @@
+package netUtil
+
+import (
+	"net"
+	"net/http"
+	"strconv"
+	"strings"
+
+	"github.com/gorilla/websocket"
+)
+
+// 远程地址解析器
+type RemoteAddrParser struct {
+	// 主机地址（IP）
+	Host string
+
+	// 端口
+	Port int
+}
+
+// 解析远程地址
+func (this *RemoteAddrParser) parseRemoteAddr(remoteAddr string) {
+	/*
+		http中调用JoinHostPort来给RemoteAddr赋值；它的规则如下：
+		JoinHostPort combines host and port into a network address of the
+		form "host:port" or, if host contains a colon or a percent sign,
+		"[host]:port".
+		net包中是类似的
+
+		所以现在要将RemoteAddr解析成host和port，则需要找到最后一个:，前面的部分则是host；
+		如果host包含[]，则需要去除
+	*/
+
+	// 找到分隔host、port的:
+	index := strings.LastIndex(remoteAddr, ":")
+	if index == -1 {
+		return
+	}
+
+	// 取出host部分
+	this.Host = remoteAddr[:index]
+	this.Port, _ = strconv.Atoi(remoteAddr[index+1:])
+
+	// 处理host中可能的[]
+	if strings.Index(this.Host, "[") == -1 {
+		return
+	}
+	this.Host = this.Host[1:]
+
+	if strings.Index(this.Host, "]") == -1 {
+		return
+	}
+	this.Host = this.Host[:len(this.Host)-1]
+
+	return
+}
+
+// 直接读取IP地址
+func GetHttpAddr(request *http.Request) *RemoteAddrParser {
+	this := &RemoteAddrParser{}
+	this.parseRemoteAddr(request.RemoteAddr)
+	return this
+}
+
+// 优先获取header中的代理地址，如果未设置代理地址，则使用request地址
+func GetHttpAddr2(request *http.Request) *RemoteAddrParser {
+	remoteAddr := request.Header.Get("HTTP_X_FORWARDED_FOR")
+	if len(remoteAddr) > 0 {
+		if len(remoteAddr) >= 10 {
+			strArray := strings.Split(remoteAddr, ",")
+			if len(strArray) > 0 {
+				remoteAddr = strArray[0]
+			}
+		}
+
+		if len(remoteAddr) > 0 && len(remoteAddr) <= 15 {
+			this := &RemoteAddrParser{
+				Host: remoteAddr,
+			}
+			return this
+		}
+	}
+
+	remoteAddr = request.Header.Get("X-Real-IP")
+	if len(remoteAddr) > 0 {
+		this := &RemoteAddrParser{
+			Host: remoteAddr,
+		}
+		return this
+	}
+
+	remoteAddr = request.Header.Get("X-Forwarded-For")
+	if len(remoteAddr) > 0 {
+		this := &RemoteAddrParser{
+			Host: remoteAddr,
+		}
+		return this
+	}
+
+	this := &RemoteAddrParser{}
+	this.parseRemoteAddr(request.RemoteAddr)
+	return this
+}
+
+func GetWebSocketAddr(conn *websocket.Conn) *RemoteAddrParser {
+	this := &RemoteAddrParser{}
+	this.parseRemoteAddr(conn.RemoteAddr().String())
+	return this
+}
+
+func GetConnAddr(conn net.Conn) *RemoteAddrParser {
+	this := &RemoteAddrParser{}
+	this.parseRemoteAddr(conn.RemoteAddr().String())
+	return this
+}