You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
206 lines
5.0 KiB
206 lines
5.0 KiB
// Copyright © 2023 OpenIM. All rights reserved.
|
|
//
|
|
// 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 cmd
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/OpenIMSDK/protocol/constant"
|
|
"github.com/OpenIMSDK/tools/log"
|
|
|
|
"github.com/openimsdk/open-im-server/v3/pkg/common/config"
|
|
)
|
|
|
|
type RootCmdPt interface {
|
|
GetPortFromConfig(portType string) int
|
|
}
|
|
type RootCmd struct {
|
|
Command cobra.Command
|
|
Name string
|
|
port int
|
|
prometheusPort int
|
|
cmdItf RootCmdPt
|
|
}
|
|
|
|
type CmdOpts struct {
|
|
loggerPrefixName string
|
|
}
|
|
|
|
func WithCronTaskLogName() func(*CmdOpts) {
|
|
return func(opts *CmdOpts) {
|
|
opts.loggerPrefixName = "OpenIM.CronTask.log.all"
|
|
}
|
|
}
|
|
|
|
func WithLogName(logName string) func(*CmdOpts) {
|
|
return func(opts *CmdOpts) {
|
|
opts.loggerPrefixName = logName
|
|
}
|
|
}
|
|
|
|
func NewRootCmd(name string, opts ...func(*CmdOpts)) *RootCmd {
|
|
rootCmd := &RootCmd{Name: name}
|
|
cmd := cobra.Command{
|
|
Use: "Start openIM application",
|
|
Short: fmt.Sprintf(`Start %s `, name),
|
|
Long: fmt.Sprintf(`Start %s `, name),
|
|
PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
|
|
if err := rootCmd.getConfFromCmdAndInit(cmd); err != nil {
|
|
panic(err)
|
|
}
|
|
cmdOpts := &CmdOpts{}
|
|
for _, opt := range opts {
|
|
opt(cmdOpts)
|
|
}
|
|
if cmdOpts.loggerPrefixName == "" {
|
|
cmdOpts.loggerPrefixName = "OpenIM.log.all"
|
|
}
|
|
err := log.InitFromConfig(cmdOpts.loggerPrefixName, name, config.Config.Log.RemainLogLevel,
|
|
config.Config.Log.IsStdout, config.Config.Log.IsJson, config.Config.Log.StorageLocation,
|
|
config.Config.Log.RemainRotationCount, config.Config.Log.RotationTime)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return nil
|
|
},
|
|
}
|
|
rootCmd.Command = cmd
|
|
rootCmd.addConfFlag()
|
|
|
|
return rootCmd
|
|
}
|
|
|
|
// func (rc *RootCmd) persistentPreRun(cmd *cobra.Command, opts ...func(*CmdOpts)) error {
|
|
// if err := rc.initializeConfiguration(cmd); err != nil {
|
|
// return fmt.Errorf("failed to get configuration from command: %w", err)
|
|
// }
|
|
|
|
// cmdOpts := rc.applyOptions(opts...)
|
|
|
|
// if err := rc.initializeLogger(cmdOpts); err != nil {
|
|
// return fmt.Errorf("failed to initialize from config: %w", err)
|
|
// }
|
|
|
|
// return nil
|
|
// }
|
|
|
|
//nolint:unused //unused work wrongly
|
|
func (rc *RootCmd) initializeConfiguration(cmd *cobra.Command) error {
|
|
return rc.getConfFromCmdAndInit(cmd)
|
|
}
|
|
|
|
// func (rc *RootCmd) applyOptions(opts ...func(*CmdOpts)) *CmdOpts {
|
|
// cmdOpts := defaultCmdOpts()
|
|
// for _, opt := range opts {
|
|
// opt(cmdOpts)
|
|
// }
|
|
|
|
// return cmdOpts
|
|
// }
|
|
|
|
// func (rc *RootCmd) initializeLogger(cmdOpts *CmdOpts) error {
|
|
// logConfig := config.Config.Log
|
|
|
|
// return log.InitFromConfig(
|
|
|
|
// cmdOpts.loggerPrefixName,
|
|
// rc.Name,
|
|
// logConfig.RemainLogLevel,
|
|
// logConfig.IsStdout,
|
|
// logConfig.IsJson,
|
|
// logConfig.StorageLocation,
|
|
// logConfig.RemainRotationCount,
|
|
// logConfig.RotationTime,
|
|
// )
|
|
// }
|
|
|
|
// func defaultCmdOpts() *CmdOpts {
|
|
// return &CmdOpts{
|
|
// loggerPrefixName: "OpenIM.log.all",
|
|
// }
|
|
// }
|
|
|
|
func (r *RootCmd) SetRootCmdPt(cmdItf RootCmdPt) {
|
|
r.cmdItf = cmdItf
|
|
}
|
|
|
|
func (r *RootCmd) addConfFlag() {
|
|
r.Command.Flags().StringP(constant.FlagConf, "c", "", "path to config file folder")
|
|
}
|
|
|
|
func (r *RootCmd) AddPortFlag() {
|
|
r.Command.Flags().IntP(constant.FlagPort, "p", 0, "server listen port")
|
|
}
|
|
|
|
func (r *RootCmd) getPortFlag(cmd *cobra.Command) int {
|
|
port, _ := cmd.Flags().GetInt(constant.FlagPort)
|
|
if port == 0 {
|
|
port = r.PortFromConfig(constant.FlagPort)
|
|
}
|
|
|
|
return port
|
|
}
|
|
|
|
func (r *RootCmd) GetPortFlag() int {
|
|
return r.port
|
|
}
|
|
|
|
func (r *RootCmd) AddPrometheusPortFlag() {
|
|
r.Command.Flags().IntP(constant.FlagPrometheusPort, "", 0, "server prometheus listen port")
|
|
}
|
|
|
|
func (r *RootCmd) getPrometheusPortFlag(cmd *cobra.Command) int {
|
|
port, _ := cmd.Flags().GetInt(constant.FlagPrometheusPort)
|
|
if port == 0 {
|
|
port = r.PortFromConfig(constant.FlagPrometheusPort)
|
|
}
|
|
|
|
return port
|
|
}
|
|
|
|
func (r *RootCmd) GetPrometheusPortFlag() int {
|
|
return r.prometheusPort
|
|
}
|
|
|
|
func (r *RootCmd) getConfFromCmdAndInit(cmdLines *cobra.Command) error {
|
|
configFolderPath, _ := cmdLines.Flags().GetString(constant.FlagConf)
|
|
fmt.Println("configFolderPath:", configFolderPath)
|
|
|
|
return config.InitConfig(configFolderPath)
|
|
}
|
|
|
|
func (r *RootCmd) Execute() error {
|
|
return r.Command.Execute()
|
|
}
|
|
|
|
func (r *RootCmd) AddCommand(cmds ...*cobra.Command) {
|
|
r.Command.AddCommand(cmds...)
|
|
}
|
|
|
|
func (r *RootCmd) GetPortFromConfig(portType string) int {
|
|
fmt.Println("RootCmd.GetPortFromConfig:", portType)
|
|
|
|
return 0
|
|
}
|
|
|
|
func (r *RootCmd) PortFromConfig(portType string) int {
|
|
fmt.Println("PortFromConfig:", portType)
|
|
|
|
return r.cmdItf.GetPortFromConfig(portType)
|
|
}
|