mirrors/imgsrv
1
0
Fork 0
mirror of https://github.com/vbatts/imgsrv.git synced 2024-11-30 12:05:42 +00:00
Code Releases Activity

dbutil: abstract out the last mongo type

Browse source 
Signed-off-by: Vincent Batts <vbatts@hashbangbash.com>
This commit is contained in:
Vincent Batts 2017-04-07 11:56:05 -04:00
parent efa8ac6511
commit b80d34f423
Signed by: vbatts
GPG key ID: 10937E57733F1362
3 changed files with 31 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
dbutil/dbutil.go
View file

@ -1,8 +1,9 @@
package dbutil package dbutil
import ( import (
"io"
"github.com/vbatts/imgsrv/types" "github.com/vbatts/imgsrv/types"
"labix.org/v2/mgo"
) )
// Handles are all the register backing Handlers // Handles are all the register backing Handlers
@ -13,8 +14,8 @@ type Handler interface {
Init(config interface{}) error Init(config interface{}) error
Close() error Close() error
Open(filename string) (*mgo.GridFile, error) Open(filename string) (File, error)
Create(filename string) (*mgo.GridFile, error) Create(filename string) (File, error)
Remove(filename string) error Remove(filename string) error
//HasFileByMd5(md5 string) (exists bool, err error) //HasFileByMd5(md5 string) (exists bool, err error)
@ -31,3 +32,15 @@ type Handler interface {
GetExtensions() (kp []types.IdCount, err error) GetExtensions() (kp []types.IdCount, err error)
GetKeywords() (kp []types.IdCount, err error) GetKeywords() (kp []types.IdCount, err error)
} }
type MetaDataer interface {
GetMeta(result interface{}) (err error)
SetMeta(metadata interface{})
}
type File interface {
io.Reader
io.Writer
io.Closer
MetaDataer
}

4
dbutil/mongo/handle.go
View file

@ -59,12 +59,12 @@ func (h mongoHandle) Close() error {
} }
// pass through for GridFs // pass through for GridFs
func (h mongoHandle) Open(filename string) (file *mgo.GridFile, err error) { func (h mongoHandle) Open(filename string) (file dbutil.File, err error) {
return h.Gfs.Open(strings.ToLower(filename)) return h.Gfs.Open(strings.ToLower(filename))
} }
// pass through for GridFs // pass through for GridFs
func (h mongoHandle) Create(filename string) (file *mgo.GridFile, err error) { func (h mongoHandle) Create(filename string) (file dbutil.File, err error) {
return h.Gfs.Create(strings.ToLower(filename)) return h.Gfs.Create(strings.ToLower(filename))
} }

19
types/types.go
View file

@ -21,31 +21,26 @@ type File struct {
UploadDate time.Time UploadDate time.Time
Length int64 Length int64
Filename string ",omitempty" Filename string ",omitempty"
ContentType string "contentType,omitempty"
} }
func (f *File) SetContentType() { // ContentType guesses the mime-type by the file's extension
f.ContentType = mime.TypeByExtension(filepath.Ext(f.Filename)) func (f *File) ContentType() string {
return mime.TypeByExtension(filepath.Ext(f.Filename))
} }
func (f *File) IsImage() bool { func (f *File) IsImage() bool {
f.SetContentType() return strings.HasPrefix(f.ContentType(), "image")
return strings.HasPrefix(f.ContentType, "image")
} }
func (f *File) IsVideo() bool { func (f *File) IsVideo() bool {
f.SetContentType() return strings.HasPrefix(f.ContentType(), "video")
return strings.HasPrefix(f.ContentType, "video")
} }
func (f *File) IsAudio() bool { func (f *File) IsAudio() bool {
f.SetContentType() return strings.HasPrefix(f.ContentType(), "audio")
return strings.HasPrefix(f.ContentType, "audio")
} }
/* // IdCount structure used for collecting values for a tag cloud
Structure used for collecting values from mongo for a tag cloud
*/
type IdCount struct { type IdCount struct {
Id string "_id" Id string "_id"
Value int Value int