mirror of
https://git.sr.ht/~sircmpwn/tokidoki
synced 2025-12-12 14:17:21 +01:00
storage: streamline ETag calculation
This commit introduces some helpers so that ETags can be calculated at the same time that files get read or written. Besides looking nicer, it should also help reduce lock contention around file access, as files do not need to be opened twice anymore.
This commit is contained in:
parent
fe0a0d0d00
commit
665b206709
4 changed files with 89 additions and 54 deletions
|
|
@ -47,22 +47,23 @@ func (b *filesystemBackend) safeLocalCalDAVPath(ctx context.Context, urlPath str
|
|||
return b.safeLocalPath(homeSetPath, urlPath)
|
||||
}
|
||||
|
||||
func calendarFromFile(path string, propFilter []string) (*ical.Calendar, error) {
|
||||
func calendarAndEtagFromFile(path string, propFilter []string) (*ical.Calendar, string, error) {
|
||||
f, err := os.Open(path)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
return nil, "", err
|
||||
}
|
||||
defer f.Close()
|
||||
|
||||
dec := ical.NewDecoder(f)
|
||||
src := NewETagReader(f)
|
||||
dec := ical.NewDecoder(src)
|
||||
cal, err := dec.Decode()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
return nil, "", err
|
||||
}
|
||||
|
||||
return cal, nil
|
||||
// TODO implement
|
||||
//return icalPropFilter(cal, propFilter), nil
|
||||
return cal, src.ETag(), nil
|
||||
}
|
||||
|
||||
func (b *filesystemBackend) loadAllCalendarObjects(ctx context.Context, urlPath string, propFilter []string) ([]caldav.CalendarObject, error) {
|
||||
|
|
@ -88,17 +89,12 @@ func (b *filesystemBackend) loadAllCalendarObjects(ctx context.Context, urlPath
|
|||
b.RLock(filename)
|
||||
defer b.RUnlock(filename)
|
||||
|
||||
cal, err := calendarFromFile(filename, propFilter)
|
||||
cal, etag, err := calendarAndEtagFromFile(filename, propFilter)
|
||||
if err != nil {
|
||||
fmt.Printf("load calendar error for %s: %v\n", filename, err)
|
||||
return err
|
||||
}
|
||||
|
||||
etag, err := etagForFile(filename)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// TODO can this potentially be called on a calendar object resource?
|
||||
// Would work (as Walk() includes root), except for the path construction below
|
||||
obj := caldav.CalendarObject{
|
||||
|
|
@ -108,6 +104,7 @@ func (b *filesystemBackend) loadAllCalendarObjects(ctx context.Context, urlPath
|
|||
ETag: etag,
|
||||
Data: cal,
|
||||
}
|
||||
log.Debug().Str("path", obj.Path).Str("etag", etag).Int64("size", info.Size()).Msg("calendar object loaded")
|
||||
result = append(result, obj)
|
||||
return nil
|
||||
})
|
||||
|
|
@ -268,17 +265,12 @@ func (b *filesystemBackend) GetCalendarObject(ctx context.Context, objPath strin
|
|||
propFilter = req.Props
|
||||
}
|
||||
|
||||
calendar, err := calendarFromFile(localPath, propFilter)
|
||||
calendar, etag, err := calendarAndEtagFromFile(localPath, propFilter)
|
||||
if err != nil {
|
||||
log.Debug().Str("path", localPath).Err(err).Msg("error reading calendar")
|
||||
return nil, err
|
||||
}
|
||||
|
||||
etag, err := etagForFile(localPath)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
obj := caldav.CalendarObject{
|
||||
Path: objPath,
|
||||
ModTime: info.ModTime(),
|
||||
|
|
@ -286,6 +278,7 @@ func (b *filesystemBackend) GetCalendarObject(ctx context.Context, objPath strin
|
|||
ETag: etag,
|
||||
Data: calendar,
|
||||
}
|
||||
log.Debug().Str("path", objPath).Str("etag", etag).Int64("size", info.Size()).Msg("returning calendar object")
|
||||
return &obj, nil
|
||||
}
|
||||
|
||||
|
|
@ -373,16 +366,13 @@ func (b *filesystemBackend) PutCalendarObject(ctx context.Context, objPath strin
|
|||
}
|
||||
defer f.Close()
|
||||
|
||||
enc := ical.NewEncoder(f)
|
||||
out := NewETagWriter(f)
|
||||
enc := ical.NewEncoder(out)
|
||||
err = enc.Encode(calendar)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
etag, err := etagForFile(localPath)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
info, err := f.Stat()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
|
|
@ -392,9 +382,10 @@ func (b *filesystemBackend) PutCalendarObject(ctx context.Context, objPath strin
|
|||
Path: objPath,
|
||||
ModTime: info.ModTime(),
|
||||
ContentLength: info.Size(),
|
||||
ETag: etag,
|
||||
ETag: out.ETag(),
|
||||
Data: calendar,
|
||||
}
|
||||
log.Debug().Str("path", r.Path).Str("etag", r.ETag).Int64("size", info.Size()).Msg("calendar object updated")
|
||||
return &r, nil
|
||||
}
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue