123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104 |
- // Copyright GoFrame Author(https://goframe.org). All Rights Reserved.
- //
- // This Source Code Form is subject to the terms of the MIT License.
- // If a copy of the MIT was not distributed with this file,
- // You can obtain one at https://github.com/gogf/gf.
- package gins
- import (
- "context"
- "fmt"
- "github.com/gogf/gf/v2/internal/consts"
- "github.com/gogf/gf/v2/internal/intlog"
- "github.com/gogf/gf/v2/net/ghttp"
- "github.com/gogf/gf/v2/util/gconv"
- "github.com/gogf/gf/v2/util/gutil"
- )
- const (
- frameCoreComponentNameServer = "gf.core.component.server" // Prefix for HTTP server instance.
- )
- // Server returns an instance of http server with specified name.
- // Note that it panics if any error occurs duration instance creating.
- func Server(name ...interface{}) *ghttp.Server {
- var (
- err error
- ctx = context.Background()
- instanceName = ghttp.DefaultServerName
- instanceKey = fmt.Sprintf("%s.%v", frameCoreComponentNameServer, name)
- )
- if len(name) > 0 && name[0] != "" {
- instanceName = gconv.String(name[0])
- }
- return localInstances.GetOrSetFuncLock(instanceKey, func() interface{} {
- server := ghttp.GetServer(instanceName)
- if Config().Available(ctx) {
- // Server initialization from configuration.
- var (
- configMap map[string]interface{}
- serverConfigMap map[string]interface{}
- serverLoggerConfigMap map[string]interface{}
- configNodeName string
- )
- if configMap, err = Config().Data(ctx); err != nil {
- intlog.Errorf(ctx, `retrieve config data map failed: %+v`, err)
- }
- // Find possible server configuration item by possible names.
- if len(configMap) > 0 {
- if v, _ := gutil.MapPossibleItemByKey(configMap, consts.ConfigNodeNameServer); v != "" {
- configNodeName = v
- }
- if configNodeName == "" {
- if v, _ := gutil.MapPossibleItemByKey(configMap, consts.ConfigNodeNameServerSecondary); v != "" {
- configNodeName = v
- }
- }
- }
- // Automatically retrieve configuration by instance name.
- serverConfigMap = Config().MustGet(
- ctx,
- fmt.Sprintf(`%s.%s`, configNodeName, instanceName),
- ).Map()
- if len(serverConfigMap) == 0 {
- serverConfigMap = Config().MustGet(ctx, configNodeName).Map()
- }
- if len(serverConfigMap) > 0 {
- if err = server.SetConfigWithMap(serverConfigMap); err != nil {
- panic(err)
- }
- } else {
- // The configuration is not necessary, so it just prints internal logs.
- intlog.Printf(
- ctx,
- `missing configuration from configuration component for HTTP server "%s"`,
- instanceName,
- )
- }
- // Server logger configuration checks.
- serverLoggerConfigMap = Config().MustGet(
- ctx,
- fmt.Sprintf(`%s.%s.%s`, configNodeName, instanceName, consts.ConfigNodeNameLogger),
- ).Map()
- if len(serverLoggerConfigMap) == 0 && len(serverConfigMap) > 0 {
- serverLoggerConfigMap = gconv.Map(serverConfigMap[consts.ConfigNodeNameLogger])
- }
- if len(serverLoggerConfigMap) > 0 {
- if err = server.Logger().SetConfigWithMap(serverLoggerConfigMap); err != nil {
- panic(err)
- }
- }
- }
- // The server name is necessary. It sets a default server name is it is not configured.
- if server.GetName() == "" || server.GetName() == ghttp.DefaultServerName {
- server.SetName(instanceName)
- }
- // As it might use template feature,
- // it initializes the view instance as well.
- _ = getViewInstance()
- return server
- }).(*ghttp.Server)
- }
|