messages.sql.go 2.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117
  1. // Code generated by sqlc. DO NOT EDIT.
  2. // versions:
  3. // sqlc v1.27.0
  4. // source: messages.sql
  5. package db
  6. import (
  7. "context"
  8. )
  9. const createMessage = `-- name: CreateMessage :one
  10. INSERT INTO messages (
  11. id,
  12. session_id,
  13. message_data,
  14. created_at,
  15. updated_at
  16. ) VALUES (
  17. ?, ?, ?, strftime('%s', 'now'), strftime('%s', 'now')
  18. )
  19. RETURNING id, session_id, message_data, created_at, updated_at
  20. `
  21. type CreateMessageParams struct {
  22. ID string `json:"id"`
  23. SessionID string `json:"session_id"`
  24. MessageData string `json:"message_data"`
  25. }
  26. func (q *Queries) CreateMessage(ctx context.Context, arg CreateMessageParams) (Message, error) {
  27. row := q.queryRow(ctx, q.createMessageStmt, createMessage, arg.ID, arg.SessionID, arg.MessageData)
  28. var i Message
  29. err := row.Scan(
  30. &i.ID,
  31. &i.SessionID,
  32. &i.MessageData,
  33. &i.CreatedAt,
  34. &i.UpdatedAt,
  35. )
  36. return i, err
  37. }
  38. const deleteMessage = `-- name: DeleteMessage :exec
  39. DELETE FROM messages
  40. WHERE id = ?
  41. `
  42. func (q *Queries) DeleteMessage(ctx context.Context, id string) error {
  43. _, err := q.exec(ctx, q.deleteMessageStmt, deleteMessage, id)
  44. return err
  45. }
  46. const deleteSessionMessages = `-- name: DeleteSessionMessages :exec
  47. DELETE FROM messages
  48. WHERE session_id = ?
  49. `
  50. func (q *Queries) DeleteSessionMessages(ctx context.Context, sessionID string) error {
  51. _, err := q.exec(ctx, q.deleteSessionMessagesStmt, deleteSessionMessages, sessionID)
  52. return err
  53. }
  54. const getMessage = `-- name: GetMessage :one
  55. SELECT id, session_id, message_data, created_at, updated_at
  56. FROM messages
  57. WHERE id = ? LIMIT 1
  58. `
  59. func (q *Queries) GetMessage(ctx context.Context, id string) (Message, error) {
  60. row := q.queryRow(ctx, q.getMessageStmt, getMessage, id)
  61. var i Message
  62. err := row.Scan(
  63. &i.ID,
  64. &i.SessionID,
  65. &i.MessageData,
  66. &i.CreatedAt,
  67. &i.UpdatedAt,
  68. )
  69. return i, err
  70. }
  71. const listMessagesBySession = `-- name: ListMessagesBySession :many
  72. SELECT id, session_id, message_data, created_at, updated_at
  73. FROM messages
  74. WHERE session_id = ?
  75. ORDER BY created_at ASC
  76. `
  77. func (q *Queries) ListMessagesBySession(ctx context.Context, sessionID string) ([]Message, error) {
  78. rows, err := q.query(ctx, q.listMessagesBySessionStmt, listMessagesBySession, sessionID)
  79. if err != nil {
  80. return nil, err
  81. }
  82. defer rows.Close()
  83. items := []Message{}
  84. for rows.Next() {
  85. var i Message
  86. if err := rows.Scan(
  87. &i.ID,
  88. &i.SessionID,
  89. &i.MessageData,
  90. &i.CreatedAt,
  91. &i.UpdatedAt,
  92. ); err != nil {
  93. return nil, err
  94. }
  95. items = append(items, i)
  96. }
  97. if err := rows.Close(); err != nil {
  98. return nil, err
  99. }
  100. if err := rows.Err(); err != nil {
  101. return nil, err
  102. }
  103. return items, nil
  104. }