Something went wrong. Try again.
Something went wrong. Try again.
2.2 kB · 65 lines
Go
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566package model
import ( "context" "errors" "fmt"
"github.com/bluesky-social/indigo/atproto/syntax" glex "github.com/streamplace/glex/runtime" "gorm.io/gorm" "gorm.io/gorm/clause" "stream.place/streamplace/pkg/appbsky" "stream.place/streamplace/pkg/spid")
type BskyProfile struct { URI string `json:"uri" gorm:"primaryKey;column:uri"` CID string `json:"cid" gorm:"column:cid"` RepoDID string `json:"repoDID" gorm:"column:repo_did"` Repo *Repo `json:"repo,omitempty" gorm:"foreignKey:DID;references:RepoDID"` Record *[]byte `json:"record" gorm:"column:record"` WasStreamplace bool `json:"wasStreamplace" gorm:"primaryKey;column:was_streamplace"`}
func (m *DBModel) UpsertBskyProfile(ctx context.Context, aturi syntax.ATURI, recBs []byte, wasStreamplace bool) error { cid, err := spid.GetCIDFromBytes(recBs) if err != nil { return fmt.Errorf("failed to get cid: %w", err) } dbProfile := &BskyProfile{ URI: aturi.String(), CID: cid.String(), RepoDID: aturi.Authority().String(), Record: &recBs, WasStreamplace: wasStreamplace, }
// Use GORM's OnConflict to handle unique/primary conflicts // If a conflict (same PK), then update the relevant fields return m.DB. Clauses( // Conflict columns: uri, was_streamplace (matching primaryKey in struct) clause.OnConflict{ Columns: []clause.Column{{Name: "uri"}, {Name: "was_streamplace"}}, DoUpdates: clause.AssignmentColumns([]string{"cid", "repo_did", "record"}), }, ). Create(dbProfile).Error}
func (m *DBModel) GetBskyProfile(ctx context.Context, did string, wasStreamplace bool) (*appbsky.ActorProfile, error) { var profile BskyProfile err := m.DB.Where("uri = ? AND was_streamplace = ?", fmt.Sprintf("at://%s/app.bsky.actor.profile/self", did), wasStreamplace).First(&profile).Error if errors.Is(err, gorm.ErrRecordNotFound) { return nil, nil } if err != nil { return nil, err } var bskyProfile appbsky.ActorProfile if err := glex.DecodeCBOR(*profile.Record, &bskyProfile); err != nil { return nil, fmt.Errorf("failed to decode profile record: %w", err) } return &bskyProfile, nil}