71 lines
1.7 KiB
Go
71 lines
1.7 KiB
Go
package postgres
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"git.dragse.it/anthrove/otter-space-sdk/pkg/models"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// CreateSource creates a pgModels.Source
|
|
func CreateSource(ctx context.Context, db *gorm.DB, anthroveSource *models.Source) error {
|
|
|
|
if anthroveSource.Domain == "" {
|
|
return fmt.Errorf("anthroveSource domain is required")
|
|
}
|
|
|
|
result := db.WithContext(ctx).Where(models.Source{Domain: anthroveSource.Domain}).FirstOrCreate(anthroveSource)
|
|
|
|
if result.Error != nil {
|
|
return result.Error
|
|
}
|
|
|
|
log.WithFields(log.Fields{
|
|
"node_source_url": anthroveSource.Domain,
|
|
"node_source_displayName": anthroveSource.DisplayName,
|
|
"node_source_icon": anthroveSource.Icon,
|
|
}).Trace("database: created source node")
|
|
|
|
return nil
|
|
}
|
|
|
|
// GetAllSource returns a list of all pgModels.Source
|
|
func GetAllSource(ctx context.Context, db *gorm.DB) ([]models.Source, error) {
|
|
var sources []models.Source
|
|
|
|
result := db.WithContext(ctx).Find(&sources)
|
|
|
|
if result.Error != nil {
|
|
return nil, result.Error
|
|
}
|
|
|
|
log.WithFields(log.Fields{
|
|
"tag_amount": result.RowsAffected,
|
|
}).Trace("database: get all source nodes")
|
|
|
|
return sources, nil
|
|
}
|
|
|
|
// GetSourceByURL returns the first source it finds based on the domain
|
|
func GetSourceByURL(ctx context.Context, db *gorm.DB, domain string) (*models.Source, error) {
|
|
var sources models.Source
|
|
|
|
if domain == "" {
|
|
return nil, fmt.Errorf("domain is required")
|
|
}
|
|
|
|
result := db.WithContext(ctx).Where("domain = ?", domain).First(&sources)
|
|
|
|
if result.Error != nil {
|
|
return nil, result.Error
|
|
}
|
|
|
|
log.WithFields(log.Fields{
|
|
"tag_amount": result.RowsAffected,
|
|
}).Trace("database: get all source nodes")
|
|
|
|
return &sources, nil
|
|
}
|