37 lines
839 B
Go
37 lines
839 B
Go
package pds
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/uabluerail/indexer/models"
|
|
)
|
|
|
|
const Unknown models.ID = 0
|
|
|
|
type PDS struct {
|
|
ID models.ID `gorm:"primarykey"`
|
|
CreatedAt time.Time
|
|
UpdatedAt time.Time
|
|
Host string `gorm:"uniqueIndex"`
|
|
Cursor int64
|
|
FirstCursorSinceReset int64
|
|
LastList time.Time
|
|
CrawlLimit int
|
|
}
|
|
|
|
func AutoMigrate(db *gorm.DB) error {
|
|
return db.AutoMigrate(&PDS{})
|
|
}
|
|
|
|
func EnsureExists(ctx context.Context, db *gorm.DB, host string) (*PDS, error) {
|
|
remote := PDS{Host: host}
|
|
if err := db.Model(&remote).Where(&PDS{Host: host}).FirstOrCreate(&remote).Error; err != nil {
|
|
return nil, fmt.Errorf("failed to get PDS record from DB for %q: %w", remote.Host, err)
|
|
}
|
|
return &remote, nil
|
|
}
|