1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283 |
- package sqlite
- import (
- "context"
- "strings"
- storepb "github.com/usememos/memos/proto/gen/store"
- "github.com/usememos/memos/store"
- )
- func (d *DB) UpsertReaction(ctx context.Context, upsert *store.Reaction) (*store.Reaction, error) {
- fields := []string{"`creator_id`", "`content_id`", "`reaction_type`"}
- placeholder := []string{"?", "?", "?"}
- args := []interface{}{upsert.CreatorID, upsert.ContentID, upsert.ReactionType.String()}
- stmt := "INSERT INTO `reaction` (" + strings.Join(fields, ", ") + ") VALUES (" + strings.Join(placeholder, ", ") + ") RETURNING `id`, `created_ts`"
- if err := d.db.QueryRowContext(ctx, stmt, args...).Scan(
- &upsert.ID,
- &upsert.CreatedTs,
- ); err != nil {
- return nil, err
- }
- reaction := upsert
- return reaction, nil
- }
- func (d *DB) ListReactions(ctx context.Context, find *store.FindReaction) ([]*store.Reaction, error) {
- where, args := []string{"1 = 1"}, []interface{}{}
- if find.ID != nil {
- where, args = append(where, "id = ?"), append(args, *find.ID)
- }
- if find.CreatorID != nil {
- where, args = append(where, "creator_id = ?"), append(args, *find.CreatorID)
- }
- if find.ContentID != nil {
- where, args = append(where, "content_id = ?"), append(args, *find.ContentID)
- }
- rows, err := d.db.QueryContext(ctx, `
- SELECT
- id,
- created_ts,
- creator_id,
- content_id,
- reaction_type
- FROM reaction
- WHERE `+strings.Join(where, " AND ")+`
- ORDER BY id ASC`,
- args...,
- )
- if err != nil {
- return nil, err
- }
- defer rows.Close()
- list := []*store.Reaction{}
- for rows.Next() {
- reaction := &store.Reaction{}
- var reactionType string
- if err := rows.Scan(
- &reaction.ID,
- &reaction.CreatedTs,
- &reaction.CreatorID,
- &reaction.ContentID,
- &reactionType,
- ); err != nil {
- return nil, err
- }
- reaction.ReactionType = storepb.ReactionType(storepb.ReactionType_value[reactionType])
- list = append(list, reaction)
- }
- if err := rows.Err(); err != nil {
- return nil, err
- }
- return list, nil
- }
- func (d *DB) DeleteReaction(ctx context.Context, delete *store.DeleteReaction) error {
- _, err := d.db.ExecContext(ctx, "DELETE FROM `reaction` WHERE `id` = ?", delete.ID)
- return err
- }
|