74 lines
1.7 KiB
Go
74 lines
1.7 KiB
Go
|
package metajson
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"encoding/json"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/andreimarcu/linx-server/backends"
|
||
|
|
)
|
||
|
|
|
||
|
|
type MetadataJSON struct {
|
||
|
|
DeleteKey string `json:"delete_key"`
|
||
|
|
Sha256sum string `json:"sha256sum"`
|
||
|
|
Mimetype string `json:"mimetype"`
|
||
|
|
Size int64 `json:"size"`
|
||
|
|
Expiry int64 `json:"expiry"`
|
||
|
|
ArchiveFiles []string `json:"archive_files,omitempty"`
|
||
|
|
ShortURL string `json:"short_url"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type MetaJSONBackend struct {
|
||
|
|
storage backends.MetaStorageBackend
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m MetaJSONBackend) Put(key string, metadata *backends.Metadata) error {
|
||
|
|
mjson := MetadataJSON{}
|
||
|
|
mjson.DeleteKey = metadata.DeleteKey
|
||
|
|
mjson.Mimetype = metadata.Mimetype
|
||
|
|
mjson.ArchiveFiles = metadata.ArchiveFiles
|
||
|
|
mjson.Sha256sum = metadata.Sha256sum
|
||
|
|
mjson.Expiry = metadata.Expiry.Unix()
|
||
|
|
mjson.Size = metadata.Size
|
||
|
|
mjson.ShortURL = metadata.ShortURL
|
||
|
|
|
||
|
|
byt, err := json.Marshal(mjson)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
if _, err := m.storage.Put(key, bytes.NewBuffer(byt)); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m MetaJSONBackend) Get(key string) (metadata backends.Metadata, err error) {
|
||
|
|
b, err := m.storage.Get(key)
|
||
|
|
if err != nil {
|
||
|
|
return metadata, backends.BadMetadata
|
||
|
|
}
|
||
|
|
|
||
|
|
mjson := MetadataJSON{}
|
||
|
|
|
||
|
|
err = json.Unmarshal(b, &mjson)
|
||
|
|
if err != nil {
|
||
|
|
return metadata, backends.BadMetadata
|
||
|
|
}
|
||
|
|
|
||
|
|
metadata.DeleteKey = mjson.DeleteKey
|
||
|
|
metadata.Mimetype = mjson.Mimetype
|
||
|
|
metadata.ArchiveFiles = mjson.ArchiveFiles
|
||
|
|
metadata.Sha256sum = mjson.Sha256sum
|
||
|
|
metadata.Expiry = time.Unix(mjson.Expiry, 0)
|
||
|
|
metadata.Size = mjson.Size
|
||
|
|
metadata.ShortURL = mjson.ShortURL
|
||
|
|
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewMetaJSONBackend(storage backends.MetaStorageBackend) MetaJSONBackend {
|
||
|
|
return MetaJSONBackend{storage: storage}
|
||
|
|
}
|