src/server/database/image.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

package database

import (
	"database/sql"
)

func getThumbnailOrImage(thumbnail bool) string {
	if thumbnail {
		return "Posts.thumbnail "
	}
	return "Posts.image "
}

func GetImage(uid int, thumbnail bool) ([]byte, error) {
	var db = GetDb()

	var image []byte
	row := db.QueryRow("SELECT "+getThumbnailOrImage(thumbnail)+
		"FROM Posts "+
		"WHERE Posts.uid = ?;", uid)
	if err := row.Scan(&image); err != nil {

		if err == sql.ErrNoRows {
			return nil, nil
		}

		return nil, err
	}

	return image, nil
}