gf/g/net/ghttp/http_server_domain.go

package ghttp

import (
    "strings"
    "gitee.com/johng/gf/g/container/gmap"
)

// 域名管理器对象
type Domain struct {
    s *Server         // 所属Server
    m map[string]bool // 多域名
}

// 域名对象表，用以存储和检索域名(支持多域名)与域名对象之间的关联关系
var domains *gmap.StringInterfaceMap = gmap.NewStringInterfaceMap()

// 生成一个域名对象
func (s *Server) Domain(domain string) *Domain {
    if r := domains.Get(domain); r != nil {
        return r.(*Domain)
    }
    d      := &Domain{
        s : s,
        m : make(map[string]bool),
    }
    result := strings.Split(domain, ",")
    for _, v := range result {
        d.m[strings.TrimSpace(v)] = true
    }
    domains.Set(domain, d)
    return d
}

// 在当前域名中绑定回调函数
func (d *Domain) BindHandler(pattern string, handler HandlerFunc) error {
    for domain, _ := range d.m {
        if err := d.s.BindHandler(pattern + "@" + domain, handler); err != nil {
            return err
        }
    }
    return nil
}

// 批量绑定
func (d *Domain) BindHandlerByMap(m HandlerMap) error {
    for p, f := range m {
        if err := d.s.BindHandler(p, f); err != nil {
            return err
        }
    }
    return nil
}

// 绑定REST控制器
func (d *Domain) BindControllerRest(uri string, c ControllerRest) error {
    for domain, _ := range d.m {
        if err := d.s.BindControllerRest(uri + "@" + domain, c); err != nil {
            return err
        }
    }
    return nil
}