You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
85 lines
2.0 KiB
85 lines
2.0 KiB
package main
|
|
|
|
import (
|
|
"log"
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/gofiber/fiber/v2/middleware/logger"
|
|
|
|
_ "github.com/mattn/go-sqlite3"
|
|
"github.com/jmoiron/sqlx"
|
|
sq "github.com/Masterminds/squirrel"
|
|
)
|
|
|
|
type Link struct {
|
|
Id int `db:"id" json:"id"`
|
|
StreamId int `db:"stream_id" json:"stream_id"`
|
|
Url string `db:"url" json:"url"`
|
|
Description string `db:"description" json:"description"`
|
|
}
|
|
|
|
type Stream struct {
|
|
Id int `db:"id" json:"id"`
|
|
Title string `db:"title" json:"title"`
|
|
Description string `db:"description" json:"description"`
|
|
}
|
|
|
|
|
|
func SelectJson[T any](db *sqlx.DB, c *fiber.Ctx, err error, sql string, args ...interface{}) error {
|
|
if(err != nil) {
|
|
log.Fatalln(err)
|
|
}
|
|
var result []T
|
|
|
|
err = db.Select(&result, sql, args...)
|
|
if(err != nil) {
|
|
log.Fatalln(err);
|
|
}
|
|
|
|
return c.JSON(&result);
|
|
}
|
|
|
|
func GetJson[T any](db *sqlx.DB, c *fiber.Ctx, err error, sql string, args ...interface{}) error {
|
|
if(err != nil) {
|
|
log.Fatalln(err)
|
|
}
|
|
var result T
|
|
|
|
err = db.Get(&result, sql, args...)
|
|
if(err != nil) {
|
|
log.Fatalln(err);
|
|
}
|
|
|
|
return c.JSON(&result);
|
|
}
|
|
|
|
func main() {
|
|
log.SetFlags(log.LstdFlags | log.Lshortfile)
|
|
|
|
db, err := sqlx.Connect("sqlite3", "db.sqlite3")
|
|
if err != nil {
|
|
log.Fatalln(err)
|
|
}
|
|
|
|
app := fiber.New()
|
|
app.Use(logger.New())
|
|
|
|
// handler that returns one json from a sql database
|
|
app.Get("/api/stream/", func (c *fiber.Ctx) error {
|
|
sql, args, err := sq.Select("*").From("stream").ToSql()
|
|
return SelectJson[Stream](db, c, err, sql, args...)
|
|
})
|
|
|
|
app.Get("/api/stream/:id", func (c *fiber.Ctx) error {
|
|
sql, args, err := sq.Select("*").From("stream").Where("id", c.Params("id")).ToSql()
|
|
return GetJson[Stream](db, c, err, sql, args...)
|
|
})
|
|
|
|
app.Get("/api/stream/:id/links", func (c *fiber.Ctx) error {
|
|
sql, args, err := sq.Select("*").From("stream_link").Where("stream_id", c.Params("id")).ToSql()
|
|
return SelectJson[Link](db, c, err, sql, args...)
|
|
})
|
|
|
|
app.Static("/", "./public")
|
|
|
|
log.Fatal(app.Listen(":5001"))
|
|
}
|
|
|