Torjus Håkestad
17a484db91
All checks were successful
ci/woodpecker/push/woodpecker Pipeline was successful
86 lines
1.3 KiB
Go
86 lines
1.3 KiB
Go
package files
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
type fileData struct {
|
|
ID string
|
|
Body bytes.Buffer
|
|
|
|
MaxViews uint
|
|
ExpiresOn time.Time
|
|
FileSize int64
|
|
}
|
|
|
|
type MemoryFileStore struct {
|
|
lock sync.RWMutex
|
|
data map[string]*fileData
|
|
}
|
|
|
|
func NewMemoryFileStore() *MemoryFileStore {
|
|
return &MemoryFileStore{
|
|
data: make(map[string]*fileData),
|
|
}
|
|
}
|
|
|
|
func (s *MemoryFileStore) Store(f *File) error {
|
|
|
|
data := &fileData{
|
|
ID: f.ID,
|
|
MaxViews: f.MaxViews,
|
|
ExpiresOn: f.ExpiresOn,
|
|
}
|
|
|
|
n, err := io.Copy(&data.Body, f.Body)
|
|
_ = f.Body.Close()
|
|
|
|
data.FileSize = n
|
|
|
|
s.lock.Lock()
|
|
defer s.lock.Unlock()
|
|
|
|
s.data[f.ID] = data
|
|
return err
|
|
}
|
|
|
|
func (s *MemoryFileStore) Get(id string) (*File, error) {
|
|
s.lock.RLock()
|
|
defer s.lock.RUnlock()
|
|
|
|
fd, ok := s.data[id]
|
|
if !ok {
|
|
return nil, fmt.Errorf("no such item")
|
|
}
|
|
f := &File{
|
|
ID: fd.ID,
|
|
MaxViews: fd.MaxViews,
|
|
ExpiresOn: fd.ExpiresOn,
|
|
Body: io.NopCloser(&fd.Body),
|
|
}
|
|
|
|
return f, nil
|
|
}
|
|
|
|
func (s *MemoryFileStore) Delete(id string) error {
|
|
s.lock.Lock()
|
|
defer s.lock.Unlock()
|
|
delete(s.data, id)
|
|
return nil
|
|
}
|
|
|
|
func (s *MemoryFileStore) List() ([]string, error) {
|
|
var ids []string
|
|
|
|
s.lock.RLock()
|
|
defer s.lock.RUnlock()
|
|
for id := range s.data {
|
|
ids = append(ids, id)
|
|
}
|
|
return ids, nil
|
|
}
|