Добавление логирования в файл

This commit is contained in:
2025-09-25 20:24:33 +03:00
parent d6fe11c8e3
commit b826188019
5 changed files with 239 additions and 1 deletions

View File

@@ -249,6 +249,44 @@ run_every.week(check_python_packages)
---
## Логирование
ALR-Updater поддерживает логирование как в консоль (stderr), так и в файл с автоматической ротацией:
### Настройка логирования в файл
В файле `/etc/alr-updater/config.toml`:
```toml
[logging]
# Включить логирование в файл
enable_file = true
# Путь к файлу логов
log_file = "/var/log/alr-updater.log"
# Максимальный размер файла в байтах (100MB)
max_size = 104857600
```
### Особенности:
- При достижении максимального размера файл автоматически ротируется
- Старый файл сохраняется с временной меткой (например, `alr-updater.log.20250125-143022`)
- Хранится до 5 резервных копий логов
- Логи пишутся одновременно в stderr и файл
### Просмотр логов:
```bash
# Через systemd
journalctl -u alr-updater -f
# Из файла
tail -f /var/log/alr-updater.log
# Поиск ошибок
grep ERROR /var/log/alr-updater.log
```
---
## Запуск
### Ручной запуск

View File

@@ -34,4 +34,13 @@ reposBaseDir = "/var/cache/alr-updater"
[webhook]
# Хэш пароля для webhook. Сгенерируйте его, используя `alr-updater -g`.
pwd_hash = "CHANGE ME"
pwd_hash = "CHANGE ME"
[logging]
# Включить логирование в файл
enable_file = false
# Путь к файлу логов (по умолчанию /var/log/alr-updater.log)
log_file = "/var/log/alr-updater.log"
# Максимальный размер файла логов в байтах (по умолчанию 100MB)
# При достижении этого размера файл будет ротирован
max_size = 104857600

View File

