route.go

package viot

import(
	"sync"
	"regexp"
	"fmt"
	"path"
	"strings"
)

type Route struct{
	HandlerError	func(w ResponseWriter, r *Request)
	rt       		sync.Map		// 路由表 map[string]
}


//HandleFunc 绑定处理函数，匹配的网址支持正则，这说明你要严格的检查。
//	url string                                          网址，支持正则匹配
//	handler func(w ResponseWriter, r *Request)    		处理函数
func (T *Route) HandleFunc(url string,  handler func(w ResponseWriter, r *Request)){
	if handler == nil {
    	T.rt.Delete(url)
		return
	}
    T.rt.Store(url, HandlerFunc(handler))
}

//ServeIOT 服务IOT
//	w ResponseWriter    响应
//	r *Request          请求
func (T *Route) ServeIOT(w ResponseWriter, r *Request){
	upath := r.URL.Path
	inf, ok := T.rt.Load(r.URL.Path)
	if ok {
		inf.(Handler).ServeIOT(w, r)
		if upath == r.URL.Path {
			return
		}
	}else{
		var handleFunc Handler
		T.rt.Range(func(k, v interface{}) bool {
			pattern := k.(string)
			//正则
			if strings.HasPrefix(pattern, "^") || strings.HasSuffix(pattern, "$") {
		        regexpRegexp, err := regexp.Compile(pattern)
		        if err != nil {
		            return true
		        }
		        _, complete := regexpRegexp.LiteralPrefix()
		        if !complete {
	           		regexpRegexp.Longest()
			        if regexpRegexp.MatchString(r.URL.Path) {
			        	ok = true
			            handleFunc = v.(Handler)
			            return false
			        }
		        }
				return true
			}
			//通配符
			matched, _ := path.Match(pattern, r.URL.Path)
			if matched {
	        	ok = true
	            handleFunc = v.(Handler)
	            return false
			}
			return true
		});
		if ok {
			handleFunc.ServeIOT(w, r)
			if upath == r.URL.Path {
				return
			}
		}
	}
	
	//处理错误的请求
	if T.HandlerError != nil {
		T.HandlerError(w, r)
		return
	}
	
	//默认的错误处理
	w.Status(404)
	w.Header().Set("Connection","close")
	w.SetBody(fmt.Sprintf("The path does not exist %s", upath))
}