woodpecker/server/services/log/file/file.go
qwerty287 c72468478d
Allow to store logs in files (#3568)
Co-authored-by: Anbraten <6918444+anbraten@users.noreply.github.com>
2024-06-06 14:34:57 +02:00

82 lines
1.6 KiB
Go

package file
import (
"bufio"
"encoding/json"
"fmt"
"os"
"path/filepath"
"strings"
"go.woodpecker-ci.org/woodpecker/v2/server/model"
"go.woodpecker-ci.org/woodpecker/v2/server/services/log"
)
type logStore struct {
base string
}
func NewLogStore(base string) (log.Service, error) {
if base == "" {
return nil, fmt.Errorf("file storage base path is required")
}
if _, err := os.Stat(base); err != nil && os.IsNotExist(err) {
err = os.MkdirAll(base, 0o600)
if err != nil {
return nil, err
}
}
return logStore{base: base}, nil
}
func (l logStore) filePath(id int64) string {
return filepath.Join(l.base, fmt.Sprintf("%d.json", id))
}
func (l logStore) LogFind(step *model.Step) ([]*model.LogEntry, error) {
filename := l.filePath(step.ID)
file, err := os.Open(filename)
if err != nil {
if os.IsNotExist(err) {
return nil, nil
}
return nil, err
}
s := bufio.NewScanner(file)
var entries []*model.LogEntry
for s.Scan() {
j := s.Text()
if len(strings.TrimSpace(j)) == 0 {
continue
}
entry := &model.LogEntry{}
err = json.Unmarshal([]byte(j), entry)
if err != nil {
return nil, err
}
entries = append(entries, entry)
}
return entries, nil
}
func (l logStore) LogAppend(logEntry *model.LogEntry) error {
file, err := os.OpenFile(l.filePath(logEntry.StepID), os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0o600)
if err != nil {
return err
}
jsonData, err := json.Marshal(logEntry)
if err != nil {
return err
}
_, err = file.Write(append(jsonData, byte('\n')))
if err != nil {
return err
}
return file.Close()
}
func (l logStore) LogDelete(step *model.Step) error {
return os.Remove(l.filePath(step.ID))
}