135 lines
4.9 KiB
Go
135 lines
4.9 KiB
Go
// Copyright 2019 clair authors
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package pgsql
|
|
|
|
import (
|
|
"database/sql"
|
|
"time"
|
|
|
|
"github.com/coreos/clair/database/pgsql/keyvalue"
|
|
"github.com/coreos/clair/database/pgsql/vulnerability"
|
|
|
|
"github.com/coreos/clair/database"
|
|
"github.com/coreos/clair/database/pgsql/ancestry"
|
|
"github.com/coreos/clair/database/pgsql/detector"
|
|
"github.com/coreos/clair/database/pgsql/feature"
|
|
"github.com/coreos/clair/database/pgsql/layer"
|
|
"github.com/coreos/clair/database/pgsql/lock"
|
|
"github.com/coreos/clair/database/pgsql/namespace"
|
|
"github.com/coreos/clair/database/pgsql/notification"
|
|
"github.com/coreos/clair/pkg/pagination"
|
|
)
|
|
|
|
// Enforce the interface at compile time.
|
|
var _ database.Session = &pgSession{}
|
|
|
|
type pgSession struct {
|
|
*sql.Tx
|
|
|
|
key pagination.Key
|
|
}
|
|
|
|
func (tx *pgSession) UpsertAncestry(a database.Ancestry) error {
|
|
return ancestry.UpsertAncestry(tx.Tx, a)
|
|
}
|
|
|
|
func (tx *pgSession) FindAncestry(name string) (database.Ancestry, bool, error) {
|
|
return ancestry.FindAncestry(tx.Tx, name)
|
|
}
|
|
|
|
func (tx *pgSession) PersistDetectors(detectors []database.Detector) error {
|
|
return detector.PersistDetectors(tx.Tx, detectors)
|
|
}
|
|
|
|
func (tx *pgSession) PersistFeatures(features []database.Feature) error {
|
|
return feature.PersistFeatures(tx.Tx, features)
|
|
}
|
|
|
|
func (tx *pgSession) PersistNamespacedFeatures(features []database.NamespacedFeature) error {
|
|
return feature.PersistNamespacedFeatures(tx.Tx, features)
|
|
}
|
|
|
|
func (tx *pgSession) CacheAffectedNamespacedFeatures(features []database.NamespacedFeature) error {
|
|
return vulnerability.CacheAffectedNamespacedFeatures(tx.Tx, features)
|
|
}
|
|
|
|
func (tx *pgSession) FindAffectedNamespacedFeatures(features []database.NamespacedFeature) ([]database.NullableAffectedNamespacedFeature, error) {
|
|
return vulnerability.FindAffectedNamespacedFeatures(tx.Tx, features)
|
|
}
|
|
|
|
func (tx *pgSession) PersistNamespaces(namespaces []database.Namespace) error {
|
|
return namespace.PersistNamespaces(tx.Tx, namespaces)
|
|
}
|
|
|
|
func (tx *pgSession) PersistLayer(hash string, features []database.LayerFeature, namespaces []database.LayerNamespace, detectedBy []database.Detector) error {
|
|
return layer.PersistLayer(tx.Tx, hash, features, namespaces, detectedBy)
|
|
}
|
|
|
|
func (tx *pgSession) FindLayer(hash string) (database.Layer, bool, error) {
|
|
return layer.FindLayer(tx.Tx, hash)
|
|
}
|
|
|
|
func (tx *pgSession) InsertVulnerabilities(vulns []database.VulnerabilityWithAffected) error {
|
|
return vulnerability.InsertVulnerabilities(tx.Tx, vulns)
|
|
}
|
|
|
|
func (tx *pgSession) FindVulnerabilities(ids []database.VulnerabilityID) ([]database.NullableVulnerability, error) {
|
|
return vulnerability.FindVulnerabilities(tx.Tx, ids)
|
|
}
|
|
|
|
func (tx *pgSession) DeleteVulnerabilities(ids []database.VulnerabilityID) error {
|
|
return vulnerability.DeleteVulnerabilities(tx.Tx, ids)
|
|
}
|
|
|
|
func (tx *pgSession) InsertVulnerabilityNotifications(notifications []database.VulnerabilityNotification) error {
|
|
return notification.InsertVulnerabilityNotifications(tx.Tx, notifications)
|
|
}
|
|
|
|
func (tx *pgSession) FindNewNotification(notifiedBefore time.Time) (hook database.NotificationHook, found bool, err error) {
|
|
return notification.FindNewNotification(tx.Tx, notifiedBefore)
|
|
}
|
|
|
|
func (tx *pgSession) FindVulnerabilityNotification(name string, limit int, oldVulnerabilityPage pagination.Token, newVulnerabilityPage pagination.Token) (noti database.VulnerabilityNotificationWithVulnerable, found bool, err error) {
|
|
return notification.FindVulnerabilityNotification(tx.Tx, name, limit, oldVulnerabilityPage, newVulnerabilityPage, tx.key)
|
|
}
|
|
|
|
func (tx *pgSession) MarkNotificationAsRead(name string) error {
|
|
return notification.MarkNotificationAsRead(tx.Tx, name)
|
|
}
|
|
|
|
func (tx *pgSession) DeleteNotification(name string) error {
|
|
return notification.DeleteNotification(tx.Tx, name)
|
|
}
|
|
|
|
func (tx *pgSession) UpdateKeyValue(key, value string) error {
|
|
return keyvalue.UpdateKeyValue(tx.Tx, key, value)
|
|
}
|
|
|
|
func (tx *pgSession) FindKeyValue(key string) (value string, found bool, err error) {
|
|
return keyvalue.FindKeyValue(tx.Tx, key)
|
|
}
|
|
|
|
func (tx *pgSession) AcquireLock(name, owner string, duration time.Duration) (acquired bool, expiration time.Time, err error) {
|
|
return lock.AcquireLock(tx.Tx, name, owner, duration)
|
|
}
|
|
|
|
func (tx *pgSession) ExtendLock(name, owner string, duration time.Duration) (extended bool, expiration time.Time, err error) {
|
|
return lock.ExtendLock(tx.Tx, name, owner, duration)
|
|
}
|
|
|
|
func (tx *pgSession) ReleaseLock(name, owner string) error {
|
|
return lock.ReleaseLock(tx.Tx, name, owner)
|
|
}
|