blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-10-27 05:52:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

update check logic.

Browse Source
This commit is contained in:
Monet Lee 2024-11-13 12:28:54 +08:00
parent 201f1f009e
commit ce27a4acdf
1 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
tools/check-component/main.go
View File

@ -18,7 +18,7 @@ import (
"context" "context"
"flag" "flag"
"fmt" "fmt"
"io/ioutil" "io"
"log" "log"
"os" "os"
"path/filepath" "path/filepath"
@ -37,10 +37,16 @@ import (
const maxRetry = 180 const maxRetry = 180
const (
MountConfigFilePath = "CONFIG_PATH"
DeploymentType = "DEPLOYMENT_TYPE"
KUBERNETES = "kubernetes"
)
func CheckZookeeper(ctx context.Context, config *config.ZooKeeper) error { func CheckZookeeper(ctx context.Context, config *config.ZooKeeper) error {
// Temporary disable logging // Temporary disable logging
originalLogger := log.Default().Writer() originalLogger := log.Default().Writer()
log.SetOutput(ioutil.Discard) log.SetOutput(io.Discard)
defer log.SetOutput(originalLogger) // Ensure logging is restored defer log.SetOutput(originalLogger) // Ensure logging is restored
return zookeeper.Check(ctx, config.Address, config.Schema, zookeeper.WithUserNameAndPassword(config.Username, config.Password)) return zookeeper.Check(ctx, config.Address, config.Schema, zookeeper.WithUserNameAndPassword(config.Username, config.Password))
} }
@ -78,6 +84,13 @@ func initConfig(configDir string) (*config.Mongo, *config.Redis, *config.Kafka,
discovery = &config.Discovery{} discovery = &config.Discovery{}
thirdConfig = &config.Third{} thirdConfig = &config.Third{}
) )
if os.Getenv(DeploymentType) == KUBERNETES {
configDir = os.Getenv(MountConfigFilePath)
if configDir == "" {
return nil, nil, nil, nil, nil, fmt.Errorf("%s env is empty", MountConfigFilePath)
}
}
err := config.Load(configDir, cmd.MongodbConfigFileName, cmd.ConfigEnvPrefixMap[cmd.MongodbConfigFileName], mongoConfig) err := config.Load(configDir, cmd.MongodbConfigFileName, cmd.ConfigEnvPrefixMap[cmd.MongodbConfigFileName], mongoConfig)
if err != nil { if err != nil {
return nil, nil, nil, nil, nil, err return nil, nil, nil, nil, nil, err