61 lines
1.9 KiB
Go
61 lines
1.9 KiB
Go
package models
|
|
|
|
import (
|
|
"database/sql"
|
|
"time"
|
|
)
|
|
|
|
type Post struct {
|
|
ID int
|
|
ThreadID int
|
|
UserID int
|
|
PostTime time.Time
|
|
EditTime *time.Time
|
|
Content string
|
|
AttachmentHash *int64
|
|
AttachmentName *string
|
|
Title string
|
|
ReplyTo int
|
|
}
|
|
|
|
func GetPostsByThreadID(db *sql.DB, threadID int) ([]Post, error) {
|
|
query := "SELECT id, thread_id, user_id, post_time, edit_time, content, attachment_hash, attachment_name, title, reply_to FROM posts WHERE thread_id = ? ORDER BY post_time ASC"
|
|
rows, err := db.Query(query, threadID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
var posts []Post
|
|
for rows.Next() {
|
|
post := Post{}
|
|
var postTimeStr string
|
|
var editTimeStr sql.NullString
|
|
err := rows.Scan(&post.ID, &post.ThreadID, &post.UserID, &postTimeStr, &editTimeStr, &post.Content, &post.AttachmentHash, &post.AttachmentName, &post.Title, &post.ReplyTo)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
post.PostTime, err = time.Parse("2006-01-02 15:04:05", postTimeStr)
|
|
if err != nil {
|
|
post.PostTime = time.Time{}
|
|
}
|
|
if editTimeStr.Valid {
|
|
editTime, err := time.Parse("2006-01-02 15:04:05", editTimeStr.String)
|
|
if err != nil {
|
|
post.EditTime = nil
|
|
} else {
|
|
post.EditTime = &editTime
|
|
}
|
|
} else {
|
|
post.EditTime = nil
|
|
}
|
|
posts = append(posts, post)
|
|
}
|
|
return posts, nil
|
|
}
|
|
|
|
func CreatePost(db *sql.DB, post Post) error {
|
|
query := "INSERT INTO posts (thread_id, user_id, content, title, reply_to, post_time) VALUES (?, ?, ?, ?, ?, NOW())"
|
|
_, err := db.Exec(query, post.ThreadID, post.UserID, post.Content, post.Title, post.ReplyTo)
|
|
return err
|
|
} |