working commit
This commit is contained in:
+27
-42
@@ -45,22 +45,27 @@ func (db *Database) InsertManifestWithLayers(ctx context.Context, manifest *desc
|
||||
tx, err := db.db.BeginTx(ctx, nil)
|
||||
// Manifest
|
||||
request = `
|
||||
INSERT INTO manifests(id, name, reference, contentType, payload, digest, created_at, updated_at)
|
||||
VALUES ($1, $2, $3, $4, $5, $6, $7, $8);`
|
||||
INSERT INTO manifests(id, name, reference, contentType, payload, digest,
|
||||
created_at, updated_at, created_by, updated_by)
|
||||
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10);`
|
||||
_, err = tx.Exec(request, manifest.ID, manifest.Name, manifest.Reference,
|
||||
manifest.ContentType, manifest.Payload, manifest.Digest,
|
||||
manifest.CreatedAt, manifest.UpdatedAt)
|
||||
manifest.CreatedAt, manifest.UpdatedAt,
|
||||
manifest.CreatedBy, manifest.UpdatedBy)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
// Layers
|
||||
for _, layer := range layers {
|
||||
request = `
|
||||
INSERT INTO blobs(id, name, reference, mediaType, digest, size, created_at, updated_at)
|
||||
VALUES ($1, $2, $3, $4, $5, $6, $7, $8)`
|
||||
INSERT INTO blobs(id, name, reference, mediaType, digest, size,
|
||||
created_at, updated_at, created_by, updated_by)
|
||||
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10)`
|
||||
_, err = tx.Exec(request,
|
||||
layer.ID, layer.Name, layer.Reference, layer.MediaType,
|
||||
layer.Digest, layer.Size, layer.CreatedAt, layer.UpdatedAt)
|
||||
layer.Digest, layer.Size,
|
||||
layer.CreatedAt, layer.UpdatedAt,
|
||||
layer.CreatedBy, layer.UpdatedBy)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
@@ -74,7 +79,8 @@ func (db *Database) InsertManifestWithLayers(ctx context.Context, manifest *desc
|
||||
return err
|
||||
}
|
||||
|
||||
func (db *Database) UpdateManifestWithBlobs(ctx context.Context, manifest *descr.Manifest, newLayers []descr.Blob, oldLayers []descr.Blob) error {
|
||||
func (db *Database) UpdateManifestWithBlobs(ctx context.Context, manifest *descr.Manifest,
|
||||
newLayers []descr.Blob, oldLayers []descr.Blob) error {
|
||||
var err error
|
||||
var request string
|
||||
|
||||
@@ -83,20 +89,19 @@ func (db *Database) UpdateManifestWithBlobs(ctx context.Context, manifest *descr
|
||||
// TODO: tx.Rollback()
|
||||
|
||||
// Manifest
|
||||
request = `
|
||||
UPDATE manifests
|
||||
SET contentType = $1, payload = $2, digest = $3, updated_at = $4
|
||||
WHERE name = $5 AND reference = $6`
|
||||
request = `UPDATE manifests SET contentType = $1, payload = $2, digest = $3, updated_at = $4, updated_by = $5
|
||||
WHERE name = $6 AND reference = $7`
|
||||
_, err = tx.Exec(request, manifest.ContentType, manifest.Payload, manifest.Digest,
|
||||
manifest.UpdatedAt, manifest.Name, manifest.Reference)
|
||||
manifest.UpdatedAt, manifest.UpdatedBy,
|
||||
manifest.Name, manifest.Reference)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
// New layers
|
||||
for _, layer := range newLayers {
|
||||
request = `
|
||||
INSERT INTO blobs(id, name, reference, mediaType, digest, size, created_at, updated_at)
|
||||
VALUES ($1, $2, $3, $4, $5, $6, $7, $8)`
|
||||
request = `INSERT INTO blobs(id, name, reference, mediaType, digest, size,
|
||||
created_at, updated_at, created_by, updated_by)
|
||||
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10)`
|
||||
_, err = tx.Exec(request, layer.ID, layer.Name, layer.Reference, layer.MediaType,
|
||||
layer.Digest, layer.Size, layer.CreatedAt, layer.UpdatedAt)
|
||||
if err != nil {
|
||||
@@ -105,11 +110,7 @@ func (db *Database) UpdateManifestWithBlobs(ctx context.Context, manifest *descr
|
||||
}
|
||||
// Old layers
|
||||
for _, layer := range oldLayers {
|
||||
request = `
|
||||
DELETE FROM blobs
|
||||
WHERE name == $1
|
||||
AND reference == $2
|
||||
AND digest == $3`
|
||||
request = `DELETE FROM blobs WHERE name == $1 AND reference == $2 AND digest == $3`
|
||||
_, err = tx.Exec(request, layer.Name, layer.Reference, layer.Digest)
|
||||
}
|
||||
// Commit
|
||||
@@ -124,11 +125,7 @@ func (db *Database) ManifestExistsByReference(ctx context.Context, name, referen
|
||||
var err error
|
||||
var count int
|
||||
var exists bool
|
||||
request := `
|
||||
SELECT count(id) AS count FROM manifests
|
||||
WHERE name = $1
|
||||
AND reference = $2
|
||||
LIMIT 1`
|
||||
request := `SELECT count(id) AS count FROM manifests WHERE name = $1 AND reference = $2 LIMIT 1`
|
||||
err = db.db.Get(&count, request, name, reference)
|
||||
if err != nil {
|
||||
return exists, err
|
||||
@@ -143,9 +140,7 @@ func (db *Database) ManifestExistsByDigest(ctx context.Context, name, digest str
|
||||
var err error
|
||||
var count int
|
||||
var exists bool
|
||||
request := `
|
||||
SELECT count(id) AS count FROM manifests
|
||||
WHERE name = $1 AND digest = $2 LIMIT 1`
|
||||
request := `SELECT count(id) AS count FROM manifests WHERE name = $1 AND digest = $2 LIMIT 1`
|
||||
err = db.db.Get(&count, request, name, digest)
|
||||
if err != nil {
|
||||
return exists, err
|
||||
@@ -161,10 +156,7 @@ func (db *Database) GetManifestByDigest(ctx context.Context, name, digest string
|
||||
manifest := descr.Manifest{}
|
||||
manifests := make([]descr.Manifest, 0)
|
||||
exists := false
|
||||
request := `
|
||||
SELECT * FROM manifests
|
||||
WHERE name = $1 AND digest = $2
|
||||
LIMIT 1`
|
||||
request := `SELECT * FROM manifests WHERE name = $1 AND digest = $2 LIMIT 1`
|
||||
err = db.db.Select(&manifests, request, name, digest)
|
||||
if err != nil {
|
||||
return exists, manifest, err
|
||||
@@ -181,10 +173,7 @@ func (db *Database) GetManifestByReference(ctx context.Context, name, reference
|
||||
manifest := descr.Manifest{}
|
||||
exists := false
|
||||
manifests := make([]descr.Manifest, 0)
|
||||
request := `
|
||||
SELECT * FROM manifests
|
||||
WHERE name = $1 AND reference = $2
|
||||
LIMIT 1`
|
||||
request := `SELECT * FROM manifests WHERE name = $1 AND reference = $2 LIMIT 1`
|
||||
err = db.db.Select(&manifests, request, name, reference)
|
||||
if err != nil {
|
||||
return exists, manifest, err
|
||||
@@ -200,9 +189,7 @@ func (db *Database) GetManifestByReference(ctx context.Context, name, reference
|
||||
func (db *Database) ListManifestsByName(ctx context.Context, name string) ([]descr.Manifest, error) {
|
||||
var err error
|
||||
manifests := make([]descr.Manifest, 0)
|
||||
request := `
|
||||
SELECT id, name, reference, contentType, payload FROM manifests
|
||||
WHERE name = $1`
|
||||
request := `SELECT id, name, reference, contentType, payload FROM manifests WHERE name = $1`
|
||||
err = db.db.Select(&manifests, request, name)
|
||||
if err != nil {
|
||||
return manifests, err
|
||||
@@ -241,9 +228,7 @@ func (db *Database) DeleteManifest(ctx context.Context, name, reference string)
|
||||
// TODO: tx.Rollback()
|
||||
|
||||
// Blobs
|
||||
request = `
|
||||
DELETE FROM blobs
|
||||
WHERE name = $1 AND reference = $2`
|
||||
request = `DELETE FROM blobs WHERE name = $1 AND reference = $2`
|
||||
_, err = tx.Exec(request, name, reference)
|
||||
if err != nil {
|
||||
return err
|
||||
|
||||
Reference in New Issue
Block a user