telegram_loki/internal/logic/sql/logic_sql.go

package sql

import (
	"context"
	"fmt"
	"github.com/ManyakRus/starter/contextmain"
	"github.com/ManyakRus/starter/log"
	"github.com/ManyakRus/starter/micro"
	"github.com/ManyakRus/starter/postgres_pgx"
	"github.com/ManyakRus/starter/telegram_client"
	"github.com/ManyakRus/telegram_loki/internal/config"
	"github.com/jackc/pgx/v5"
	"os"
	"path"
	"time"
)

// Ticker - таймер, запускается каждые INTERVAL_SEND_MINUTES (10) минут
var Ticker *time.Ticker

// Start - старт работы чтения логов LOKI
func Start() {
	RunSQL_and_Send()

	Ticker = time.NewTicker(time.Duration(config.Settings.DATABASE_CHECKER_INTERVAL_MINUTES) * time.Minute)
	defer Ticker.Stop()

	go ReadTicker()
}

// ReadTicker - запускается каждые INTERVAL_SEND_MINUTES минут
func ReadTicker() {
	for {
		select {
		case <-contextmain.GetContext().Done():
			return
		case <-Ticker.C:
			RunSQL_and_Send()
		}
	}
}

func RunSQL_and_Send() {
	err := RunSQL()
	if err != nil {
		//отправим в Telegram
		_, err = telegram_client.SendMessage(config.Settings.TELEGRAM_CHAT_NAME, err.Error())
		if err != nil {
			log.Error("SendMessage() error: ", err)
		}
	}

}

// RunSQL - запускает все скрипты .sql
func RunSQL() error {
	var err error

	log.Debug("Start run .sql scripts")

	dir := micro.ProgramDir_bin()
	DirFilename := dir + config.Settings.SQL_FILES_FOLDER
	Files, err := os.ReadDir(DirFilename)
	if err != nil {
		err = fmt.Errorf("os.ReadDir() error: %w", err)
		log.Error(err)
		return err
	}

	//пройдемся по всем файлам
	for _, file1 := range Files {
		//пропускаем файлы, которые не .sql
		//FileInfo, err := file1.Info()
		//if err != nil {
		//	err = fmt.Errorf("file1.Info() error: %w", err)
		//	log.Error(err)
		//	return err
		//}

		Filename := DirFilename + micro.SeparatorFile() + file1.Name()
		if file1.Type().IsRegular() == false || len(Filename) < 4 || Filename[len(Filename)-4:] != ".sql" {
			continue
		}

		//запускаем скрипт
		err = RunSQL1(Filename)
		if err != nil {
			log.Warn(err)
			return err
		}
	}

	return err
}

// RunSQL1 - запускает 1 скрипт
func RunSQL1(Filename string) error {
	var err error

	//подключимся к БД
	db := postgres_pgx.GetConnection()

	ctxMain := contextmain.GetContext()
	ctx, cancel := context.WithTimeout(ctxMain, 10*time.Minute)
	defer cancel()

	//прочитаем файл
	bytes, err := os.ReadFile(Filename)
	if err != nil {
		err = fmt.Errorf("ReadFile() Filename: %s error: %w", Filename, err)
		log.Error(err)

		return err
	}
	TextSQL := string(bytes)

	//запустим запрос
	ResultSQL := ""
	err = db.QueryRow(ctx, TextSQL).Scan(&ResultSQL)

	//нет строк - это хорошо
	if err == pgx.ErrNoRows {
		err = nil
		return err
	}

	//ошибки не должно быть
	if err != nil {
		err = fmt.Errorf("db.Exec() Filename: %s, error: %w", Filename, err)
		log.Error(err)

		return err
	}

	//запрос вернул число(строку)
	FilenameShort := path.Base(Filename)
	err = fmt.Errorf("скрипт %s вернул значение: %s", FilenameShort, ResultSQL)

	return err
}