mirror of
https://github.com/ollama/ollama.git
synced 2025-05-17 23:14:23 +02:00
62 lines
1.5 KiB
Go
62 lines
1.5 KiB
Go
package lifecycle
|
|
|
|
import (
|
|
"fmt"
|
|
"log/slog"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/ollama/ollama/envconfig"
|
|
"github.com/ollama/ollama/logutil"
|
|
)
|
|
|
|
func InitLogging() {
|
|
var logFile *os.File
|
|
var err error
|
|
// Detect if we're a GUI app on windows, and if not, send logs to console
|
|
if os.Stderr.Fd() != 0 {
|
|
// Console app detected
|
|
logFile = os.Stderr
|
|
// TODO - write one-line to the app.log file saying we're running in console mode to help avoid confusion
|
|
} else {
|
|
rotateLogs(AppLogFile)
|
|
logFile, err = os.OpenFile(AppLogFile, os.O_APPEND|os.O_WRONLY|os.O_CREATE, 0o755)
|
|
if err != nil {
|
|
slog.Error(fmt.Sprintf("failed to create server log %v", err))
|
|
return
|
|
}
|
|
}
|
|
|
|
slog.SetDefault(logutil.NewLogger(logFile, envconfig.LogLevel()))
|
|
slog.Info("ollama app started")
|
|
}
|
|
|
|
func rotateLogs(logFile string) {
|
|
if _, err := os.Stat(logFile); os.IsNotExist(err) {
|
|
return
|
|
}
|
|
index := strings.LastIndex(logFile, ".")
|
|
pre := logFile[:index]
|
|
post := "." + logFile[index+1:]
|
|
for i := LogRotationCount; i > 0; i-- {
|
|
older := pre + "-" + strconv.Itoa(i) + post
|
|
newer := pre + "-" + strconv.Itoa(i-1) + post
|
|
if i == 1 {
|
|
newer = pre + post
|
|
}
|
|
if _, err := os.Stat(newer); err == nil {
|
|
if _, err := os.Stat(older); err == nil {
|
|
err := os.Remove(older)
|
|
if err != nil {
|
|
slog.Warn("Failed to remove older log", "older", older, "error", err)
|
|
continue
|
|
}
|
|
}
|
|
err := os.Rename(newer, older)
|
|
if err != nil {
|
|
slog.Warn("Failed to rotate log", "older", older, "newer", newer, "error", err)
|
|
}
|
|
}
|
|
}
|
|
}
|