101 lines
2.5 KiB
Go
101 lines
2.5 KiB
Go
package maindb
|
|
|
|
import (
|
|
"context"
|
|
|
|
"mstore/app/descr"
|
|
)
|
|
|
|
func (db *Database) InsertBlob(ctx context.Context, layer *descr.Blob) error {
|
|
var err error
|
|
request := `
|
|
INSERT INTO blobs(id, name, reference, mediaType, digest, size, created_at, updated_at)
|
|
VALUES ($1, $2, $3, $4, $5, $6, $7, $8)`
|
|
_, err = db.db.Exec(request,
|
|
layer.ID, layer.Name, layer.Reference, layer.MediaType,
|
|
layer.Digest, layer.Size, layer.CreatedAt, layer.UpdatedAt)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (db *Database) GetBlobByDigest(ctx context.Context, digest string) (bool, descr.Blob, error) {
|
|
var err error
|
|
blobs := make([]descr.Blob, 0)
|
|
res := descr.Blob{}
|
|
exists := false
|
|
request := `SELECT * FROM blobs WHERE digest = $1 LIMIT 1`
|
|
err = db.db.Select(&blobs, request, digest)
|
|
if err != nil {
|
|
return exists, res, err
|
|
}
|
|
if len(blobs) > 0 {
|
|
res = blobs[0]
|
|
exists = true
|
|
}
|
|
return exists, res, err
|
|
}
|
|
|
|
func (db *Database) ListAllBlobs(ctx context.Context) ([]descr.Blob, error) {
|
|
var err error
|
|
blobs := make([]descr.Blob, 0)
|
|
request := `SELECT * FROM blobs`
|
|
err = db.db.Select(&blobs, request)
|
|
if err != nil {
|
|
return blobs, err
|
|
}
|
|
return blobs, err
|
|
}
|
|
|
|
func (db *Database) xxxBlobExists(ctx context.Context, name, reference, digest string) (bool, error) {
|
|
var err error
|
|
blobs := make([]descr.Blob, 0)
|
|
request := `
|
|
SELECT * FROM blobs WHERE name = $1 AND reference = $2 AND digest = $3 LIMIT 1`
|
|
err = db.db.Select(&blobs, request, name, reference, digest)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
if len(blobs) > 0 {
|
|
return true, err
|
|
}
|
|
return false, err
|
|
}
|
|
|
|
func (db *Database) GetBlobsByReferense(ctx context.Context, name, reference string) ([]descr.Blob, error) {
|
|
var err error
|
|
blobs := make([]descr.Blob, 0)
|
|
request := `
|
|
SELECT * FROM blobs WHERE name = $1 AND reference = $2`
|
|
err = db.db.Select(&blobs, request, name, reference)
|
|
if err != nil {
|
|
return blobs, err
|
|
}
|
|
return blobs, err
|
|
}
|
|
|
|
func (db *Database) GetBlobUsage(ctx context.Context, digest string) (int64, error) {
|
|
var err error
|
|
var usage int64
|
|
count := make([]int64, 0)
|
|
request := `
|
|
SELECT count(id) AS count FROM blobs WHERE digest = $1`
|
|
err = db.db.Select(&count, request, digest)
|
|
if err != nil {
|
|
return usage, err
|
|
}
|
|
usage = count[0]
|
|
return usage, err
|
|
}
|
|
|
|
func (db *Database) DeleteBlobByDigest(ctx context.Context, digest string) error {
|
|
var err error
|
|
request := `DELETE FROM blobs WHERE digest = $1;`
|
|
_, err = db.db.Exec(request, digest)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return err
|
|
}
|