mirror of
https://github.com/woodpecker-ci/woodpecker.git
synced 2025-10-26 21:31:38 +00:00
added blobstore and capability packages
This commit is contained in:
55
server/blobstore/blobsql/blobstore.go
Normal file
55
server/blobstore/blobsql/blobstore.go
Normal file
@@ -0,0 +1,55 @@
|
||||
package blobsql
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"io"
|
||||
"io/ioutil"
|
||||
|
||||
"github.com/russross/meddler"
|
||||
)
|
||||
|
||||
type Blobstore struct {
|
||||
meddler.DB
|
||||
}
|
||||
|
||||
// Del removes an object from the blobstore.
|
||||
func (b *Blobstore) Del(path string) error {
|
||||
var _, err = b.Exec(deleteBlob, path)
|
||||
return err
|
||||
}
|
||||
|
||||
// Get retrieves an object from the blobstore.
|
||||
func (b *Blobstore) Get(path string) ([]byte, error) {
|
||||
var blob = Blob{}
|
||||
var err = meddler.QueryRow(b, &blob, queryBlob, path)
|
||||
return []byte(blob.Data), err
|
||||
}
|
||||
|
||||
// GetReader retrieves an object from the blobstore.
|
||||
// It is the caller's responsibility to call Close on
|
||||
// the ReadCloser when finished reading.
|
||||
func (b *Blobstore) GetReader(path string) (io.ReadCloser, error) {
|
||||
var blob, err = b.Get(path)
|
||||
var buf = bytes.NewBuffer(blob)
|
||||
return ioutil.NopCloser(buf), err
|
||||
}
|
||||
|
||||
// Put inserts an object into the blobstore.
|
||||
func (b *Blobstore) Put(path string, data []byte) error {
|
||||
var blob = Blob{}
|
||||
meddler.QueryRow(b, &blob, queryBlob, path)
|
||||
blob.Path = path
|
||||
blob.Data = string(data)
|
||||
return meddler.Save(b, tableBlob, &blob)
|
||||
}
|
||||
|
||||
// PutReader inserts an object into the blobstore by
|
||||
// consuming data from r until EOF.
|
||||
func (b *Blobstore) PutReader(path string, r io.Reader) error {
|
||||
var data, _ = ioutil.ReadAll(r)
|
||||
return b.Put(path, data)
|
||||
}
|
||||
|
||||
func New(db meddler.DB) *Blobstore {
|
||||
return &Blobstore{db}
|
||||
}
|
||||
Reference in New Issue
Block a user