@@ -22,6 +22,7 @@ type Config struct {
ReposBaseDir string `toml:"reposBaseDir" env:"REPOS_BASE_DIR"`
Repositories map[string]GitRepo `toml:"repositories"`
Webhook Webhook `toml:"webhook" envPrefix:"WEBHOOK_"`
Logging Logging `toml:"logging" envPrefix:"LOGGING_"`
}
type GitRepo struct {
@@ -43,3 +44,9 @@ type Commit struct {
type Webhook struct {
PasswordHash string `toml:"pwd_hash" env:"PASSWORD_HASH"`
}
type Logging struct {
LogFile string `toml:"log_file" env:"LOG_FILE"`
MaxSize int64 `toml:"max_size" env:"MAX_SIZE"`
EnableFile bool `toml:"enable_file" env:"ENABLE_FILE"`
}

View File

@@ -0,0 +1,149 @@
/*
* ALR Updater - Automated updater bot for ALR packages
* Copyright (C) 2025 The ALR Authors
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package logger
import (
"fmt"
"io"
"os"
"path/filepath"
"sync"
"time"
)
type RotatingFileWriter struct {
mu sync.Mutex
file *os.File
filename string
maxSize int64
currentSize int64
}
func NewRotatingFileWriter(filename string, maxSize int64) (*RotatingFileWriter, error) {
dir := filepath.Dir(filename)
if err := os.MkdirAll(dir, 0755); err != nil {
return nil, fmt.Errorf("failed to create log directory: %w", err)
}
rfw := &RotatingFileWriter{
filename: filename,
maxSize: maxSize,
}
if err := rfw.openFile(); err != nil {
return nil, err
}
return rfw, nil
}
func (rfw *RotatingFileWriter) openFile() error {
file, err := os.OpenFile(rfw.filename, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0644)
if err != nil {
return fmt.Errorf("failed to open log file: %w", err)
}
info, err := file.Stat()
if err != nil {
file.Close()
return fmt.Errorf("failed to stat log file: %w", err)
}
rfw.file = file
rfw.currentSize = info.Size()
return nil
}
func (rfw *RotatingFileWriter) rotate() error {
if rfw.file != nil {
rfw.file.Close()
}
// Переименовываем текущий файл с временной меткой
backupName := fmt.Sprintf("%s.%s", rfw.filename, time.Now().Format("20060102-150405"))
if err := os.Rename(rfw.filename, backupName); err != nil && !os.IsNotExist(err) {
return fmt.Errorf("failed to rotate log file: %w", err)
}
// Создаем новый файл
if err := rfw.openFile(); err != nil {
return err
}
// Удаляем старые резервные копии (оставляем только последние 5)
pattern := rfw.filename + ".*"
matches, _ := filepath.Glob(pattern)
if len(matches) > 5 {
// Сортировка не требуется, так как имена файлов содержат временную метку
for i := 0; i < len(matches)-5; i++ {
os.Remove(matches[i])
}
}
return nil
}
func (rfw *RotatingFileWriter) Write(p []byte) (n int, err error) {
rfw.mu.Lock()
defer rfw.mu.Unlock()
// Проверяем, нужна ли ротация
if rfw.currentSize+int64(len(p)) > rfw.maxSize {
if err := rfw.rotate(); err != nil {
return 0, err
}
}
n, err = rfw.file.Write(p)
if err != nil {
return n, err
}
rfw.currentSize += int64(n)
return n, nil
}
func (rfw *RotatingFileWriter) Close() error {
rfw.mu.Lock()
defer rfw.mu.Unlock()
if rfw.file != nil {
return rfw.file.Close()
}
return nil
}
// MultiWriter объединяет вывод в несколько writers
type MultiWriter struct {
writers []io.Writer
}
func NewMultiWriter(writers ...io.Writer) *MultiWriter {
return &MultiWriter{writers: writers}
}
func (mw *MultiWriter) Write(p []byte) (n int, err error) {
for _, w := range mw.writers {
n, err = w.Write(p)
if err != nil {
return
}
}
return len(p), nil
}

35
main.go
View File

@@ -21,6 +21,7 @@ package main
import (
"bufio"
"fmt"
"io"
"net/http"
"os"
"path/filepath"
@@ -37,12 +38,15 @@ import (
"gitea.plemya-x.ru/Plemya-x/ALR-updater/internal/builtins"
"gitea.plemya-x.ru/Plemya-x/ALR-updater/internal/config"
"gitea.plemya-x.ru/Plemya-x/ALR-updater/internal/generator"
filelogger "gitea.plemya-x.ru/Plemya-x/ALR-updater/internal/logger"
"go.etcd.io/bbolt"
"go.starlark.net/starlark"
"golang.org/x/crypto/bcrypt"
"golang.org/x/term"
)
var fileWriter *filelogger.RotatingFileWriter
func init() {
log.Logger = logger.NewPretty(os.Stderr)
}
@@ -143,6 +147,37 @@ func main() {
fl.Close()
}
// Настройка логирования в файл
if cfg.Logging.EnableFile {
logFile := cfg.Logging.LogFile
if logFile == "" {
logFile = "/var/log/alr-updater.log"
}
maxSize := cfg.Logging.MaxSize
if maxSize == 0 {
maxSize = 100 * 1024 * 1024 // 100 MB по умолчанию
}
var err error
fileWriter, err = filelogger.NewRotatingFileWriter(logFile, maxSize)
if err != nil {
log.Error("Failed to create file logger, continuing with stderr only").Err(err).Send()
} else {
// Создаем MultiWriter для вывода в stderr и файл одновременно
multiWriter := filelogger.NewMultiWriter(os.Stderr, fileWriter)
log.Logger = logger.NewPretty(multiWriter)
log.Info("File logging enabled").Str("file", logFile).Int64("maxSize", maxSize).Send()
// Закрываем файл при завершении
defer func() {
if fileWriter != nil {
fileWriter.Close()
}
}()
}
}
// Обработка генерации плагинов
if *generatePlugins {
log.Info("Starting automatic plugin generation...")