blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-04-24 02:16:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

if [ -eq 1 ]; then

Browse Source 
Signed-off-by: hanzhixiao <709674996@qq.com>
This commit is contained in:
hanzhixiao 2023-08-10 16:37:59 +08:00
parent b01f66b54e
commit ba65647e66
1 changed files with 31 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
tools/component/main.go
View File

@ -2,6 +2,7 @@ package main
import (
"context"
"database/sql"
"fmt"
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/config"
"github.com/OpenIMSDK/tools/errs"
@ -129,15 +130,20 @@ func exactIP(urll string) string {
}
func checkMysql() error {
var sqlDB *sql.DB
defer func() {
if sqlDB != nil {
sqlDB.Close()
}
}()
dsn := fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8mb4&parseTime=true&loc=Local",
config.Config.Mysql.Username, config.Config.Mysql.Password, config.Config.Mysql.Address[0], "mysql")
db, err := gorm.Open(mysql.Open(dsn), nil)
if err != nil {
return err
} else {
sqlDB, err := db.DB()
sqlDB, err = db.DB()
err = sqlDB.Ping()
sqlDB.Close()
if err != nil {
return err
}
@ -146,6 +152,12 @@ func checkMysql() error {
}
func checkMongo() error {
var client *mongo.Client
defer func() {
if client != nil {
client.Disconnect(context.TODO())
}
}()
mongodbHosts := ""
for i, v := range config.Config.Mongo.Address {
if i == len(config.Config.Mongo.Address)-1 {
@ -161,7 +173,6 @@ func checkMongo() error {
return err
} else {
err = client.Ping(context.TODO(), &readpref.ReadPref{})
client.Disconnect(context.TODO())
if err != nil {
return err
}
@ -199,6 +210,11 @@ func checkMinio() error {
func checkRedis() error {
var redisClient redis.UniversalClient
defer func() {
if redisClient != nil {
redisClient.Close()
}
}()
if len(config.Config.Redis.Address) > 1 {
redisClient = redis.NewClusterClient(&redis.ClusterOptions{
Addrs: config.Config.Redis.Address,
@ -220,19 +236,23 @@ func checkRedis() error {
}
func checkZookeeper() error {
var c *zk.Conn
defer func() {
if c != nil {
c.Close()
}
}()
c, _, err := zk.Connect(config.Config.Zookeeper.ZkAddr, time.Second)
if err != nil {
return err
} else {
if config.Config.Zookeeper.Username != "" && config.Config.Zookeeper.Password != "" {
if err := c.AddAuth("digest", []byte(config.Config.Zookeeper.Username+":"+config.Config.Zookeeper.Password)); err != nil {
c.Close()
return err
}
}
_, _, err = c.Get("/")
if err != nil {
c.Close()
return err
}
}
@ -240,6 +260,12 @@ func checkZookeeper() error {
}
func checkKafka() error {
var kafkaClient sarama.Client
defer func() {
if kafkaClient != nil {
kafkaClient.Close()
}
}()
cfg := sarama.NewConfig()
if config.Config.Kafka.Username != "" && config.Config.Kafka.Password != "" {
cfg.Net.SASL.Enable = true
@ -251,7 +277,6 @@ func checkKafka() error {
return err
} else {
topics, err := kafkaClient.Topics()
kafkaClient.Close()
if err != nil {
return err
}