Added some initial code.
This commit is contained in:
parent
917e3634be
commit
d6ceac0a53
35
entry.go
Normal file
35
entry.go
Normal file
@ -0,0 +1,35 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"os"
|
||||
)
|
||||
|
||||
// Entry represents a file in storage.
|
||||
type Entry struct {
|
||||
URL string `json:"url"`
|
||||
ContentLength int64 `json:"content_length"`
|
||||
ContentType string `json:"content_type"`
|
||||
}
|
||||
|
||||
// LoadEntry loads an entry from disk.
|
||||
func LoadEntry(filename string) (*Entry, error) {
|
||||
f, err := os.Open(filename)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
e := &Entry{}
|
||||
if err = json.NewDecoder(f).Decode(e); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return e, nil
|
||||
}
|
||||
|
||||
// Save writes the entry to disk.
|
||||
func (e *Entry) Save(filename string) error {
|
||||
f, err := os.Create(filename)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return json.NewEncoder(f).Encode(e)
|
||||
}
|
39
storage.go
Normal file
39
storage.go
Normal file
@ -0,0 +1,39 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"crypto/md5"
|
||||
"fmt"
|
||||
"io"
|
||||
"os"
|
||||
"path"
|
||||
"sync"
|
||||
)
|
||||
|
||||
// Storage provides read and write access to items in the cache. In order to
|
||||
// avoid race conditions, adding and testing for entries in the cache are
|
||||
// protected by a mutex.
|
||||
type Storage struct {
|
||||
Directory string
|
||||
mutex sync.Mutex
|
||||
}
|
||||
|
||||
// GetReader returns an io.Reader for the specified URL. If the file does not
|
||||
// exist, both a writer (for downloading the file) and a reader are created.
|
||||
func (s *Storage) GetReader(url string) (io.Reader, error) {
|
||||
var (
|
||||
hash = string(md5.Sum([]byte(url)))
|
||||
jsonFilename = path.Join(s.Directory, fmt.Sprintf("%s.json", hash))
|
||||
dataFilename = path.Join(s.Directory, fmt.Sprintf("%s.data", hash))
|
||||
)
|
||||
s.mutex.Lock()
|
||||
defer s.mutex.Unlock()
|
||||
i, err := os.Stat(jsonFilename)
|
||||
if err != nil {
|
||||
if os.IsNotExist(err) {
|
||||
NewWriter(url, jsonFilename, dataFilename)
|
||||
} else {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
return NewReader(url, jsonFilename, dataFilename)
|
||||
}
|
Loading…
Reference in New Issue
Block a user