mirror of
https://github.com/ManyakRus/starter.git
synced 2024-11-28 09:33:45 +02:00
124 lines
2.9 KiB
Go
124 lines
2.9 KiB
Go
package nats_connect
|
|
|
|
import (
|
|
"github.com/nats-io/nats.go"
|
|
"github.com/ManyakRus/starter/logger"
|
|
"github.com/ManyakRus/starter/ping"
|
|
"os"
|
|
//"github.com/ManyakRus/starter/common/v0/micro"
|
|
"github.com/ManyakRus/starter/contextmain"
|
|
"github.com/ManyakRus/starter/stopapp"
|
|
)
|
|
|
|
// Conn - соединение к серверу nats
|
|
var Conn *nats.Conn
|
|
|
|
// log - глобальный логгер
|
|
var log = logger.GetLog()
|
|
|
|
// Settings хранит все нужные переменные окружения
|
|
var Settings SettingsINI
|
|
|
|
// SettingsINI - структура для хранения всех нужных переменных окружения
|
|
type SettingsINI struct {
|
|
NATS_HOST string
|
|
NATS_PORT string
|
|
NATS_LOGIN string
|
|
NATS_PASSWORD string
|
|
}
|
|
|
|
// Connect - подключается к серверу Nats
|
|
func Connect() {
|
|
var err error
|
|
|
|
err = Connect_err()
|
|
if err != nil {
|
|
log.Panicln("NATS Connect() error: ", err)
|
|
} else {
|
|
log.Info("NATS Connect() ok ")
|
|
}
|
|
}
|
|
|
|
// Connect_err - подключается к серверу Nats и возвращает ошибку
|
|
func Connect_err() error {
|
|
var err error
|
|
|
|
if Settings.NATS_HOST == "" {
|
|
FillSettings()
|
|
}
|
|
|
|
ping.Ping(Settings.NATS_HOST, Settings.NATS_PORT)
|
|
|
|
sNATS_PORT := Settings.NATS_PORT
|
|
URL := "nats://" + Settings.NATS_HOST + ":" + sNATS_PORT
|
|
UserInfo := nats.UserInfo(Settings.NATS_LOGIN, Settings.NATS_PASSWORD)
|
|
Conn, err = nats.Connect(URL, UserInfo)
|
|
|
|
//nats.ManualAck()
|
|
return err
|
|
}
|
|
|
|
// StartNats - необходимые процедуры для подключения к серверу Nats
|
|
func StartNats() {
|
|
Connect()
|
|
|
|
stopapp.GetWaitGroup_Main().Add(1)
|
|
go WaitStop()
|
|
|
|
}
|
|
|
|
// CloseConnection - закрывает соединение с сервером Nats
|
|
func CloseConnection() {
|
|
//var err error
|
|
|
|
if Conn == nil {
|
|
return
|
|
}
|
|
|
|
Conn.Close()
|
|
|
|
log.Info("NATS stopped")
|
|
|
|
return
|
|
}
|
|
|
|
// WaitStop - ожидает отмену глобального контекста или сигнала завершения приложения
|
|
func WaitStop() {
|
|
|
|
select {
|
|
case <-contextmain.GetContext().Done():
|
|
log.Warn("Context app is canceled. NATS_Connect.")
|
|
}
|
|
|
|
CloseConnection()
|
|
|
|
stopapp.GetWaitGroup_Main().Done()
|
|
}
|
|
|
|
// FillSettings загружает переменные окружения в структуру из файла или из переменных окружения
|
|
func FillSettings() {
|
|
Settings = SettingsINI{}
|
|
Settings.NATS_HOST = os.Getenv("NATS_HOST")
|
|
Settings.NATS_PORT = os.Getenv("NATS_PORT")
|
|
Settings.NATS_LOGIN = os.Getenv("NATS_LOGIN")
|
|
Settings.NATS_PASSWORD = os.Getenv("NATS_PASSWORD")
|
|
|
|
if Settings.NATS_HOST == "" {
|
|
log.Panicln("Need fill NATS_HOST ! in os.ENV ")
|
|
}
|
|
|
|
if Settings.NATS_PORT == "" {
|
|
log.Panicln("Need fill NATS_PORT ! in os.ENV ")
|
|
}
|
|
|
|
//if Settings.NATS_LOGIN == "" {
|
|
// log.Panicln("Need fill NATS_LOGIN ! in os.ENV ")
|
|
//}
|
|
//
|
|
//if Settings.NATS_PASSWORD == "" {
|
|
// log.Panicln("Need fill NATS_PASSWORD ! in os.ENV ")
|
|
//}
|
|
|
|
//
|
|
}
|