feat: support fasthttp server (#288)

Author: sysulq

pkg/server/xfasthttp/config.go

@@ -0,0 +1,127 @@
+// Copyright 2022 Douyu
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package xfasthttp
+
+import (
+	"fmt"
+
+	"github.com/douyu/jupiter/pkg/conf"
+	"github.com/douyu/jupiter/pkg/constant"
+	"github.com/douyu/jupiter/pkg/ecode"
+	"github.com/douyu/jupiter/pkg/flag"
+	"github.com/douyu/jupiter/pkg/xlog"
+	"github.com/pkg/errors"
+)
+
+//ModName named a mod
+const ModName = "server.fasthttp"
+
+//Config HTTP config
+type Config struct {
+	Host              string
+	Port              int
+	Deployment        string
+	Debug             bool
+	DisableMetric     bool
+	DisableTrace      bool
+	DisablePrintStack bool
+	// ServiceAddress service address in registry info, default to 'Host:Port'
+	ServiceAddress string
+	CertFile       string
+	PrivateFile    string
+	EnableTLS      bool
+
+	SlowQueryThresholdInMilli int64
+	ReadBufferSize            int
+	WriteBufferSize           int
+	ReduceMemoryUsage         bool
+
+	logger *xlog.Logger
+}
+
+// DefaultConfig ...
+func DefaultConfig() *Config {
+	return &Config{
+		Host:                      flag.String("host"),
+		Port:                      9091,
+		Debug:                     false,
+		Deployment:                constant.DefaultDeployment,
+		SlowQueryThresholdInMilli: 500,  // 500ms
+		ReadBufferSize:            1024, // 1KB
+		WriteBufferSize:           1024, // 1KB
+		ReduceMemoryUsage:         true,
+		logger:                    xlog.JupiterLogger.With(xlog.FieldMod(ModName)),
+		DisablePrintStack:         false,
+		EnableTLS:                 false,
+		CertFile:                  "cert.pem",
+		PrivateFile:               "private.pem",
+	}
+}
+
+// StdConfig Jupiter Standard HTTP Server config
+func StdConfig(name string) *Config {
+	return RawConfig("jupiter.server." + name)
+}
+
+// RawConfig ...
+func RawConfig(key string) *Config {
+	var config = DefaultConfig()
+	if err := conf.UnmarshalKey(key, &config); err != nil &&
+		errors.Cause(err) != conf.ErrInvalidKey {
+		config.logger.Panic("http server parse config panic", xlog.FieldErrKind(ecode.ErrKindUnmarshalConfigErr), xlog.FieldErr(err), xlog.FieldKey(key), xlog.FieldValueAny(config))
+	}
+	return config
+}
+
+// WithLogger ...
+func (config *Config) WithLogger(logger *xlog.Logger) *Config {
+	config.logger = logger
+	return config
+}
+
+// WithHost ...
+func (config *Config) WithHost(host string) *Config {
+	config.Host = host
+	return config
+}
+
+// WithPort ...
+func (config *Config) WithPort(port int) *Config {
+	config.Port = port
+	return config
+}
+
+func (config *Config) MustBuild() *Server {
+	server, err := config.Build()
+	if err != nil {
+		xlog.Panicf("build echo server failed: %v", err)
+	}
+	return server
+}
+
+// Build create server instance, then initialize it with necessary interceptor
+func (config *Config) Build() (*Server, error) {
+	server, err := newServer(config)
+	if err != nil {
+		return nil, err
+	}
+
+	return server, nil
+}
+
+// Address ...
+func (config *Config) Address() string {
+	return fmt.Sprintf("%s:%d", config.Host, config.Port)
+}

pkg/server/xfasthttp/const.go

@@ -0,0 +1,38 @@
+// Copyright 2022 Douyu
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package xfasthttp
+
+// Headers
+const (
+	// HeaderAcceptEncoding ...
+	HeaderAcceptEncoding = "Accept-Encoding"
+	// HeaderContentType ...
+	HeaderContentType = "Content-Type"
+	// HRPC Errord
+	HeaderHRPCErr = "HRPC-Errord"
+)
+
+// MIME types
+const (
+	// MIMEApplicationJSON ...
+	MIMEApplicationJSON = "application/json"
+	// MIMEApplicationJSONCharsetUTF8 ...
+	MIMEApplicationJSONCharsetUTF8 = MIMEApplicationJSON + "; " + charsetUTF8
+	// MIMEApplicationProtobuf ...
+	MIMEApplicationProtobuf = "application/protobuf"
+)
+const (
+	charsetUTF8 = "charset=utf-8"
+)

pkg/server/xfasthttp/go.mod

@@ -0,0 +1,30 @@
+module github.com/douyu/jupiter/pkg/server/xfasthttp
+
+go 1.17
+
+replace github.com/douyu/jupiter => ../../../
+
+require (
+	github.com/douyu/jupiter v0.3.5
+	github.com/pkg/errors v0.9.1
+	github.com/valyala/fasthttp v1.33.0
+	go.uber.org/zap v1.19.1
+)
+
+require (
+	github.com/BurntSushi/toml v0.3.1 // indirect
+	github.com/andybalholm/brotli v1.0.4 // indirect
+	github.com/davecgh/go-spew v1.1.1 // indirect
+	github.com/golang/protobuf v1.5.2 // indirect
+	github.com/json-iterator/go v1.1.11 // indirect
+	github.com/klauspost/compress v1.14.1 // indirect
+	github.com/mitchellh/mapstructure v1.3.2 // indirect
+	github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
+	github.com/modern-go/reflect2 v1.0.1 // indirect
+	github.com/valyala/bytebufferpool v1.0.0 // indirect
+	go.uber.org/atomic v1.7.0 // indirect
+	go.uber.org/multierr v1.6.0 // indirect
+	google.golang.org/genproto v0.0.0-20210602131652-f16073e35f0c // indirect
+	google.golang.org/grpc v1.43.0 // indirect
+	google.golang.org/protobuf v1.26.0 // indirect
+)