54 lines
1.2 KiB
Go
54 lines
1.2 KiB
Go
|
package db
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"lmika.dev/lmika/weiro/models"
|
||
|
|
"lmika.dev/lmika/weiro/providers/db/gen/sqlgen"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (db *Provider) SelectPostsOfSite(ctx context.Context, siteID int64) ([]*models.Post, error) {
|
||
|
|
rows, err := db.queries.SelectPostsOfSite(ctx, siteID)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
posts := make([]*models.Post, len(rows))
|
||
|
|
for i, row := range rows {
|
||
|
|
posts[i] = &models.Post{
|
||
|
|
ID: row.ID,
|
||
|
|
SiteID: row.SiteID,
|
||
|
|
GUID: row.Guid,
|
||
|
|
Title: row.Title,
|
||
|
|
Body: row.Body,
|
||
|
|
Slug: row.Slug,
|
||
|
|
CreatedAt: time.Unix(row.CreatedAt, 0).UTC(),
|
||
|
|
PublishedAt: time.Unix(row.PublishedAt, 0).UTC(),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return posts, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (db *Provider) SavePost(ctx context.Context, post *models.Post) error {
|
||
|
|
if post.ID == 0 {
|
||
|
|
newID, err := db.queries.InsertPost(ctx, sqlgen.InsertPostParams{
|
||
|
|
SiteID: post.SiteID,
|
||
|
|
Guid: post.GUID,
|
||
|
|
Title: post.Title,
|
||
|
|
Body: post.Body,
|
||
|
|
Slug: post.Slug,
|
||
|
|
CreatedAt: post.CreatedAt.Unix(),
|
||
|
|
PublishedAt: post.PublishedAt.Unix(),
|
||
|
|
})
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
post.ID = newID
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// No update query defined in sqlgen yet
|
||
|
|
return nil
|
||
|
|
}
|