mirror of
https://github.com/siderolabs/talos.git
synced 2025-08-30 02:51:11 +02:00
40a5b7c177
This adds generic goroutine runner which simply wraps service as process goroutine. It supports log redirection and basic panic handling. DHCP-related part of the network package was slightly adjusted to run as service with logging updates (to redirect logs to a file) and context canceling. Signed-off-by: Andrey Smirnov <smirnov.andrey@gmail.com>
80 lines
1.6 KiB
Go
80 lines
1.6 KiB
Go
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
package log
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"sync"
|
|
|
|
filechunker "github.com/talos-systems/talos/internal/pkg/chunker/file"
|
|
)
|
|
|
|
var instance = map[string]*Log{}
|
|
var mu sync.Mutex
|
|
|
|
// Log represents the log of a service. It supports streaming of the contents of
|
|
// the log file by way of implementing the chunker.Chunker interface.
|
|
type Log struct {
|
|
Name string
|
|
Path string
|
|
source filechunker.Source
|
|
}
|
|
|
|
// New initializes and registers a log for a service.
|
|
func New(name, rootPath string) (*Log, error) {
|
|
mu.Lock()
|
|
if l, ok := instance[name]; ok {
|
|
mu.Unlock()
|
|
return l, nil
|
|
}
|
|
mu.Unlock()
|
|
|
|
logpath := FormatLogPath(name, rootPath)
|
|
w, err := os.Create(logpath)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("create log file: %s", err.Error())
|
|
}
|
|
|
|
l := &Log{
|
|
Name: name,
|
|
Path: logpath,
|
|
source: w,
|
|
}
|
|
|
|
mu.Lock()
|
|
instance[name] = l
|
|
mu.Unlock()
|
|
|
|
return l, nil
|
|
}
|
|
|
|
// Write implements io.WriteCloser.
|
|
func (l *Log) Write(p []byte) (n int, err error) {
|
|
return l.source.Write(p)
|
|
}
|
|
|
|
// Close implements io.WriteCloser.
|
|
func (l *Log) Close() error {
|
|
mu.Lock()
|
|
delete(instance, l.Name)
|
|
mu.Unlock()
|
|
|
|
return l.source.Close()
|
|
}
|
|
|
|
// Read implements chunker.Chunker.
|
|
func (l *Log) Read(ctx context.Context) <-chan []byte {
|
|
c := filechunker.NewChunker(l.source)
|
|
return c.Read(ctx)
|
|
}
|
|
|
|
// FormatLogPath formats the path the log file.
|
|
func FormatLogPath(p, rootPath string) string {
|
|
return filepath.Join(rootPath, p+".log")
|
|
}
|