Beispiel #1
0
func (this *Reply) write() {
	if this.isWebSocket {
		return
	}
	this.writeHeader()
	if this.redirect {
		return
	}
	if this.data != nil {
		if reader, ok := this.data.(io.Reader); ok {
			_, err := io.Copy(this.w, reader)
			if err != nil {
				logger.Error("数据输出出现错误:%s", err)
				return
			}
			if limitReader, ok := this.data.(*io.LimitedReader); ok {
				reader = limitReader.R
				if closer, ok := reader.(io.Closer); ok {
					closer.Close()
				}
			}
			if closer, ok := this.data.(io.Closer); ok {
				closer.Close()
			}
		} else {
			err := this.transport.Out(this.w, this.data)
			if err != nil {
				logger.Error("数据序列化失败:%s", err)
			}
		}
	}
}
Beispiel #2
0
func buildActionHandler(action *defauleAction) (*actionHandler, error) {
	path := action.GetPath()
	if !strings.HasPrefix(path, PATH_SEPARATOR) {
		return nil, errors.New(logger.Error("定义的Uri必须是%s前缀", PATH_SEPARATOR))
	}
	paths := strings.Split(path, PATH_SEPARATOR)
	uriConversions := make(map[string]int, 0)
	conversionUri := ""
	pathSize := 0
	for index, p := range paths {
		pathSize++
		if p == "" {
			continue
		}
		if strings.HasPrefix(p, WILDCARD_PREFIX) && strings.HasSuffix(p, WILDCARD_SUFFIX) {
			name := string([]byte(p)[len(WILDCARD_PREFIX) : len(p)-len(WILDCARD_SUFFIX)])
			uriConversions[name] = index
			p = _conversion
		}
		conversionUri += (PATH_SEPARATOR + p)
	}
	if conversionUri == "" {
		conversionUri = PATH_SEPARATOR
	}
	exp, err := regexp.Compile(conversionUri)
	if err != nil {
		return nil, err
	}
	return &actionHandler{exp: exp, method: action.GetMethod(), defineUri: action.GetPath(), hasPathFragments: len(uriConversions) > 0, uriConversions: uriConversions, pathSize: pathSize, service: action.Service}, nil
}
Beispiel #3
0
func (this *Server) Start() error {
	logger.Debug("serverConfig is %#v", this.config)
	this.router.matcher.sort()
	conf := this.config
	server := &http.Server{Handler: http.HandlerFunc(this.serverHttpHandler), MaxHeaderBytes: conf.MaxHeaderBytes, TLSConfig: conf.TLSConfig}
	if conf.ReadTimeout > 0 {
		server.ReadTimeout = conf.ReadTimeout
	}
	if conf.WriteTimeout > 0 {
		server.WriteTimeout = conf.WriteTimeout
	}
	http2.ConfigureServer(server, &http2.Server{})
	var err error
	this.listener, err = net.Listen("tcp", conf.serverAddr)
	if err != nil {
		return errors.New(logger.Error("listen [%s] fail:%s", conf.serverAddr, err))
	}
	logger.Info("start HttpServer :%s", conf.serverAddr)
	keepAliveListrener := tcpKeepAliveListener{this.listener.(*net.TCPListener)}
	if conf.TLSConfig != nil {
		conf.TLSConfig.NextProtos = append(conf.TLSConfig.NextProtos, "http/1.1")
		go server.Serve(tls.NewListener(keepAliveListrener, conf.TLSConfig))
	} else {
		go server.Serve(keepAliveListrener)
	}
	return nil
}
Beispiel #4
0
func (this *pathMatcher) regeditAction(action *defauleAction) error {
	newActionHandler, err := buildActionHandler(action)
	if err != nil {
		logger.Error("添加Action出错:%s", err)
		return err
	}
	actionHandlers, ok := this.actionHandlerMap[action.GetMethod()]
	if !ok {
		actionHandlers = make(_actionHandler, 0)
	}
	for _, handler := range actionHandlers {
		if handler.exp.String() == newActionHandler.exp.String() {
			return errors.New(logger.Error("定义的uri[%s]与[%s]产生冲突,不能添加", handler.defineUri, newActionHandler.defineUri))
		}
	}
	this.actionHandlerMap[action.GetMethod()] = append(actionHandlers, newActionHandler)
	return nil
}
Beispiel #5
0
func newRegexUriPatternMatcher(uriPattern string) uriPatternMatcher {
	matcher := new(regexUriPatternMatcher)
	var err error
	matcher.pattern, err = regexp.Compile(uriPattern)
	if err != nil {
		logger.Error("编译正则表达式[%s]异常,%s", uriPattern, err)
		return nil
	}
	return uriPatternMatcher(matcher)
}
Beispiel #6
0
//创建一个Server,参数可以为空,默认使用0.0.0.0:8888
func NewServer(serverConfig *ServerConfig) *Server {
	if serverConfig == nil {
		serverConfig = &ServerConfig{Addr: "0.0.0.0"}
	}
	if serverConfig.Port == 0 {
		serverConfig.Port = 8888
	}
	if serverConfig.OpenHttp2 && serverConfig.TLSConfig == nil {
		logger.Error("open http2 need TLS support")
		return nil
	}
	addr := net.JoinHostPort(serverConfig.Addr, strconv.Itoa(serverConfig.Port))
	serverAddr, err := net.ResolveTCPAddr("tcp", addr)
	if err != nil {
		logger.Error("can't parse the server addr [%s],cause:%s", addr, err)
		return nil
	}
	serverConfig.serverAddr = serverAddr.String()
	return &Server{router: newRoutingDispatcher(), config: serverConfig}
}
Beispiel #7
0
func (this *actionHandler) doAction(request *http.Request, reply *Reply) {
	requestUri := request.URL.Path
	param := make(map[string]string, 0)
	if this.hasPathFragments {
		paths := strings.Split(requestUri, PATH_SEPARATOR)
		if this.pathSize != len(paths) {
			panic(errors.New(logger.Error("需要解析的uri[%s]不匹配定义的uri[%s]", requestUri, this.defineUri)))
		}
		for name, index := range this.uriConversions {
			param[name] = paths[index]
		}
	}
	this.service(request, param, reply)
}
Beispiel #8
0
func (route *routingDispatcher) handle(request *http.Request, reply *Reply) {
	defer func() {
		if err := recover(); err != nil {
			reply.SetCode(500).With(err)
		}
	}()
	handler := route.matcher.getActionHandler(request.URL.Path, strings.ToUpper(request.Method))
	if handler == nil {
		reply.SetCode(404).With("404:you are lost")
		logger.Error("Not found Handler for[%s] [%s]", strings.ToUpper(request.Method), request.URL.Path)
		return
	}
	handler.doAction(request, reply)
}
Beispiel #9
0
// Index responds with the pprof-formatted profile named by the request.
// For example, "/debug/pprof/heap" serves the "heap" profile.
// Index responds to a request for "/debug/pprof/" with an HTML page
// listing the available profiles.
func Index(request *http.Request, pathFragments map[string]string, reply *web.Reply) {
	if strings.HasPrefix(request.URL.Path, "/debug/pprof/") {
		name := strings.TrimPrefix(request.URL.Path, "/debug/pprof/")
		if name != "" {
			handler(name).RequestHandler(request, pathFragments, reply)
			return
		}
	}
	r, w := io.Pipe()
	go func() {
		defer w.Close()
		profiles := pprof.Profiles()
		if err := indexTmpl.Execute(w, profiles); err != nil {
			logger.Error("出现错误:%s", err)
		}
	}()
	reply.With(r)
}
Beispiel #10
0
func main() {
	tlsConfig, err := web.NewTLSConfig("server.crt", "server.key")
	if err != nil {
		logger.Error("证书初始化失败,%s", err)
		time.Sleep(time.Second)
		return
	}
	serConfig := new(web.ServerConfig)
	serConfig.TLSConfig = tlsConfig
	serConfig.OpenHttp2 = true
	server := web.NewServer(serConfig)
	server.Regedit("/test", web.GET, TestService)
	server.Regedit("/reqinfo", web.GET, reqInfoHandler)
	server.Regedit("/a/{name}/123", web.GET, Service)
	server.AddFilter("/*", web.AccessLogFilter)
	server.Start()
	time.Sleep(time.Second * 120)
	server.Stop()
}