speatzle/morffix
1
0
Fork 0
Code Issues Pull requests Projects Releases 2 Packages Wiki Activity Actions

Rework chart gen. Add Health, Tanscode, Task Status Charts
All checks were successful
/ release (push) Successful in 32s
Details

Browse source
This commit is contained in:
Samuel Lorch 2024-07-13 23:18:47 +02:00
parent 091ef322d5
commit 8e15cdd6e8
3 changed files with 164 additions and 81 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

157
server/stats.go
View file

@ -2,9 +2,12 @@ package server
import (
"bytes"
"context"
"fmt"
"html/template"
"log/slog"
"net/http"
"strings"
"git.lastassault.de/speatzle/morffix/constants"
"github.com/go-echarts/go-echarts/v2/charts"
@ -13,40 +16,25 @@ import (
)
type StatsDisplay struct {
CodecCounts []CodecCount
Charts []ChartData
}
type ChartData struct {
Element template.HTML
Script template.HTML
}
type CodecCount struct {
Codec string
Count int
type PieValue struct {
Name string
Value int
}
func handleStats(w http.ResponseWriter, r *http.Request) {
data := StatsDisplay{}
type PieIntValue struct {
Id int
Value int
}
rows, err := db.Query(r.Context(), `SELECT COALESCE(jsonb_path_query_first(ffprobe_data, '$.streams[*] ? (@.codec_type == "video") ? (@.disposition.attached_pic == 0).codec_name')::text, 'Unknown') AS codec, COUNT(*) AS count FROM files WHERE ffprobe_data IS NOT NULL GROUP BY jsonb_path_query_first(ffprobe_data, '$.streams[*] ? (@.codec_type == "video") ? (@.disposition.attached_pic == 0).codec_name');`)
if err != nil {
slog.ErrorContext(r.Context(), "Query Stats", "err", err)
http.Error(w, "Error Query Stats: "+err.Error(), http.StatusInternalServerError)
return
}
codecCounts, err := pgx.CollectRows[CodecCount](rows, pgx.RowToStructByName[CodecCount])
if err != nil {
slog.ErrorContext(r.Context(), "Collect Rows", "err", err)
http.Error(w, "Error Query Libraries: "+err.Error(), http.StatusInternalServerError)
return
}
data.CodecCounts = codecCounts
chartData := []opts.PieData{}
for _, c := range codecCounts {
chartData = append(chartData, opts.PieData{
Name: c.Codec,
Value: c.Count,
})
}
func generatePie(name string, data []opts.PieData) ChartData {
pie := charts.NewPie()
pie.SetGlobalOptions(
@ -55,10 +43,10 @@ func handleStats(w http.ResponseWriter, r *http.Request) {
BackgroundColor: "#111",
}),
charts.WithTitleOpts(opts.Title{
Title: "Codecs",
Title: name,
}))
pie.AddSeries("Codecs", chartData).SetSeriesOptions(
pie.AddSeries(name, data).SetSeriesOptions(
charts.WithLabelOpts(
opts.Label{
Show: opts.Bool(true),
@ -68,11 +56,116 @@ func handleStats(w http.ResponseWriter, r *http.Request) {
snippet := pie.RenderSnippet()
data.Element = template.HTML(snippet.Element)
data.Script = template.HTML(snippet.Script)
return ChartData{
Element: template.HTML(snippet.Element),
Script: template.HTML(snippet.Script),
}
}
func generateStats(ctx context.Context) ([]ChartData, error) {
data := []ChartData{}
rows, err := db.Query(ctx,
`SELECT COALESCE(jsonb_path_query_first(ffprobe_data, '$.streams[*] ? (@.codec_type == "video") ? (@.disposition.attached_pic == 0).codec_name')::text, 'Unknown') AS name, COUNT(*) AS value
FROM files
WHERE ffprobe_data IS NOT NULL
GROUP BY jsonb_path_query_first(ffprobe_data, '$.streams[*] ? (@.codec_type == "video") ? (@.disposition.attached_pic == 0).codec_name');`)
if err != nil {
return nil, fmt.Errorf("Query Codecs: %w", err)
}
codecCounts, err := pgx.CollectRows(rows, pgx.RowToStructByName[PieValue])
if err != nil {
return nil, fmt.Errorf("Collect Codec Data: %w", err)
}
res := []opts.PieData{}
for _, v := range codecCounts {
res = append(res, opts.PieData{
Name: strings.ReplaceAll(v.Name, "\"", ""),
Value: v.Value,
})
}
data = append(data, generatePie("Codecs", res))
rows, err = db.Query(ctx,
`SELECT health AS id, COUNT(*) AS value
FROM files
GROUP BY health;`)
if err != nil {
return nil, fmt.Errorf("Query Health: %w", err)
}
healthCounts, err := pgx.CollectRows(rows, pgx.RowToStructByName[PieIntValue])
if err != nil {
return nil, fmt.Errorf("Collect Health Data: %w", err)
}
res = []opts.PieData{}
for _, v := range healthCounts {
res = append(res, opts.PieData{
Name: constants.FileHealth(v.Id).String(),
Value: v.Value,
})
}
data = append(data, generatePie("Health", res))
rows, err = db.Query(ctx,
`SELECT transcode AS id, COUNT(*) AS value
FROM files
GROUP BY transcode;`)
if err != nil {
return nil, fmt.Errorf("Query Transcode: %w", err)
}
transcodeCounts, err := pgx.CollectRows(rows, pgx.RowToStructByName[PieIntValue])
if err != nil {
return nil, fmt.Errorf("Collect Transcode Data: %w", err)
}
res = []opts.PieData{}
for _, v := range transcodeCounts {
res = append(res, opts.PieData{
Name: constants.FileTranscode(v.Id).String(),
Value: v.Value,
})
}
data = append(data, generatePie("Transcode Status", res))
rows, err = db.Query(ctx,
`SELECT status AS id, COUNT(*) AS value
FROM tasks
GROUP BY status;`)
if err != nil {
return nil, fmt.Errorf("Query Task Status: %w", err)
}
taskStatusCounts, err := pgx.CollectRows(rows, pgx.RowToStructByName[PieIntValue])
if err != nil {
return nil, fmt.Errorf("Collect Task Status Data: %w", err)
}
res = []opts.PieData{}
for _, v := range taskStatusCounts {
res = append(res, opts.PieData{
Name: constants.TaskStatus(v.Id).String(),
Value: v.Value,
})
}
data = append(data, generatePie("Task Status", res))
return data, nil
}
func handleStats(w http.ResponseWriter, r *http.Request) {
data, err := generateStats(r.Context())
if err != nil {
slog.ErrorContext(r.Context(), "Generate Stats:", "err", err)
http.Error(w, "Generate Stats: "+err.Error(), http.StatusInternalServerError)
return
}
buf := bytes.Buffer{}
err = templates.ExecuteTemplate(&buf, constants.STATS_TEMPLATE_NAME, data)
err = templates.ExecuteTemplate(&buf, constants.STATS_TEMPLATE_NAME, StatsDisplay{
Charts: data,
})
if err != nil {
slog.ErrorContext(r.Context(), "Executing Stats Template", "err", err)
http.Error(w, "Error Executing Template: "+err.Error(), http.StatusInternalServerError)

13
static/style/style.css
View file

@ -32,21 +32,21 @@ body {
gap: 2rem;
}
.workers>* {
.workers > * {
gap: 0.5rem;
border: 1px solid var(--fg);
}
.log>p {
.log > p {
padding: 0.25rem 0;
user-select: text;
}
.log>p:hover {
.log > p:hover {
background: var(--cl-hl);
}
nav> :first-child {
nav > :first-child {
font-size: 2rem;
font-weight: bold;
}
@ -58,3 +58,8 @@ nav> :first-child {
.button-list {
flex-direction: row;
}
.stats {
flex-flow: row wrap;
gap: 2rem;
}

21
tmpl/stats.tmpl
View file

@ -1,23 +1,8 @@
{{template "head"}}
<h2>Stats</h2>
<h2>Codec Counts</h2>
<div>
<table>
<tr>
<th>Codec</th>
<th>Count</th>
</tr>
{{range $f := .CodecCounts}}
<tr>
<td>
{{ $f.Codec }}
</td>
<td>
{{ $f.Count }}
</td>
</tr>
<div class="stats">
{{range $c := .Charts}}
{{$c.Element}} {{$c.Script}}
{{end}}
</table>
</div>
{{.Element}} {{.Script}}
{{template "tail"}}