Compare commits
17 Commits
Author | SHA1 | Date |
---|---|---|
|
db65d47ea9 | |
|
8c7ccf7e9f | |
|
0e4ee6fcf8 | |
|
1f47ca4ee6 | |
|
fefb874f95 | |
|
804f9a337c | |
|
8c6189b4b8 | |
|
4acbfefd43 | |
|
08dab37c40 | |
|
a505ef3c3f | |
|
2ffd30dd0f | |
|
b5d374c0ae | |
|
af80fc6c92 | |
|
a94acd17ad | |
|
81c0ed4793 | |
|
14d35a9170 | |
|
2b504289ea |
|
@ -0,0 +1,9 @@
|
|||
#!/bin/bash
|
||||
|
||||
result="$(gofmt -s -l . | grep -v '^vendor/' )"
|
||||
if [ -n "$result" ]; then
|
||||
echo "Go code is not formatted, run 'gofmt -s -w .'" >&2
|
||||
echo "$result"
|
||||
exit 1
|
||||
fi
|
||||
|
|
@ -0,0 +1,26 @@
|
|||
#!/usr/bin/env python3
|
||||
# checks if every desired package has test files
|
||||
|
||||
import os
|
||||
import re
|
||||
import sys
|
||||
|
||||
source_re = re.compile(".*\.go")
|
||||
test_re = re.compile(".*_test\.go")
|
||||
missing = False
|
||||
|
||||
for root, dirs, files in os.walk("."):
|
||||
# ignore some paths
|
||||
if root == "." or root.startswith("./vendor") or root.startswith("./.") or root.startswith("./docs"):
|
||||
continue
|
||||
|
||||
# source files but not test files?
|
||||
if len([f for f in files if source_re.match(f)]) > 0 and len([f for f in files if test_re.match(f)]) == 0:
|
||||
print("no test files for {}".format(root))
|
||||
missing = True
|
||||
|
||||
if missing:
|
||||
sys.exit(1)
|
||||
else:
|
||||
print("every package has test files")
|
||||
|
|
@ -0,0 +1,80 @@
|
|||
image: "golang:latest"
|
||||
|
||||
variables:
|
||||
# Tell 'docker:dind' to enable TLS (recommended)
|
||||
# and generate certificates in the specified directory.
|
||||
DOCKER_TLS_CERTDIR: "/certs"
|
||||
|
||||
stages:
|
||||
- test
|
||||
- build
|
||||
|
||||
test-lint:
|
||||
stage: test
|
||||
script:
|
||||
- go install github.com/client9/misspell/cmd/misspell@latest
|
||||
- find . -type f -not -path "./webroot/assets" | grep -v "models/.*_testdata.*.go" | xargs misspell -error
|
||||
- ./.ci/check-gofmt
|
||||
- ./.ci/check-testfiles
|
||||
|
||||
build-linux:
|
||||
stage: build
|
||||
except:
|
||||
- tags
|
||||
- master
|
||||
- main
|
||||
script:
|
||||
# build app with version
|
||||
- go install
|
||||
- mv "/go/bin/$CI_PROJECT_NAME" "$CI_PROJECT_DIR/$CI_PROJECT_NAME"
|
||||
artifacts:
|
||||
paths:
|
||||
- config_example.toml
|
||||
- "$CI_PROJECT_NAME"
|
||||
|
||||
build-docker-latest:
|
||||
stage: build
|
||||
image: docker:latest
|
||||
services:
|
||||
- docker:dind
|
||||
only:
|
||||
- main
|
||||
- master
|
||||
script:
|
||||
- docker login -u $CI_REGISTRY_USER -p $CI_REGISTRY_PASSWORD $CI_REGISTRY
|
||||
- docker build --build-arg VERSION=$CI_COMMIT_TAG
|
||||
-t $CI_REGISTRY_IMAGE:latest
|
||||
.
|
||||
- docker push $CI_REGISTRY_IMAGE:latest
|
||||
|
||||
build-docker-release:
|
||||
stage: build
|
||||
image: docker:latest
|
||||
services:
|
||||
- docker:dind
|
||||
only:
|
||||
- tags
|
||||
script:
|
||||
- docker login -u $CI_REGISTRY_USER -p $CI_REGISTRY_PASSWORD $CI_REGISTRY
|
||||
- docker build --build-arg VERSION=$CI_COMMIT_TAG
|
||||
-t $CI_REGISTRY_IMAGE:$CI_COMMIT_TAG
|
||||
-t $CI_REGISTRY_IMAGE:${CI_COMMIT_TAG%\.[0-9]*}
|
||||
-t $CI_REGISTRY_IMAGE:${CI_COMMIT_TAG%\.[0-9]*\.[0-9]*}
|
||||
.
|
||||
- docker push $CI_REGISTRY_IMAGE:$CI_COMMIT_TAG
|
||||
- docker push $CI_REGISTRY_IMAGE:${CI_COMMIT_TAG%\.[0-9]*}
|
||||
- docker push $CI_REGISTRY_IMAGE:${CI_COMMIT_TAG%\.[0-9]*\.[0-9]*}
|
||||
|
||||
build-release:
|
||||
stage: build
|
||||
only:
|
||||
- tags
|
||||
script:
|
||||
# build app with version
|
||||
- go install
|
||||
- mv "/go/bin/$CI_PROJECT_NAME" "$CI_PROJECT_DIR/$CI_PROJECT_NAME"
|
||||
artifacts:
|
||||
paths:
|
||||
- config_example.toml
|
||||
- "$CI_PROJECT_NAME"
|
||||
expire_in: never
|
|
@ -0,0 +1,20 @@
|
|||
##
|
||||
# Compile application
|
||||
##
|
||||
FROM docker.io/library/golang:alpine AS build-env
|
||||
WORKDIR /app
|
||||
COPY . .
|
||||
# ge dependencies
|
||||
RUN go mod tidy
|
||||
# build binary
|
||||
RUN CGO_ENABLED=0 go build -ldflags="-w -s" -o oven-exporter
|
||||
|
||||
|
||||
##
|
||||
# Build Image
|
||||
##
|
||||
FROM scratch
|
||||
COPY --from=build-env ["/etc/ssl/cert.pem", "/etc/ssl/certs/ca-certificates.crt"]
|
||||
COPY --from=build-env /app/oven-exporter /oven-exporter
|
||||
WORKDIR /
|
||||
ENTRYPOINT ["/oven-exporter", "-c", ""]
|
|
@ -0,0 +1,59 @@
|
|||
== Oven-Exporter
|
||||
|
||||
An Prometheus Exporter for OvenMediaEngine
|
||||
|
||||
(it provides also a small API-Client for OvenMediaEngine) Be welcome to improve it.
|
||||
|
||||
=== Configure OvenMediaEngine
|
||||
|
||||
This Exporter use the REST-API of OvenMediaEngine, to setting it up, that a look in there Documentation https://airensoft.gitbook.io/ovenmediaengine/rest-api[OvenMediaEngine REST-API].
|
||||
|
||||
=== Setup Exporter
|
||||
|
||||
==== Compile
|
||||
|
||||
Install https://golang.org/doc/install[golang].
|
||||
|
||||
Run: `go install -v codeberg.org/Mediathek/oven-exporter@latest`
|
||||
|
||||
==== Configuration
|
||||
|
||||
Read comments in config_example.toml for more information.
|
||||
|
||||
Maybe a good place to store this file is: `/etc/ovenmediaengine/exporter.conf`
|
||||
|
||||
OR use env variables:
|
||||
|
||||
....
|
||||
OVEN_E_LISTEN=:8080
|
||||
OVEN_E_API__URL=http://1.2.3.4:8081
|
||||
OVEN_E_API__TOKEN=ome-access-token
|
||||
OVEN_E_API__DEFAULT_VHOST=
|
||||
OVEN_E_API__DEFAULT_APP=
|
||||
....
|
||||
|
||||
(File read could be disabled by call `oven-exporter -c ''`
|
||||
|
||||
==== Startup
|
||||
|
||||
Create a systemd.service file e.g. under `/etc/systemd/system/oven-exporter.service` with maybe a content like this:
|
||||
|
||||
[source,ini]
|
||||
----
|
||||
[Unit]
|
||||
Description = Prometheus exporter for OvenMediaEngine
|
||||
|
||||
[Service]
|
||||
Type=simple
|
||||
ExecStart=/usr/local/bin/oven-exporter -c /etc/ovenmediaengine/exporter.conf
|
||||
Restart=always
|
||||
RestartSec=5s
|
||||
Environment=PATH=/usr/bin:/usr/local/bin
|
||||
|
||||
[Install]
|
||||
WantedBy=multi-user.target
|
||||
----
|
||||
|
||||
PS: maybe you need to adjust the binary path and configuration path.
|
||||
|
||||
Start and enable on boot: `systemctl enable --now oven-exporter.service`
|
|
@ -1,5 +0,0 @@
|
|||
# Oven-Exporter
|
||||
An Prometheus Exporter for OvenMediaEngine
|
||||
|
||||
(it provides also a small API-Client for OvenMediaEngine)
|
||||
Be welcome to improve it.
|
|
@ -4,14 +4,16 @@ import "encoding/base64"
|
|||
|
||||
// A Client for the API
|
||||
type Client struct {
|
||||
Token string `toml:"token"`
|
||||
Host string `toml:"host"`
|
||||
Token string `config:"token"`
|
||||
URL string `config:"url"`
|
||||
DefaultVHost string `config:"default_vhost"`
|
||||
DefaultApp string `config:"default_app"`
|
||||
}
|
||||
|
||||
// New Client from host and token
|
||||
func New(host, token string) *Client {
|
||||
func New(url, token string) *Client {
|
||||
c := &Client{
|
||||
Host: host,
|
||||
URL: url,
|
||||
}
|
||||
c.SetToken(token)
|
||||
return c
|
||||
|
|
|
@ -0,0 +1 @@
|
|||
package api
|
|
@ -1,18 +1,33 @@
|
|||
package api
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
// Request to API and unmarshal result
|
||||
func (c *Client) Request(url string, value interface{}) error {
|
||||
func (c *Client) Request(method, url string, body, value interface{}) error {
|
||||
netClient := &http.Client{
|
||||
Timeout: time.Second * 20,
|
||||
}
|
||||
req, err := http.NewRequest(http.MethodGet, c.Host+url, nil)
|
||||
var jsonBody io.Reader
|
||||
if body != nil {
|
||||
if strBody, ok := body.(string); ok {
|
||||
jsonBody = strings.NewReader(strBody)
|
||||
} else {
|
||||
jsonBodyArray, err := json.Marshal(body)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
jsonBody = bytes.NewBuffer(jsonBodyArray)
|
||||
}
|
||||
}
|
||||
req, err := http.NewRequest(method, c.URL+url, jsonBody)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
|
|
@ -2,6 +2,7 @@ package api
|
|||
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
const (
|
||||
|
@ -16,29 +17,42 @@ type ResponseList struct {
|
|||
Data []string `json:"response,omitempty"`
|
||||
}
|
||||
|
||||
// RequestListVHosts to get list of vhosts
|
||||
func (c *Client) RequestListVHosts() (*ResponseList, error) {
|
||||
req := ResponseList{}
|
||||
url := fmt.Sprintf(URLRequestListVHost)
|
||||
if err := c.Request(url, &req); err != nil {
|
||||
if err := c.Request(http.MethodGet, url, nil, &req); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &req, nil
|
||||
}
|
||||
|
||||
// RequestListApps to get list of apps on given vhost
|
||||
func (c *Client) RequestListApps(vhost string) (*ResponseList, error) {
|
||||
req := ResponseList{}
|
||||
url := fmt.Sprintf(URLRequestListApp, vhost)
|
||||
if err := c.Request(url, &req); err != nil {
|
||||
if err := c.Request(http.MethodGet, url, nil, &req); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &req, nil
|
||||
}
|
||||
|
||||
// RequestDefaultListApps to get list of apps on default vhost
|
||||
func (c *Client) RequestDefaultListApps() (*ResponseList, error) {
|
||||
return c.RequestListApps(c.DefaultVHost)
|
||||
}
|
||||
|
||||
// RequestDefaultListStreams to get list of streams on given vhost and app
|
||||
func (c *Client) RequestListStreams(vhost, app string) (*ResponseList, error) {
|
||||
req := ResponseList{}
|
||||
url := fmt.Sprintf(URLRequestListStream, vhost, app)
|
||||
if err := c.Request(url, &req); err != nil {
|
||||
if err := c.Request(http.MethodGet, url, nil, &req); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &req, nil
|
||||
}
|
||||
|
||||
// RequestDefaultListStreams to get list of streams on default vhost and app
|
||||
func (c *Client) RequestDefaultListStreams() (*ResponseList, error) {
|
||||
return c.RequestListStreams(c.DefaultVHost, c.DefaultApp)
|
||||
}
|
||||
|
|
|
@ -0,0 +1,106 @@
|
|||
package api
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
// API URLS for Push
|
||||
const (
|
||||
URLRequestPushStatus = "/v1/vhosts/%s/apps/%s:pushes"
|
||||
URLRequestPushStart = "/v1/vhosts/%s/apps/%s:startPush"
|
||||
URLRequestPushStop = "/v1/vhosts/%s/apps/%s:stopPush"
|
||||
)
|
||||
|
||||
// ResponsePushStatus JSON Message with Push status data
|
||||
type ResponsePushStatus struct {
|
||||
Message string `json:"message"`
|
||||
StatusCode int `json:"statusCode"`
|
||||
Data []*ResponsePushData `json:"response,omitempty"`
|
||||
}
|
||||
|
||||
// ResponsePushData one push configuration
|
||||
type ResponsePushData struct {
|
||||
VHost string `json:"vhost" example:"default"`
|
||||
App string `json:"app" example:"live"`
|
||||
ID string `json:"id" example:"youtube"`
|
||||
Stream *ResponsePushDataStream `json:"stream"`
|
||||
State string `json:"state" example:"ready"`
|
||||
Protocol string `json:"protocol" example:"rtmp"`
|
||||
URL string `json:"url" example:"rtmp://a.rtmp.youtube.com/live2"`
|
||||
StreamKey string `json:"streamKey" example:"SUPERSECRET"`
|
||||
// - timestamp - time.TIme has problem with nanosecond in JSON
|
||||
CreatedTime string `json:"createdTime" example:"2021-07-19T23:13:12.162+0200"`
|
||||
FinishedTime string `json:"finishedTime" example:"2021-07-19T23:23:27.274+0200"`
|
||||
StartTime string `json:"startTime" example:"2021-07-19T23:23:27.077+0200"`
|
||||
// - coonnections
|
||||
Sequence int `json:"sequence" example:"1"`
|
||||
// - traffic
|
||||
SentBytes uint64 `json:"sentBytes" example:"0"`
|
||||
SentTime uint64 `json:"sentTime" example:"0"`
|
||||
TotalSentBytes uint64 `json:"totalsentBytes" example:"356233652"`
|
||||
TotalSentTime uint64 `json:"totalsentTime" example:"933808"`
|
||||
}
|
||||
|
||||
// ResponsePushStart one push configuration
|
||||
type ResponsePushStart struct {
|
||||
ID string `json:"id" example:"youtube"`
|
||||
Stream *ResponsePushDataStream `json:"stream"`
|
||||
Protocol string `json:"protocol" example:"rtmp"`
|
||||
URL string `json:"url" example:"rtmp://a.rtmp.youtube.com/live2"`
|
||||
StreamKey string `json:"streamKey" example:"SUPERSECRET"`
|
||||
}
|
||||
|
||||
// ResponsePushDataStream of data of stream
|
||||
type ResponsePushDataStream struct {
|
||||
Name string `json:"name" example:"..."`
|
||||
Tracks []int `json:"tracks" example:"[]"`
|
||||
}
|
||||
|
||||
// RequestPushStatus to get list of pushes and his configuration
|
||||
func (c *Client) RequestPushStatus(vhost, app string) (*ResponsePushStatus, error) {
|
||||
req := ResponsePushStatus{}
|
||||
url := fmt.Sprintf(URLRequestPushStatus, vhost, app)
|
||||
if err := c.Request(http.MethodGet, url, nil, &req); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &req, nil
|
||||
}
|
||||
|
||||
// RequestPushStatusDefault to get list of pushes and his configuration for default vhost and app
|
||||
func (c *Client) RequestPushStatusDefault() (*ResponsePushStatus, error) {
|
||||
return c.RequestPushStatus(c.DefaultVHost, c.DefaultApp)
|
||||
}
|
||||
|
||||
// StartPush to delete an push
|
||||
func (c *Client) StartPush(vhost, app string, data *ResponsePushStart) (*ResponsePushStatus, error) {
|
||||
req := ResponsePushStatus{}
|
||||
url := fmt.Sprintf(URLRequestPushStart, vhost, app)
|
||||
if err := c.Request(http.MethodPost, url, &data, &req); err != nil {
|
||||
return &req, err
|
||||
}
|
||||
return &req, nil
|
||||
}
|
||||
|
||||
// StartPushDefault to delete an push and on default vhost and app
|
||||
func (c *Client) StartPushDefault(data *ResponsePushStart) (*ResponsePushStatus, error) {
|
||||
return c.StartPush(c.DefaultVHost, c.DefaultApp, data)
|
||||
}
|
||||
|
||||
// DeletePush to delete an push
|
||||
func (c *Client) DeletePush(vhost, app, id string) error {
|
||||
type idJSON struct {
|
||||
ID string `json:"id"`
|
||||
}
|
||||
url := fmt.Sprintf(URLRequestPushStop, vhost, app)
|
||||
data := idJSON{ID: id}
|
||||
if err := c.Request(http.MethodPost, url, &data, nil); err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// DeletePushDefault to delete an push and on default vhost and app
|
||||
func (c *Client) DeletePushDefault(id string) error {
|
||||
return c.DeletePush(c.DefaultVHost, c.DefaultApp, id)
|
||||
}
|
|
@ -2,6 +2,7 @@ package api
|
|||
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
const (
|
||||
|
@ -34,7 +35,7 @@ type ResponseStatsData struct {
|
|||
func (c *Client) RequestStatsVHost(vhost string) (*ResponseStats, error) {
|
||||
req := ResponseStats{}
|
||||
url := fmt.Sprintf(URLRequestStatsVHost, vhost)
|
||||
if err := c.Request(url, &req); err != nil {
|
||||
if err := c.Request(http.MethodGet, url, nil, &req); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &req, nil
|
||||
|
@ -43,7 +44,7 @@ func (c *Client) RequestStatsVHost(vhost string) (*ResponseStats, error) {
|
|||
func (c *Client) RequestStatsApp(vhost, app string) (*ResponseStats, error) {
|
||||
req := ResponseStats{}
|
||||
url := fmt.Sprintf(URLRequestStatsApp, vhost, app)
|
||||
if err := c.Request(url, &req); err != nil {
|
||||
if err := c.Request(http.MethodGet, url, nil, &req); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &req, nil
|
||||
|
@ -52,7 +53,7 @@ func (c *Client) RequestStatsApp(vhost, app string) (*ResponseStats, error) {
|
|||
func (c *Client) RequestStatsStream(vhost, app, stream string) (*ResponseStats, error) {
|
||||
req := ResponseStats{}
|
||||
url := fmt.Sprintf(URLRequestStatsStream, vhost, app, stream)
|
||||
if err := c.Request(url, &req); err != nil {
|
||||
if err := c.Request(http.MethodGet, url, nil, &req); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &req, nil
|
||||
|
|
|
@ -1,6 +1,9 @@
|
|||
# (Address +) Port where the Exporter is listen for Prometheus request
|
||||
listen = ":8080"
|
||||
|
||||
[api]
|
||||
host = "http://1.2.3.4:8081"
|
||||
# URL to REST-API of OvenMediaEngine
|
||||
url = "http://1.2.3.4:8081"
|
||||
# AccessToken of OvenMediaEngine's REST-API
|
||||
token = "ome-access-token"
|
||||
|
||||
|
|
|
@ -0,0 +1,12 @@
|
|||
[Unit]
|
||||
Description = Prometheus exporter for OvenMediaEngine
|
||||
|
||||
[Service]
|
||||
Type=simple
|
||||
ExecStart=/usr/local/bin/oven-exporter -c /etc/ovenmediaengine/exporter.conf
|
||||
Restart=always
|
||||
RestartSec=5s
|
||||
Environment=PATH=/usr/bin:/usr/local/bin
|
||||
|
||||
[Install]
|
||||
WantedBy=multi-user.target
|
|
@ -0,0 +1,9 @@
|
|||
name: oven-exporter
|
||||
title: Oven Exporter
|
||||
# use from git tag
|
||||
version:
|
||||
v(?<version>+({0..9}).+({0..9}).+({0..9})): $<version>
|
||||
main: latest
|
||||
|
||||
nav:
|
||||
- "modules/ROOT/nav.adoc"
|
|
@ -0,0 +1 @@
|
|||
../../../../README.adoc
|
19
go.mod
19
go.mod
|
@ -1,13 +1,16 @@
|
|||
module dev.sum7.eu/genofire/oven-exporter
|
||||
module codeberg.org/Mediathek/oven-exporter
|
||||
|
||||
go 1.16
|
||||
|
||||
require (
|
||||
dev.sum7.eu/genofire/golang-lib v0.0.0-20210719163544-fb766ca32b7c
|
||||
github.com/bdlm/log v0.1.20
|
||||
github.com/bdlm/std v1.0.1 // indirect
|
||||
github.com/prometheus/client_golang v1.10.0
|
||||
github.com/stretchr/testify v1.7.0 // indirect
|
||||
golang.org/x/crypto v0.0.0-20210711020723-a769d52b0f97 // indirect
|
||||
golang.org/x/sys v0.0.0-20210630005230-0f9fa26af87c // indirect
|
||||
github.com/fsnotify/fsnotify v1.7.0 // indirect
|
||||
github.com/knadh/koanf v1.5.0
|
||||
github.com/pelletier/go-toml v1.9.5 // indirect
|
||||
github.com/prometheus/client_golang v1.17.0
|
||||
github.com/prometheus/client_model v0.5.0 // indirect
|
||||
github.com/prometheus/common v0.45.0 // indirect
|
||||
github.com/prometheus/procfs v0.12.0 // indirect
|
||||
github.com/stretchr/testify v1.8.4
|
||||
go.uber.org/multierr v1.11.0 // indirect
|
||||
go.uber.org/zap v1.26.0
|
||||
)
|
||||
|
|
|
@ -0,0 +1,63 @@
|
|||
package helper
|
||||
|
||||
import (
|
||||
"crypto/hmac"
|
||||
"crypto/sha1"
|
||||
"encoding/base64"
|
||||
"encoding/json"
|
||||
"net/url"
|
||||
)
|
||||
|
||||
type Policy struct {
|
||||
URLExpire int64 `json:"url_expire"`
|
||||
URLActivate int64 `json:"url_activate,omitempty"`
|
||||
StreamExpire int64 `json:"stream_expire,omitempty"`
|
||||
AllowIP string `json:"allow_ip,omitempty"`
|
||||
}
|
||||
|
||||
func (p Policy) Encode() (string, error) {
|
||||
str, err := json.Marshal(p)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
return base64.RawStdEncoding.EncodeToString(str), nil
|
||||
}
|
||||
|
||||
func SignEncodedPolicy(u *url.URL, secretKey string) string {
|
||||
hasher := hmac.New(sha1.New, []byte(secretKey))
|
||||
hasher.Write([]byte(u.String()))
|
||||
return base64.RawURLEncoding.EncodeToString(hasher.Sum(nil))
|
||||
|
||||
}
|
||||
|
||||
func (p Policy) SignWithQuery(u *url.URL, secretKey, encodeQuery string) (string, error) {
|
||||
encode, err := p.Encode()
|
||||
if err != nil {
|
||||
return "", nil
|
||||
}
|
||||
query := u.Query()
|
||||
query.Add(encodeQuery, encode)
|
||||
u.RawQuery = query.Encode()
|
||||
return SignEncodedPolicy(u, secretKey), nil
|
||||
}
|
||||
func (p Policy) Sign(u *url.URL, secretKey string) (string, error) {
|
||||
return p.SignWithQuery(u, secretKey, "policy")
|
||||
}
|
||||
|
||||
func (p Policy) SignURLWithQuery(u *url.URL, secretKey, encodeQuery, signatureQuery string) error {
|
||||
encode, err := p.Encode()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
query := u.Query()
|
||||
query.Add(encodeQuery, encode)
|
||||
u.RawQuery = query.Encode()
|
||||
|
||||
signature := SignEncodedPolicy(u, secretKey)
|
||||
query.Add(signatureQuery, signature)
|
||||
u.RawQuery = query.Encode()
|
||||
return nil
|
||||
}
|
||||
func (p Policy) SignURL(u *url.URL, secretKey string) error {
|
||||
return p.SignURLWithQuery(u, secretKey, "policy", "signature")
|
||||
}
|
|
@ -0,0 +1,57 @@
|
|||
// https://airensoft.gitbook.io/ovenmediaengine/access-control/signedpolicy
|
||||
package helper
|
||||
|
||||
import (
|
||||
"net/url"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
const (
|
||||
examplePolicyEncode = "eyJ1cmxfZXhwaXJlIjoxMzk5NzIxNTgxfQ"
|
||||
exampleSecretKey = "1kU^b6"
|
||||
exampleURL = "ws://192.168.0.100:3333/app/stream"
|
||||
exampleSignature = "dvVdBpoxAeCPl94Kt5RoiqLI0YE"
|
||||
exampleURLWithSignatureAndPolicy = "ws://192.168.0.100/app/stream?policy=eyJ1cmxfZXhwaXJlIjoxMzk5NzIxNTgxfQ&signature=dvVdBpoxAeCPl94Kt5RoiqLI0YE"
|
||||
)
|
||||
|
||||
var (
|
||||
examplePolicy = Policy{
|
||||
URLExpire: 1399721581,
|
||||
}
|
||||
)
|
||||
|
||||
func TestPolicyEncode(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
|
||||
encode, err := examplePolicy.Encode()
|
||||
assert.NoError(err)
|
||||
assert.Equal(examplePolicyEncode, encode)
|
||||
}
|
||||
|
||||
func TestPolicySign(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
|
||||
u, err := url.Parse(exampleURL)
|
||||
assert.NoError(err)
|
||||
|
||||
sign, err := examplePolicy.Sign(u, exampleSecretKey)
|
||||
assert.NoError(err)
|
||||
assert.Equal(exampleSignature, sign)
|
||||
}
|
||||
|
||||
func TestPolicySignURL(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
|
||||
u, err := url.Parse(exampleURL)
|
||||
assert.NoError(err)
|
||||
|
||||
err = examplePolicy.SignURL(u, exampleSecretKey)
|
||||
assert.NoError(err)
|
||||
|
||||
// drop port -> is not part of example
|
||||
u.Host = u.Hostname()
|
||||
|
||||
assert.Equal(exampleURLWithSignatureAndPolicy, u.String())
|
||||
}
|
72
main.go
72
main.go
|
@ -3,34 +3,86 @@ package main
|
|||
import (
|
||||
"flag"
|
||||
"net/http"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
|
||||
"dev.sum7.eu/genofire/golang-lib/file"
|
||||
"github.com/bdlm/log"
|
||||
"github.com/knadh/koanf"
|
||||
"github.com/knadh/koanf/parsers/json"
|
||||
"github.com/knadh/koanf/parsers/toml"
|
||||
"github.com/knadh/koanf/parsers/yaml"
|
||||
"github.com/knadh/koanf/providers/env"
|
||||
"github.com/knadh/koanf/providers/file"
|
||||
"github.com/prometheus/client_golang/prometheus"
|
||||
"github.com/prometheus/client_golang/prometheus/promhttp"
|
||||
"go.uber.org/zap"
|
||||
|
||||
"dev.sum7.eu/genofire/oven-exporter/api"
|
||||
"codeberg.org/Mediathek/oven-exporter/api"
|
||||
)
|
||||
|
||||
var configExtParser = map[string]koanf.Parser{
|
||||
".json": json.Parser(),
|
||||
".toml": toml.Parser(),
|
||||
".yaml": yaml.Parser(),
|
||||
".yml": yaml.Parser(),
|
||||
}
|
||||
|
||||
type configData struct {
|
||||
API *api.Client `toml:"api"`
|
||||
Listen string `toml:"listen"`
|
||||
log *zap.Logger
|
||||
Log *zap.Config `config:"log"`
|
||||
API api.Client `config:"api"`
|
||||
Listen string `config:"listen"`
|
||||
}
|
||||
|
||||
func main() {
|
||||
|
||||
configPath := "config.toml"
|
||||
|
||||
flag.StringVar(&configPath, "c", configPath, "path to configuration file")
|
||||
log, _ := zap.NewProduction()
|
||||
|
||||
flag.StringVar(&configPath, "c", configPath, "path to configuration file")
|
||||
flag.Parse()
|
||||
|
||||
config := &configData{}
|
||||
if err := file.ReadTOML(configPath, config); err != nil {
|
||||
log.Panicf("open config file: %s", err)
|
||||
k := koanf.New("/")
|
||||
|
||||
if configPath != "" {
|
||||
fileExt := filepath.Ext(configPath)
|
||||
parser, ok := configExtParser[fileExt]
|
||||
if !ok {
|
||||
log.Panic("unsupported file extension:",
|
||||
zap.String("config-path", configPath),
|
||||
zap.String("file-ext", fileExt),
|
||||
)
|
||||
}
|
||||
if err := k.Load(file.Provider(configPath), parser); err != nil {
|
||||
log.Panic("load file config:", zap.Error(err))
|
||||
}
|
||||
}
|
||||
|
||||
if err := k.Load(env.Provider("OVEN_E_", "/", func(s string) string {
|
||||
return strings.Replace(strings.ToLower(
|
||||
strings.TrimPrefix(s, "OVEN_E_")), "__", "/", -1)
|
||||
}), nil); err != nil {
|
||||
log.Panic("load env:", zap.Error(err))
|
||||
}
|
||||
|
||||
config := &configData{}
|
||||
if err := k.UnmarshalWithConf("", &config, koanf.UnmarshalConf{Tag: "config"}); err != nil {
|
||||
log.Panic("reading config", zap.Error(err))
|
||||
}
|
||||
if config.Log != nil {
|
||||
l, err := config.Log.Build()
|
||||
if err != nil {
|
||||
log.Panic("generate logger from config", zap.Error(err))
|
||||
}
|
||||
log = l
|
||||
}
|
||||
config.log = log
|
||||
//config.SetLogger(log)
|
||||
config.API.SetToken(config.API.Token)
|
||||
|
||||
prometheus.MustRegister(config)
|
||||
http.Handle("/metrics", promhttp.Handler())
|
||||
log.Fatal(http.ListenAndServe(config.Listen, nil))
|
||||
if err := http.ListenAndServe(config.Listen, nil); err != nil {
|
||||
log.Fatal("crash webserver", zap.Error(err))
|
||||
}
|
||||
}
|
||||
|
|
|
@ -1,9 +1,9 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"dev.sum7.eu/genofire/oven-exporter/api"
|
||||
"github.com/bdlm/log"
|
||||
"codeberg.org/Mediathek/oven-exporter/api"
|
||||
"github.com/prometheus/client_golang/prometheus"
|
||||
"go.uber.org/zap"
|
||||
)
|
||||
|
||||
var (
|
||||
|
@ -39,6 +39,17 @@ var (
|
|||
promDescStatsStreamBytesInTotal,
|
||||
promDescStatsStreamBytesOutTotal,
|
||||
}
|
||||
|
||||
promDescPushUp = prometheus.NewDesc("oven_push_up", "state of push", []string{"vhost", "app", "stream", "id", "state"}, prometheus.Labels{})
|
||||
promDescPushSequence = prometheus.NewDesc("oven_push_sequence", "sequence of started pushes", []string{"vhost", "app", "stream", "id"}, prometheus.Labels{})
|
||||
promDescPushSentBytes = prometheus.NewDesc("oven_push_send_byte", "bytes send on push", []string{"vhost", "app", "stream", "id"}, prometheus.Labels{})
|
||||
promDescPushTotalSentBytes = prometheus.NewDesc("oven_push_total_send_bytes", "total bytes send on push", []string{"vhost", "app", "stream", "id"}, prometheus.Labels{})
|
||||
promDescPush = []*prometheus.Desc{
|
||||
promDescPushUp,
|
||||
promDescPushSequence,
|
||||
promDescPushSentBytes,
|
||||
promDescPushTotalSentBytes,
|
||||
}
|
||||
)
|
||||
|
||||
func ResponseStatsToMetrics(resp *api.ResponseStats, descs []*prometheus.Desc, labels ...string) []prometheus.Metric {
|
||||
|
@ -49,13 +60,13 @@ func ResponseStatsToMetrics(resp *api.ResponseStats, descs []*prometheus.Desc, l
|
|||
if m, err := prometheus.NewConstMetric(descs[0], prometheus.GaugeValue, float64(resp.Data.TotalConnections), labels...); err == nil {
|
||||
list = append(list, m)
|
||||
}
|
||||
if m, err := prometheus.NewConstMetric(descs[1], prometheus.GaugeValue, float64(resp.Data.MaxTotalConnections), labels...); err == nil {
|
||||
if m, err := prometheus.NewConstMetric(descs[1], prometheus.CounterValue, float64(resp.Data.MaxTotalConnections), labels...); err == nil {
|
||||
list = append(list, m)
|
||||
}
|
||||
if m, err := prometheus.NewConstMetric(descs[2], prometheus.GaugeValue, float64(resp.Data.TotalBytesIn), labels...); err == nil {
|
||||
if m, err := prometheus.NewConstMetric(descs[2], prometheus.CounterValue, float64(resp.Data.TotalBytesIn), labels...); err == nil {
|
||||
list = append(list, m)
|
||||
}
|
||||
if m, err := prometheus.NewConstMetric(descs[3], prometheus.GaugeValue, float64(resp.Data.TotalBytesOut), labels...); err == nil {
|
||||
if m, err := prometheus.NewConstMetric(descs[3], prometheus.CounterValue, float64(resp.Data.TotalBytesOut), labels...); err == nil {
|
||||
list = append(list, m)
|
||||
}
|
||||
return list
|
||||
|
@ -71,15 +82,18 @@ func (c *configData) Describe(d chan<- *prometheus.Desc) {
|
|||
for _, desc := range promDescStatsStream {
|
||||
d <- desc
|
||||
}
|
||||
for _, desc := range promDescPush {
|
||||
d <- desc
|
||||
}
|
||||
}
|
||||
|
||||
func (c *configData) Collect(metrics chan<- prometheus.Metric) {
|
||||
respList, err := c.API.RequestListVHosts()
|
||||
if err != nil {
|
||||
log.Panicf("unable to fetch vhosts: %s", err)
|
||||
c.log.Panic("unable to fetch vhosts", zap.Error(err))
|
||||
}
|
||||
for _, vhost := range respList.Data {
|
||||
logVhost := log.WithField("vhost", vhost)
|
||||
logVhost := c.log.With(zap.String("vhost", vhost))
|
||||
if resp, err := c.API.RequestStatsVHost(vhost); err == nil {
|
||||
for _, m := range ResponseStatsToMetrics(resp, promDescStatsVHost, vhost) {
|
||||
metrics <- m
|
||||
|
@ -87,19 +101,38 @@ func (c *configData) Collect(metrics chan<- prometheus.Metric) {
|
|||
}
|
||||
respList, err = c.API.RequestListApps(vhost)
|
||||
if err != nil {
|
||||
logVhost.Errorf("unable to fetch apps: %s", err)
|
||||
logVhost.Error("unable to fetch apps", zap.Error(err))
|
||||
continue
|
||||
}
|
||||
for _, app := range respList.Data {
|
||||
logApp := logVhost.WithField("app", app)
|
||||
logApp := logVhost.With(zap.String("app", app))
|
||||
if resp, err := c.API.RequestStatsApp(vhost, app); err == nil {
|
||||
for _, m := range ResponseStatsToMetrics(resp, promDescStatsApp, vhost, app) {
|
||||
metrics <- m
|
||||
}
|
||||
}
|
||||
if resp, err := c.API.RequestPushStatus(vhost, app); err != nil {
|
||||
logApp.Error("unable to fetch pushes", zap.Error(err))
|
||||
} else {
|
||||
for _, data := range resp.Data {
|
||||
if m, err := prometheus.NewConstMetric(promDescPushUp, prometheus.GaugeValue, 1, vhost, app, data.Stream.Name, data.ID, data.State); err == nil {
|
||||
metrics <- m
|
||||
}
|
||||
labels := []string{vhost, app, data.Stream.Name, data.ID}
|
||||
if m, err := prometheus.NewConstMetric(promDescPushSequence, prometheus.CounterValue, float64(data.Sequence), labels...); err == nil {
|
||||
metrics <- m
|
||||
}
|
||||
if m, err := prometheus.NewConstMetric(promDescPushSentBytes, prometheus.GaugeValue, float64(data.SentBytes), labels...); err == nil {
|
||||
metrics <- m
|
||||
}
|
||||
if m, err := prometheus.NewConstMetric(promDescPushTotalSentBytes, prometheus.CounterValue, float64(data.TotalSentBytes), labels...); err == nil {
|
||||
metrics <- m
|
||||
}
|
||||
}
|
||||
}
|
||||
respList, err = c.API.RequestListStreams(vhost, app)
|
||||
if err != nil {
|
||||
logApp.Errorf("unable to fetch stream: %s", err)
|
||||
logApp.Error("unable to fetch stream", zap.Error(err))
|
||||
continue
|
||||
}
|
||||
for _, stream := range respList.Data {
|
||||
|
|
Loading…
Reference in New Issue