Initial commit
This commit is contained in:
commit
f8b9c4f748
GPG key ID:
7262DAFE292BCE20
20 changed files with 1494 additions and 0 deletions
3
.gitignore
vendored
Normal file
3
.gitignore
vendored
Normal file
|
|
@ -0,0 +1,3 @@
|
||||||
|
tmp/
|
||||||
|
__debug*
|
||||||
|
main
|
||||||
54
.vscode/launch.json
vendored
Normal file
54
.vscode/launch.json
vendored
Normal file
|
|
@ -0,0 +1,54 @@
|
||||||
|
{
|
||||||
|
// Use IntelliSense to learn about possible attributes.
|
||||||
|
// Hover to view descriptions of existing attributes.
|
||||||
|
// For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
|
||||||
|
"version": "0.2.0",
|
||||||
|
"configurations": [
|
||||||
|
{
|
||||||
|
"name": "Master",
|
||||||
|
"type": "go",
|
||||||
|
"request": "launch",
|
||||||
|
"mode": "auto",
|
||||||
|
"program": "${workspaceFolder}/cli",
|
||||||
|
"env": {
|
||||||
|
"DUMP_PATH": "../tmp/nodes.json",
|
||||||
|
"CONFIG_PATH": "../tmp/config.json",
|
||||||
|
"NODE_DIR_PATH": "../tmp/node",
|
||||||
|
"TOKEN": "abcd"
|
||||||
|
},
|
||||||
|
"args": [
|
||||||
|
"master"
|
||||||
|
]
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "Node",
|
||||||
|
"type": "go",
|
||||||
|
"request": "launch",
|
||||||
|
"mode": "auto",
|
||||||
|
"program": "${workspaceFolder}/cli",
|
||||||
|
"env": {
|
||||||
|
"DUMP_PATH": "../tmp/nodes.json",
|
||||||
|
"CONFIG_PATH": "../tmp/config.json",
|
||||||
|
"NODE_DIR_PATH": "../tmp/node"
|
||||||
|
},
|
||||||
|
"args": [
|
||||||
|
"node"
|
||||||
|
]
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "Print",
|
||||||
|
"type": "go",
|
||||||
|
"request": "launch",
|
||||||
|
"mode": "auto",
|
||||||
|
"program": "${workspaceFolder}/cli",
|
||||||
|
"env": {
|
||||||
|
"DUMP_PATH": "../tmp/nodes.json",
|
||||||
|
"CONFIG_PATH": "../tmp/config.json",
|
||||||
|
"NODE_DIR_PATH": "../tmp/node"
|
||||||
|
},
|
||||||
|
"args": [
|
||||||
|
"print"
|
||||||
|
]
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
5
README.md
Normal file
5
README.md
Normal file
|
|
@ -0,0 +1,5 @@
|
||||||
|
# Lobby 2 - simple service discovery
|
||||||
|
|
||||||
|
This is second version of my Lobby projects that doesn't require NATS. All clients uses single service discovery server that keeps track of their presence.
|
||||||
|
|
||||||
|
Each instance, except its liveness can share labels which describe what's hosted on this instance and that can used by others instances for various things. There is also a simple KV store for additional info.
|
||||||
9
Taskfile.yml
Normal file
9
Taskfile.yml
Normal file
|
|
@ -0,0 +1,9 @@
|
||||||
|
# https://taskfile.dev
|
||||||
|
|
||||||
|
version: '3'
|
||||||
|
|
||||||
|
tasks:
|
||||||
|
docs:
|
||||||
|
cmds:
|
||||||
|
- swag i --parseDependency --dir api
|
||||||
|
silent: true
|
||||||
41
api/auth.go
Normal file
41
api/auth.go
Normal file
|
|
@ -0,0 +1,41 @@
|
||||||
|
package api
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/labstack/echo/v4"
|
||||||
|
)
|
||||||
|
|
||||||
|
func tokenMiddlware(configuredToken string) echo.MiddlewareFunc {
|
||||||
|
return func(next echo.HandlerFunc) echo.HandlerFunc {
|
||||||
|
return func(c echo.Context) error {
|
||||||
|
// Ignore token check for swagger URLs
|
||||||
|
if strings.HasPrefix(c.Request().URL.Path, "/swagger") || c.Request().URL.Path == "/" {
|
||||||
|
return next(c)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check for token in the Authorization header
|
||||||
|
authHeader := c.Request().Header.Get("Authorization")
|
||||||
|
if authHeader == "" {
|
||||||
|
return echo.NewHTTPError(http.StatusUnauthorized, "please provide valid token")
|
||||||
|
}
|
||||||
|
|
||||||
|
// The Authorization header should be in the format "Bearer <token>"
|
||||||
|
parts := strings.Split(authHeader, " ")
|
||||||
|
if len(parts) == 1 && parts[0] == configuredToken {
|
||||||
|
return next(c)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(parts) != 2 || parts[0] != "Bearer" {
|
||||||
|
return echo.NewHTTPError(http.StatusUnauthorized, "please provide valid token")
|
||||||
|
}
|
||||||
|
|
||||||
|
if parts[1] != configuredToken {
|
||||||
|
return echo.NewHTTPError(http.StatusUnauthorized, "please provide valid token")
|
||||||
|
}
|
||||||
|
|
||||||
|
return next(c)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
113
api/main.go
Normal file
113
api/main.go
Normal file
|
|
@ -0,0 +1,113 @@
|
||||||
|
package api
|
||||||
|
|
||||||
|
import (
|
||||||
|
"log"
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
_ "gitea.ceperka.net/rosti/lobby2/docs" // This line is necessary for swag to find your docs!
|
||||||
|
"gitea.ceperka.net/rosti/lobby2/nodes"
|
||||||
|
"github.com/labstack/echo/v4"
|
||||||
|
"github.com/labstack/echo/v4/middleware"
|
||||||
|
echoSwagger "github.com/swaggo/echo-swagger"
|
||||||
|
)
|
||||||
|
|
||||||
|
// @title Lobby2 API
|
||||||
|
// @version 2.0
|
||||||
|
// @description API of Lobby 2 project that helps to discover and connect to other nodes and their services.
|
||||||
|
// @BasePath /
|
||||||
|
// @securityDefinitions.apikey Bearer
|
||||||
|
// @in header
|
||||||
|
// @name Authorization
|
||||||
|
|
||||||
|
type API struct {
|
||||||
|
listen string
|
||||||
|
token string
|
||||||
|
np *nodes.NodesProcessor
|
||||||
|
|
||||||
|
e *echo.Echo
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewAPI(np *nodes.NodesProcessor, listen string, token string) *API {
|
||||||
|
return &API{
|
||||||
|
listen: listen,
|
||||||
|
token: token,
|
||||||
|
np: np,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *API) Run() error {
|
||||||
|
if a.token == "" {
|
||||||
|
log.Fatalln("TOKEN is required")
|
||||||
|
}
|
||||||
|
|
||||||
|
a.e = echo.New()
|
||||||
|
|
||||||
|
a.e.Use(middleware.Logger())
|
||||||
|
a.e.Use(tokenMiddlware(a.token))
|
||||||
|
|
||||||
|
a.e.GET("/", func(c echo.Context) error {
|
||||||
|
return c.Redirect(http.StatusTemporaryRedirect, "/swagger/index.html")
|
||||||
|
})
|
||||||
|
a.e.GET("/swagger/*", echoSwagger.WrapHandler)
|
||||||
|
|
||||||
|
a.e.GET("/nodes", a.listHandler)
|
||||||
|
a.e.GET("/nodes/:hostname", a.getHandler)
|
||||||
|
a.e.POST("/nodes/:hostname", a.refreshHandler)
|
||||||
|
|
||||||
|
// Start the server in a goroutine so that it doesn't block the signal listening
|
||||||
|
return a.e.Start(a.listen)
|
||||||
|
}
|
||||||
|
|
||||||
|
// @Summary List of nodes
|
||||||
|
// @Description List of all discovered nodes and their labels.
|
||||||
|
// @Produce application/json
|
||||||
|
// @Success 200 {object} nodes.Nodes "List of nodes"
|
||||||
|
// @Failure 401 {object} Message "Forbidden access"
|
||||||
|
// @Security Bearer
|
||||||
|
// @Router /nodes [get]
|
||||||
|
func (a *API) listHandler(c echo.Context) error {
|
||||||
|
return c.JSON(http.StatusOK, a.np.List())
|
||||||
|
}
|
||||||
|
|
||||||
|
// @Summary Get node
|
||||||
|
// @Description Return one nodes based on given hostname
|
||||||
|
// @Produce application/json
|
||||||
|
// @Param hostname path string true "Node hostname"
|
||||||
|
// @Success 200 {array} nodes.Node "Node details"
|
||||||
|
// @Failure 401 {object} Message "Forbidden access"
|
||||||
|
// @Security Bearer
|
||||||
|
// @Router /nodes/{hostname} [get]
|
||||||
|
func (a *API) getHandler(c echo.Context) error {
|
||||||
|
hostname := c.Param("hostname")
|
||||||
|
|
||||||
|
node, ok := a.np.Get(hostname)
|
||||||
|
if !ok {
|
||||||
|
return echo.NewHTTPError(http.StatusNotFound, "node not found")
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.JSON(http.StatusOK, node)
|
||||||
|
}
|
||||||
|
|
||||||
|
// @Summary Refresh node
|
||||||
|
// @Description Send new data or update existing data about a node
|
||||||
|
// @Produce application/json
|
||||||
|
// @Param hostname path string true "Node hostname"
|
||||||
|
// @Param labels body nodes.Labels true "Node labels"
|
||||||
|
// @Param kv body nodes.KV true "Key-value"
|
||||||
|
// @Success 200 {array} nodes.Node "Node details"
|
||||||
|
// @Failure 401 {object} Message "Forbidden access"
|
||||||
|
// @Security Bearer
|
||||||
|
// @Router /nodes/{hostname} [post]
|
||||||
|
func (a *API) refreshHandler(c echo.Context) error {
|
||||||
|
hostname := c.Param("hostname")
|
||||||
|
|
||||||
|
params := params{}
|
||||||
|
err := c.Bind(¶ms)
|
||||||
|
if err != nil {
|
||||||
|
return c.JSON(http.StatusBadRequest, Message{Message: err.Error()})
|
||||||
|
}
|
||||||
|
|
||||||
|
a.np.Refresh(hostname, params.Labels, params.KV)
|
||||||
|
|
||||||
|
return c.NoContent(http.StatusNoContent)
|
||||||
|
}
|
||||||
12
api/types.go
Normal file
12
api/types.go
Normal file
|
|
@ -0,0 +1,12 @@
|
||||||
|
package api
|
||||||
|
|
||||||
|
import "gitea.ceperka.net/rosti/lobby2/nodes"
|
||||||
|
|
||||||
|
type Message struct {
|
||||||
|
Message string `json:"message"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type params struct {
|
||||||
|
Labels nodes.Labels `json:"labels"`
|
||||||
|
KV nodes.KV `json:"kv"`
|
||||||
|
}
|
||||||
54
cli/actions.go
Normal file
54
cli/actions.go
Normal file
|
|
@ -0,0 +1,54 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
|
||||||
|
"gitea.ceperka.net/rosti/lobby2/api"
|
||||||
|
"gitea.ceperka.net/rosti/lobby2/nodes"
|
||||||
|
"gitea.ceperka.net/rosti/lobby2/refresher"
|
||||||
|
"github.com/urfave/cli/v2"
|
||||||
|
)
|
||||||
|
|
||||||
|
func masterAction(c *cli.Context) error {
|
||||||
|
cfg := GetConfig()
|
||||||
|
|
||||||
|
np := nodes.NewNodesProcessor(cfg.DumpPath, cfg.DropAfterSeconds)
|
||||||
|
|
||||||
|
go np.DumpLoop()
|
||||||
|
go np.GarbageCollection()
|
||||||
|
|
||||||
|
api := api.NewAPI(np, cfg.APIListen, cfg.APIToken)
|
||||||
|
api.Run()
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func nodeAction(c *cli.Context) error {
|
||||||
|
cfg := GetConfig()
|
||||||
|
|
||||||
|
r := refresher.NewRefresher(cfg.NodeDirPath, cfg.ConfigPath)
|
||||||
|
r.Loop()
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func printAction(c *cli.Context) error {
|
||||||
|
cfg := GetConfig()
|
||||||
|
|
||||||
|
np := nodes.NewNodesProcessor(cfg.DumpPath, cfg.DropAfterSeconds)
|
||||||
|
|
||||||
|
nodes := np.List()
|
||||||
|
for _, node := range nodes {
|
||||||
|
body, err := json.MarshalIndent(node, "", " ")
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("failed to marshal node: %v\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Println(string(body))
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
28
cli/config.go
Normal file
28
cli/config.go
Normal file
|
|
@ -0,0 +1,28 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"log"
|
||||||
|
|
||||||
|
"github.com/kelseyhightower/envconfig"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Config struct {
|
||||||
|
APIListen string `envconfig:"LISTEN" default:"0.0.0.0:1352"`
|
||||||
|
APIToken string `envconfig:"TOKEN" default:""`
|
||||||
|
|
||||||
|
DumpPath string `envconfig:"DUMP_PATH" default:"/var/lib/lobby2/nodes.json"`
|
||||||
|
ConfigPath string `envconfig:"CONFIG_PATH" default:"/var/lib/lobby2/config.json"`
|
||||||
|
NodeDirPath string `envconfig:"NODE_DIR_PATH" default:"/var/lib/lobby2/node"`
|
||||||
|
|
||||||
|
DropAfterSeconds int64 `envconfig:"DROP_AFTER_SECONDS" default:"60"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetConfig() Config {
|
||||||
|
var cfg Config
|
||||||
|
err := envconfig.Process("", &cfg)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return cfg
|
||||||
|
}
|
||||||
37
cli/main.go
Normal file
37
cli/main.go
Normal file
|
|
@ -0,0 +1,37 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
|
||||||
|
"github.com/urfave/cli/v2"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
app := &cli.App{
|
||||||
|
Name: "",
|
||||||
|
Usage: "",
|
||||||
|
Commands: []*cli.Command{
|
||||||
|
{
|
||||||
|
Name: "master",
|
||||||
|
Usage: "Runs master node API",
|
||||||
|
Action: masterAction,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "node",
|
||||||
|
Usage: "Runs node on local machine",
|
||||||
|
Action: nodeAction,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "print",
|
||||||
|
Usage: "Prints all discovered nodes",
|
||||||
|
Action: printAction,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
err := app.Run(os.Args)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
209
docs/docs.go
Normal file
209
docs/docs.go
Normal file
|
|
@ -0,0 +1,209 @@
|
||||||
|
// Package docs GENERATED BY SWAG; DO NOT EDIT
|
||||||
|
// This file was generated by swaggo/swag
|
||||||
|
package docs
|
||||||
|
|
||||||
|
import "github.com/swaggo/swag"
|
||||||
|
|
||||||
|
const docTemplate = `{
|
||||||
|
"schemes": {{ marshal .Schemes }},
|
||||||
|
"swagger": "2.0",
|
||||||
|
"info": {
|
||||||
|
"description": "{{escape .Description}}",
|
||||||
|
"title": "{{.Title}}",
|
||||||
|
"contact": {},
|
||||||
|
"version": "{{.Version}}"
|
||||||
|
},
|
||||||
|
"host": "{{.Host}}",
|
||||||
|
"basePath": "{{.BasePath}}",
|
||||||
|
"paths": {
|
||||||
|
"/nodes": {
|
||||||
|
"get": {
|
||||||
|
"security": [
|
||||||
|
{
|
||||||
|
"Bearer": []
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"description": "List of all discovered nodes and their labels.",
|
||||||
|
"produces": [
|
||||||
|
"application/json"
|
||||||
|
],
|
||||||
|
"summary": "List of nodes",
|
||||||
|
"responses": {
|
||||||
|
"200": {
|
||||||
|
"description": "List of nodes",
|
||||||
|
"schema": {
|
||||||
|
"type": "array",
|
||||||
|
"items": {
|
||||||
|
"$ref": "#/definitions/nodes.Node"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"401": {
|
||||||
|
"description": "Forbidden access",
|
||||||
|
"schema": {
|
||||||
|
"$ref": "#/definitions/api.Message"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"/nodes/{hostname}": {
|
||||||
|
"get": {
|
||||||
|
"security": [
|
||||||
|
{
|
||||||
|
"Bearer": []
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"description": "Return one nodes based on given hostname",
|
||||||
|
"produces": [
|
||||||
|
"application/json"
|
||||||
|
],
|
||||||
|
"summary": "Get node",
|
||||||
|
"parameters": [
|
||||||
|
{
|
||||||
|
"type": "string",
|
||||||
|
"description": "Node hostname",
|
||||||
|
"name": "hostname",
|
||||||
|
"in": "path",
|
||||||
|
"required": true
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"responses": {
|
||||||
|
"200": {
|
||||||
|
"description": "Node details",
|
||||||
|
"schema": {
|
||||||
|
"type": "array",
|
||||||
|
"items": {
|
||||||
|
"$ref": "#/definitions/nodes.Node"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"401": {
|
||||||
|
"description": "Forbidden access",
|
||||||
|
"schema": {
|
||||||
|
"$ref": "#/definitions/api.Message"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"post": {
|
||||||
|
"security": [
|
||||||
|
{
|
||||||
|
"Bearer": []
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"description": "Send new data or update existing data about a node",
|
||||||
|
"produces": [
|
||||||
|
"application/json"
|
||||||
|
],
|
||||||
|
"summary": "Refresh node",
|
||||||
|
"parameters": [
|
||||||
|
{
|
||||||
|
"type": "string",
|
||||||
|
"description": "Node hostname",
|
||||||
|
"name": "hostname",
|
||||||
|
"in": "path",
|
||||||
|
"required": true
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"description": "Node labels",
|
||||||
|
"name": "labels",
|
||||||
|
"in": "body",
|
||||||
|
"required": true,
|
||||||
|
"schema": {
|
||||||
|
"type": "array",
|
||||||
|
"items": {
|
||||||
|
"type": "string"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"description": "Key-value",
|
||||||
|
"name": "kv",
|
||||||
|
"in": "body",
|
||||||
|
"required": true,
|
||||||
|
"schema": {
|
||||||
|
"$ref": "#/definitions/nodes.KV"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"responses": {
|
||||||
|
"200": {
|
||||||
|
"description": "Node details",
|
||||||
|
"schema": {
|
||||||
|
"type": "array",
|
||||||
|
"items": {
|
||||||
|
"$ref": "#/definitions/nodes.Node"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"401": {
|
||||||
|
"description": "Forbidden access",
|
||||||
|
"schema": {
|
||||||
|
"$ref": "#/definitions/api.Message"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"definitions": {
|
||||||
|
"api.Message": {
|
||||||
|
"type": "object",
|
||||||
|
"properties": {
|
||||||
|
"message": {
|
||||||
|
"type": "string"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"nodes.KV": {
|
||||||
|
"type": "object",
|
||||||
|
"additionalProperties": {
|
||||||
|
"type": "string"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"nodes.Node": {
|
||||||
|
"type": "object",
|
||||||
|
"properties": {
|
||||||
|
"hostname": {
|
||||||
|
"type": "string"
|
||||||
|
},
|
||||||
|
"kv": {
|
||||||
|
"$ref": "#/definitions/nodes.KV"
|
||||||
|
},
|
||||||
|
"labels": {
|
||||||
|
"type": "array",
|
||||||
|
"items": {
|
||||||
|
"type": "string"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"last_update": {
|
||||||
|
"type": "integer"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"securityDefinitions": {
|
||||||
|
"Bearer": {
|
||||||
|
"type": "apiKey",
|
||||||
|
"name": "Authorization",
|
||||||
|
"in": "header"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}`
|
||||||
|
|
||||||
|
// SwaggerInfo holds exported Swagger Info so clients can modify it
|
||||||
|
var SwaggerInfo = &swag.Spec{
|
||||||
|
Version: "2.0",
|
||||||
|
Host: "",
|
||||||
|
BasePath: "/",
|
||||||
|
Schemes: []string{},
|
||||||
|
Title: "Lobby2 API",
|
||||||
|
Description: "API of Lobby 2 project that helps to discover and connect to other nodes and their services.",
|
||||||
|
InfoInstanceName: "swagger",
|
||||||
|
SwaggerTemplate: docTemplate,
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
swag.Register(SwaggerInfo.InstanceName(), SwaggerInfo)
|
||||||
|
}
|
||||||
185
docs/swagger.json
Normal file
185
docs/swagger.json
Normal file
|
|
@ -0,0 +1,185 @@
|
||||||
|
{
|
||||||
|
"swagger": "2.0",
|
||||||
|
"info": {
|
||||||
|
"description": "API of Lobby 2 project that helps to discover and connect to other nodes and their services.",
|
||||||
|
"title": "Lobby2 API",
|
||||||
|
"contact": {},
|
||||||
|
"version": "2.0"
|
||||||
|
},
|
||||||
|
"basePath": "/",
|
||||||
|
"paths": {
|
||||||
|
"/nodes": {
|
||||||
|
"get": {
|
||||||
|
"security": [
|
||||||
|
{
|
||||||
|
"Bearer": []
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"description": "List of all discovered nodes and their labels.",
|
||||||
|
"produces": [
|
||||||
|
"application/json"
|
||||||
|
],
|
||||||
|
"summary": "List of nodes",
|
||||||
|
"responses": {
|
||||||
|
"200": {
|
||||||
|
"description": "List of nodes",
|
||||||
|
"schema": {
|
||||||
|
"type": "array",
|
||||||
|
"items": {
|
||||||
|
"$ref": "#/definitions/nodes.Node"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"401": {
|
||||||
|
"description": "Forbidden access",
|
||||||
|
"schema": {
|
||||||
|
"$ref": "#/definitions/api.Message"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"/nodes/{hostname}": {
|
||||||
|
"get": {
|
||||||
|
"security": [
|
||||||
|
{
|
||||||
|
"Bearer": []
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"description": "Return one nodes based on given hostname",
|
||||||
|
"produces": [
|
||||||
|
"application/json"
|
||||||
|
],
|
||||||
|
"summary": "Get node",
|
||||||
|
"parameters": [
|
||||||
|
{
|
||||||
|
"type": "string",
|
||||||
|
"description": "Node hostname",
|
||||||
|
"name": "hostname",
|
||||||
|
"in": "path",
|
||||||
|
"required": true
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"responses": {
|
||||||
|
"200": {
|
||||||
|
"description": "Node details",
|
||||||
|
"schema": {
|
||||||
|
"type": "array",
|
||||||
|
"items": {
|
||||||
|
"$ref": "#/definitions/nodes.Node"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"401": {
|
||||||
|
"description": "Forbidden access",
|
||||||
|
"schema": {
|
||||||
|
"$ref": "#/definitions/api.Message"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"post": {
|
||||||
|
"security": [
|
||||||
|
{
|
||||||
|
"Bearer": []
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"description": "Send new data or update existing data about a node",
|
||||||
|
"produces": [
|
||||||
|
"application/json"
|
||||||
|
],
|
||||||
|
"summary": "Refresh node",
|
||||||
|
"parameters": [
|
||||||
|
{
|
||||||
|
"type": "string",
|
||||||
|
"description": "Node hostname",
|
||||||
|
"name": "hostname",
|
||||||
|
"in": "path",
|
||||||
|
"required": true
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"description": "Node labels",
|
||||||
|
"name": "labels",
|
||||||
|
"in": "body",
|
||||||
|
"required": true,
|
||||||
|
"schema": {
|
||||||
|
"type": "array",
|
||||||
|
"items": {
|
||||||
|
"type": "string"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"description": "Key-value",
|
||||||
|
"name": "kv",
|
||||||
|
"in": "body",
|
||||||
|
"required": true,
|
||||||
|
"schema": {
|
||||||
|
"$ref": "#/definitions/nodes.KV"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"responses": {
|
||||||
|
"200": {
|
||||||
|
"description": "Node details",
|
||||||
|
"schema": {
|
||||||
|
"type": "array",
|
||||||
|
"items": {
|
||||||
|
"$ref": "#/definitions/nodes.Node"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"401": {
|
||||||
|
"description": "Forbidden access",
|
||||||
|
"schema": {
|
||||||
|
"$ref": "#/definitions/api.Message"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"definitions": {
|
||||||
|
"api.Message": {
|
||||||
|
"type": "object",
|
||||||
|
"properties": {
|
||||||
|
"message": {
|
||||||
|
"type": "string"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"nodes.KV": {
|
||||||
|
"type": "object",
|
||||||
|
"additionalProperties": {
|
||||||
|
"type": "string"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"nodes.Node": {
|
||||||
|
"type": "object",
|
||||||
|
"properties": {
|
||||||
|
"hostname": {
|
||||||
|
"type": "string"
|
||||||
|
},
|
||||||
|
"kv": {
|
||||||
|
"$ref": "#/definitions/nodes.KV"
|
||||||
|
},
|
||||||
|
"labels": {
|
||||||
|
"type": "array",
|
||||||
|
"items": {
|
||||||
|
"type": "string"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"last_update": {
|
||||||
|
"type": "integer"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"securityDefinitions": {
|
||||||
|
"Bearer": {
|
||||||
|
"type": "apiKey",
|
||||||
|
"name": "Authorization",
|
||||||
|
"in": "header"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
119
docs/swagger.yaml
Normal file
119
docs/swagger.yaml
Normal file
|
|
@ -0,0 +1,119 @@
|
||||||
|
basePath: /
|
||||||
|
definitions:
|
||||||
|
api.Message:
|
||||||
|
properties:
|
||||||
|
message:
|
||||||
|
type: string
|
||||||
|
type: object
|
||||||
|
nodes.KV:
|
||||||
|
additionalProperties:
|
||||||
|
type: string
|
||||||
|
type: object
|
||||||
|
nodes.Node:
|
||||||
|
properties:
|
||||||
|
hostname:
|
||||||
|
type: string
|
||||||
|
kv:
|
||||||
|
$ref: '#/definitions/nodes.KV'
|
||||||
|
labels:
|
||||||
|
items:
|
||||||
|
type: string
|
||||||
|
type: array
|
||||||
|
last_update:
|
||||||
|
type: integer
|
||||||
|
type: object
|
||||||
|
info:
|
||||||
|
contact: {}
|
||||||
|
description: API of Lobby 2 project that helps to discover and connect to other
|
||||||
|
nodes and their services.
|
||||||
|
title: Lobby2 API
|
||||||
|
version: "2.0"
|
||||||
|
paths:
|
||||||
|
/nodes:
|
||||||
|
get:
|
||||||
|
description: List of all discovered nodes and their labels.
|
||||||
|
produces:
|
||||||
|
- application/json
|
||||||
|
responses:
|
||||||
|
"200":
|
||||||
|
description: List of nodes
|
||||||
|
schema:
|
||||||
|
items:
|
||||||
|
$ref: '#/definitions/nodes.Node'
|
||||||
|
type: array
|
||||||
|
"401":
|
||||||
|
description: Forbidden access
|
||||||
|
schema:
|
||||||
|
$ref: '#/definitions/api.Message'
|
||||||
|
security:
|
||||||
|
- Bearer: []
|
||||||
|
summary: List of nodes
|
||||||
|
/nodes/{hostname}:
|
||||||
|
get:
|
||||||
|
description: Return one nodes based on given hostname
|
||||||
|
parameters:
|
||||||
|
- description: Node hostname
|
||||||
|
in: path
|
||||||
|
name: hostname
|
||||||
|
required: true
|
||||||
|
type: string
|
||||||
|
produces:
|
||||||
|
- application/json
|
||||||
|
responses:
|
||||||
|
"200":
|
||||||
|
description: Node details
|
||||||
|
schema:
|
||||||
|
items:
|
||||||
|
$ref: '#/definitions/nodes.Node'
|
||||||
|
type: array
|
||||||
|
"401":
|
||||||
|
description: Forbidden access
|
||||||
|
schema:
|
||||||
|
$ref: '#/definitions/api.Message'
|
||||||
|
security:
|
||||||
|
- Bearer: []
|
||||||
|
summary: Get node
|
||||||
|
post:
|
||||||
|
description: Send new data or update existing data about a node
|
||||||
|
parameters:
|
||||||
|
- description: Node hostname
|
||||||
|
in: path
|
||||||
|
name: hostname
|
||||||
|
required: true
|
||||||
|
type: string
|
||||||
|
- description: Node labels
|
||||||
|
in: body
|
||||||
|
name: labels
|
||||||
|
required: true
|
||||||
|
schema:
|
||||||
|
items:
|
||||||
|
type: string
|
||||||
|
type: array
|
||||||
|
- description: Key-value
|
||||||
|
in: body
|
||||||
|
name: kv
|
||||||
|
required: true
|
||||||
|
schema:
|
||||||
|
$ref: '#/definitions/nodes.KV'
|
||||||
|
produces:
|
||||||
|
- application/json
|
||||||
|
responses:
|
||||||
|
"200":
|
||||||
|
description: Node details
|
||||||
|
schema:
|
||||||
|
items:
|
||||||
|
$ref: '#/definitions/nodes.Node'
|
||||||
|
type: array
|
||||||
|
"401":
|
||||||
|
description: Forbidden access
|
||||||
|
schema:
|
||||||
|
$ref: '#/definitions/api.Message'
|
||||||
|
security:
|
||||||
|
- Bearer: []
|
||||||
|
summary: Refresh node
|
||||||
|
securityDefinitions:
|
||||||
|
Bearer:
|
||||||
|
in: header
|
||||||
|
name: Authorization
|
||||||
|
type: apiKey
|
||||||
|
swagger: "2.0"
|
||||||
46
go.mod
Normal file
46
go.mod
Normal file
|
|
@ -0,0 +1,46 @@
|
||||||
|
module gitea.ceperka.net/rosti/lobby2
|
||||||
|
|
||||||
|
go 1.20
|
||||||
|
|
||||||
|
require (
|
||||||
|
github.com/kelseyhightower/envconfig v1.4.0
|
||||||
|
github.com/labstack/echo/v4 v4.12.0
|
||||||
|
github.com/puzpuzpuz/xsync/v3 v3.4.0
|
||||||
|
github.com/stretchr/testify v1.8.4
|
||||||
|
github.com/swaggo/echo-swagger v1.4.1
|
||||||
|
github.com/swaggo/swag v1.16.3
|
||||||
|
github.com/urfave/cli/v2 v2.3.0
|
||||||
|
)
|
||||||
|
|
||||||
|
require (
|
||||||
|
github.com/KyleBanks/depth v1.2.1 // indirect
|
||||||
|
github.com/PuerkitoBio/purell v1.1.1 // indirect
|
||||||
|
github.com/PuerkitoBio/urlesc v0.0.0-20170810143723-de5bf2ad4578 // indirect
|
||||||
|
github.com/cpuguy83/go-md2man/v2 v2.0.0-20190314233015-f79a8a8ca69d // indirect
|
||||||
|
github.com/davecgh/go-spew v1.1.1 // indirect
|
||||||
|
github.com/ghodss/yaml v1.0.0 // indirect
|
||||||
|
github.com/go-openapi/jsonpointer v0.19.5 // indirect
|
||||||
|
github.com/go-openapi/jsonreference v0.19.6 // indirect
|
||||||
|
github.com/go-openapi/spec v0.20.4 // indirect
|
||||||
|
github.com/go-openapi/swag v0.19.15 // indirect
|
||||||
|
github.com/golang-jwt/jwt v3.2.2+incompatible // indirect
|
||||||
|
github.com/josharian/intern v1.0.0 // indirect
|
||||||
|
github.com/labstack/gommon v0.4.2 // indirect
|
||||||
|
github.com/mailru/easyjson v0.7.7 // indirect
|
||||||
|
github.com/mattn/go-colorable v0.1.13 // indirect
|
||||||
|
github.com/mattn/go-isatty v0.0.20 // indirect
|
||||||
|
github.com/pmezard/go-difflib v1.0.0 // indirect
|
||||||
|
github.com/russross/blackfriday/v2 v2.0.1 // indirect
|
||||||
|
github.com/shurcooL/sanitized_anchor_name v1.0.0 // indirect
|
||||||
|
github.com/swaggo/files/v2 v2.0.0 // indirect
|
||||||
|
github.com/valyala/bytebufferpool v1.0.0 // indirect
|
||||||
|
github.com/valyala/fasttemplate v1.2.2 // indirect
|
||||||
|
golang.org/x/crypto v0.22.0 // indirect
|
||||||
|
golang.org/x/net v0.24.0 // indirect
|
||||||
|
golang.org/x/sys v0.19.0 // indirect
|
||||||
|
golang.org/x/text v0.14.0 // indirect
|
||||||
|
golang.org/x/time v0.5.0 // indirect
|
||||||
|
golang.org/x/tools v0.7.0 // indirect
|
||||||
|
gopkg.in/yaml.v2 v2.4.0 // indirect
|
||||||
|
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||||
|
)
|
||||||
111
go.sum
Normal file
111
go.sum
Normal file
|
|
@ -0,0 +1,111 @@
|
||||||
|
github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=
|
||||||
|
github.com/KyleBanks/depth v1.2.1 h1:5h8fQADFrWtarTdtDudMmGsC7GPbOAu6RVB3ffsVFHc=
|
||||||
|
github.com/KyleBanks/depth v1.2.1/go.mod h1:jzSb9d0L43HxTQfT+oSA1EEp2q+ne2uh6XgeJcm8brE=
|
||||||
|
github.com/PuerkitoBio/purell v1.1.1 h1:WEQqlqaGbrPkxLJWfBwQmfEAE1Z7ONdDLqrN38tNFfI=
|
||||||
|
github.com/PuerkitoBio/purell v1.1.1/go.mod h1:c11w/QuzBsJSee3cPx9rAFu61PvFxuPbtSwDGJws/X0=
|
||||||
|
github.com/PuerkitoBio/urlesc v0.0.0-20170810143723-de5bf2ad4578 h1:d+Bc7a5rLufV/sSk/8dngufqelfh6jnri85riMAaF/M=
|
||||||
|
github.com/PuerkitoBio/urlesc v0.0.0-20170810143723-de5bf2ad4578/go.mod h1:uGdkoq3SwY9Y+13GIhn11/XLaGBb4BfwItxLd5jeuXE=
|
||||||
|
github.com/cpuguy83/go-md2man/v2 v2.0.0-20190314233015-f79a8a8ca69d h1:U+s90UTSYgptZMwQh2aRr3LuazLJIa+Pg3Kc1ylSYVY=
|
||||||
|
github.com/cpuguy83/go-md2man/v2 v2.0.0-20190314233015-f79a8a8ca69d/go.mod h1:maD7wRr/U5Z6m/iR4s+kqSMx2CaBsrgA7czyZG/E6dU=
|
||||||
|
github.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E=
|
||||||
|
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
|
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||||
|
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
|
github.com/ghodss/yaml v1.0.0 h1:wQHKEahhL6wmXdzwWG11gIVCkOv05bNOh+Rxn0yngAk=
|
||||||
|
github.com/ghodss/yaml v1.0.0/go.mod h1:4dBDuWmgqj2HViK6kFavaiC9ZROes6MMH2rRYeMEF04=
|
||||||
|
github.com/go-openapi/jsonpointer v0.19.3/go.mod h1:Pl9vOtqEWErmShwVjC8pYs9cog34VGT37dQOVbmoatg=
|
||||||
|
github.com/go-openapi/jsonpointer v0.19.5 h1:gZr+CIYByUqjcgeLXnQu2gHYQC9o73G2XUeOFYEICuY=
|
||||||
|
github.com/go-openapi/jsonpointer v0.19.5/go.mod h1:Pl9vOtqEWErmShwVjC8pYs9cog34VGT37dQOVbmoatg=
|
||||||
|
github.com/go-openapi/jsonreference v0.19.6 h1:UBIxjkht+AWIgYzCDSv2GN+E/togfwXUJFRTWhl2Jjs=
|
||||||
|
github.com/go-openapi/jsonreference v0.19.6/go.mod h1:diGHMEHg2IqXZGKxqyvWdfWU/aim5Dprw5bqpKkTvns=
|
||||||
|
github.com/go-openapi/spec v0.20.4 h1:O8hJrt0UMnhHcluhIdUgCLRWyM2x7QkBXRvOs7m+O1M=
|
||||||
|
github.com/go-openapi/spec v0.20.4/go.mod h1:faYFR1CvsJZ0mNsmsphTMSoRrNV3TEDoAM7FOEWeq8I=
|
||||||
|
github.com/go-openapi/swag v0.19.5/go.mod h1:POnQmlKehdgb5mhVOsnJFsivZCEZ/vjK9gh66Z9tfKk=
|
||||||
|
github.com/go-openapi/swag v0.19.15 h1:D2NRCBzS9/pEY3gP9Nl8aDqGUcPFrwG2p+CNFrLyrCM=
|
||||||
|
github.com/go-openapi/swag v0.19.15/go.mod h1:QYRuS/SOXUCsnplDa677K7+DxSOj6IPNl/eQntq43wQ=
|
||||||
|
github.com/golang-jwt/jwt v3.2.2+incompatible h1:IfV12K8xAKAnZqdXVzCZ+TOjboZ2keLg81eXfW3O+oY=
|
||||||
|
github.com/golang-jwt/jwt v3.2.2+incompatible/go.mod h1:8pz2t5EyA70fFQQSrl6XZXzqecmYZeUEB8OUGHkxJ+I=
|
||||||
|
github.com/josharian/intern v1.0.0 h1:vlS4z54oSdjm0bgjRigI+G1HpF+tI+9rE5LLzOg8HmY=
|
||||||
|
github.com/josharian/intern v1.0.0/go.mod h1:5DoeVV0s6jJacbCEi61lwdGj/aVlrQvzHFFd8Hwg//Y=
|
||||||
|
github.com/kelseyhightower/envconfig v1.4.0 h1:Im6hONhd3pLkfDFsbRgu68RDNkGF1r3dvMUtDTo2cv8=
|
||||||
|
github.com/kelseyhightower/envconfig v1.4.0/go.mod h1:cccZRl6mQpaq41TPp5QxidR+Sa3axMbJDNb//FQX6Gg=
|
||||||
|
github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
|
||||||
|
github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
|
||||||
|
github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
|
||||||
|
github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
|
||||||
|
github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
|
||||||
|
github.com/labstack/echo/v4 v4.12.0 h1:IKpw49IMryVB2p1a4dzwlhP1O2Tf2E0Ir/450lH+kI0=
|
||||||
|
github.com/labstack/echo/v4 v4.12.0/go.mod h1:UP9Cr2DJXbOK3Kr9ONYzNowSh7HP0aG0ShAyycHSJvM=
|
||||||
|
github.com/labstack/gommon v0.4.2 h1:F8qTUNXgG1+6WQmqoUWnz8WiEU60mXVVw0P4ht1WRA0=
|
||||||
|
github.com/labstack/gommon v0.4.2/go.mod h1:QlUFxVM+SNXhDL/Z7YhocGIBYOiwB0mXm1+1bAPHPyU=
|
||||||
|
github.com/mailru/easyjson v0.0.0-20190614124828-94de47d64c63/go.mod h1:C1wdFJiN94OJF2b5HbByQZoLdCWB1Yqtg26g4irojpc=
|
||||||
|
github.com/mailru/easyjson v0.0.0-20190626092158-b2ccc519800e/go.mod h1:C1wdFJiN94OJF2b5HbByQZoLdCWB1Yqtg26g4irojpc=
|
||||||
|
github.com/mailru/easyjson v0.7.6/go.mod h1:xzfreul335JAWq5oZzymOObrkdz5UnU4kGfJJLY9Nlc=
|
||||||
|
github.com/mailru/easyjson v0.7.7 h1:UGYAvKxe3sBsEDzO8ZeWOSlIQfWFlxbzLZe7hwFURr0=
|
||||||
|
github.com/mailru/easyjson v0.7.7/go.mod h1:xzfreul335JAWq5oZzymOObrkdz5UnU4kGfJJLY9Nlc=
|
||||||
|
github.com/mattn/go-colorable v0.1.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA=
|
||||||
|
github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg=
|
||||||
|
github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM=
|
||||||
|
github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
|
||||||
|
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
||||||
|
github.com/niemeyer/pretty v0.0.0-20200227124842-a10e7caefd8e h1:fD57ERR4JtEqsWbfPhv4DMiApHyliiK5xCTNVSPiaAs=
|
||||||
|
github.com/niemeyer/pretty v0.0.0-20200227124842-a10e7caefd8e/go.mod h1:zD1mROLANZcx1PVRCS0qkT7pwLkGfwJo4zjcN/Tysno=
|
||||||
|
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||||
|
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||||
|
github.com/puzpuzpuz/xsync/v3 v3.4.0 h1:DuVBAdXuGFHv8adVXjWWZ63pJq+NRXOWVXlKDBZ+mJ4=
|
||||||
|
github.com/puzpuzpuz/xsync/v3 v3.4.0/go.mod h1:VjzYrABPabuM4KyBh1Ftq6u8nhwY5tBPKP9jpmh0nnA=
|
||||||
|
github.com/russross/blackfriday/v2 v2.0.1 h1:lPqVAte+HuHNfhJ/0LC98ESWRz8afy9tM/0RK8m9o+Q=
|
||||||
|
github.com/russross/blackfriday/v2 v2.0.1/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
|
||||||
|
github.com/shurcooL/sanitized_anchor_name v1.0.0 h1:PdmoCO6wvbs+7yrJyMORt4/BmY5IYyJwS/kOiWx8mHo=
|
||||||
|
github.com/shurcooL/sanitized_anchor_name v1.0.0/go.mod h1:1NzhyTcUVG4SuEtjjoZeVRXNmyL/1OwPU0+IJeTBvfc=
|
||||||
|
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||||
|
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
|
||||||
|
github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||||
|
github.com/stretchr/testify v1.8.4 h1:CcVxjf3Q8PM0mHUKJCdn+eZZtm5yQwehR5yeSVQQcUk=
|
||||||
|
github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
|
||||||
|
github.com/swaggo/echo-swagger v1.4.1 h1:Yf0uPaJWp1uRtDloZALyLnvdBeoEL5Kc7DtnjzO/TUk=
|
||||||
|
github.com/swaggo/echo-swagger v1.4.1/go.mod h1:C8bSi+9yH2FLZsnhqMZLIZddpUxZdBYuNHbtaS1Hljc=
|
||||||
|
github.com/swaggo/files/v2 v2.0.0 h1:hmAt8Dkynw7Ssz46F6pn8ok6YmGZqHSVLZ+HQM7i0kw=
|
||||||
|
github.com/swaggo/files/v2 v2.0.0/go.mod h1:24kk2Y9NYEJ5lHuCra6iVwkMjIekMCaFq/0JQj66kyM=
|
||||||
|
github.com/swaggo/swag v1.16.3 h1:PnCYjPCah8FK4I26l2F/KQ4yz3sILcVUN3cTlBFA9Pg=
|
||||||
|
github.com/swaggo/swag v1.16.3/go.mod h1:DImHIuOFXKpMFAQjcC7FG4m3Dg4+QuUgUzJmKjI/gRk=
|
||||||
|
github.com/urfave/cli/v2 v2.3.0 h1:qph92Y649prgesehzOrQjdWyxFOp/QVM+6imKHad91M=
|
||||||
|
github.com/urfave/cli/v2 v2.3.0/go.mod h1:LJmUH05zAU44vOAcrfzZQKsZbVcdbOG8rtL3/XcUArI=
|
||||||
|
github.com/valyala/bytebufferpool v1.0.0 h1:GqA5TC/0021Y/b9FG4Oi9Mr3q7XYx6KllzawFIhcdPw=
|
||||||
|
github.com/valyala/bytebufferpool v1.0.0/go.mod h1:6bBcMArwyJ5K/AmCkWv1jt77kVWyCJ6HpOuEn7z0Csc=
|
||||||
|
github.com/valyala/fasttemplate v1.2.2 h1:lxLXG0uE3Qnshl9QyaK6XJxMXlQZELvChBOCmQD0Loo=
|
||||||
|
github.com/valyala/fasttemplate v1.2.2/go.mod h1:KHLXt3tVN2HBp8eijSv/kGJopbvo7S+qRAEEKiv+SiQ=
|
||||||
|
golang.org/x/crypto v0.22.0 h1:g1v0xeRhjcugydODzvb3mEM9SQ0HGp9s/nh3COQ/C30=
|
||||||
|
golang.org/x/crypto v0.22.0/go.mod h1:vr6Su+7cTlO45qkww3VDJlzDn0ctJvRgYbC2NvXHt+M=
|
||||||
|
golang.org/x/mod v0.9.0 h1:KENHtAZL2y3NLMYZeHY9DW8HW8V+kQyJsY/V9JlKvCs=
|
||||||
|
golang.org/x/net v0.0.0-20210421230115-4e50805a0758/go.mod h1:72T/g9IO56b78aLF+1Kcs5dz7/ng1VjMUvfKvpfy+jM=
|
||||||
|
golang.org/x/net v0.24.0 h1:1PcaxkF854Fu3+lvBIx5SYn9wRlBzzcnHZSiaFFAb0w=
|
||||||
|
golang.org/x/net v0.24.0/go.mod h1:2Q7sJY5mzlzWjKtYUEXSlBWCdyaioyXzRB2RtU8KVE8=
|
||||||
|
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
|
golang.org/x/sys v0.0.0-20210420072515-93ed5bcd2bfe/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
|
golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/sys v0.19.0 h1:q5f1RH2jigJ1MoAWp2KTp3gm5zAGFUTarQZ5U386+4o=
|
||||||
|
golang.org/x/sys v0.19.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
|
||||||
|
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
|
||||||
|
golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
|
||||||
|
golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
|
||||||
|
golang.org/x/text v0.14.0 h1:ScX5w1eTa3QqT8oi6+ziP7dTV1S2+ALU0bI+0zXKWiQ=
|
||||||
|
golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
|
||||||
|
golang.org/x/time v0.5.0 h1:o7cqy6amK/52YcAKIPlM3a+Fpj35zvRj2TP+e1xFSfk=
|
||||||
|
golang.org/x/time v0.5.0/go.mod h1:3BpzKBy/shNhVucY/MWOyx10tF3SFh9QdLuxbVysPQM=
|
||||||
|
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||||
|
golang.org/x/tools v0.7.0 h1:W4OVu8VVOaIO0yzWMNdepAulS7YfoS3Zabrm8DOXXU4=
|
||||||
|
golang.org/x/tools v0.7.0/go.mod h1:4pg6aUX35JBAogB10C9AtvVL+qowtN4pT3CGSQex14s=
|
||||||
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
|
gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
|
gopkg.in/check.v1 v1.0.0-20200227125254-8fa46927fb4f h1:BLraFXnmrev5lT+xlilqcH8XK9/i0At2xKjWk4p6zsU=
|
||||||
|
gopkg.in/check.v1 v1.0.0-20200227125254-8fa46927fb4f/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
|
gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
|
||||||
|
gopkg.in/yaml.v2 v2.2.3/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
|
||||||
|
gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=
|
||||||
|
gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=
|
||||||
|
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
gopkg.in/yaml.v3 v3.0.0-20200615113413-eeeca48fe776/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||||
|
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
176
nodes/main.go
Normal file
176
nodes/main.go
Normal file
|
|
@ -0,0 +1,176 @@
|
||||||
|
package nodes
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
"path"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/puzpuzpuz/xsync/v3"
|
||||||
|
)
|
||||||
|
|
||||||
|
const dumpIntervalSeconds = 120
|
||||||
|
|
||||||
|
// Base nodes structure that covers all operations above nodes.
|
||||||
|
type NodesProcessor struct {
|
||||||
|
nodes *xsync.MapOf[string, Node]
|
||||||
|
dumpPath string
|
||||||
|
garbageCollectAfterSeconds int64
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewNodesProcessor(dumpPath string, garbageCollectAfterSeconds int64) *NodesProcessor {
|
||||||
|
p := &NodesProcessor{
|
||||||
|
dumpPath: dumpPath,
|
||||||
|
garbageCollectAfterSeconds: garbageCollectAfterSeconds,
|
||||||
|
nodes: xsync.NewMapOf[string, Node](),
|
||||||
|
}
|
||||||
|
|
||||||
|
err := p.Load()
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("failed to load nodes: %v\n", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return p
|
||||||
|
}
|
||||||
|
|
||||||
|
func (np *NodesProcessor) DumpLoop() {
|
||||||
|
log.Println(".. starting dump loop")
|
||||||
|
time.Sleep(time.Second * dumpIntervalSeconds)
|
||||||
|
for {
|
||||||
|
log.Println(".. dumping nodes")
|
||||||
|
err := np.Dump()
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("failed to dump nodes: %v\n", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
time.Sleep(time.Second * dumpIntervalSeconds)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// GarbageCollection removes nodes that haven't been updated for more than garbageCollectAfterSeconds.
|
||||||
|
func (np *NodesProcessor) GarbageCollection() {
|
||||||
|
for {
|
||||||
|
np.garbageCollect()
|
||||||
|
time.Sleep(time.Second * 10)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (np *NodesProcessor) garbageCollect() {
|
||||||
|
np.nodes.Range(func(key string, value Node) bool {
|
||||||
|
if time.Now().Unix()-value.LastUpdate > np.garbageCollectAfterSeconds {
|
||||||
|
np.nodes.Delete(value.HostName)
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// Dumps content of np.nodes to np.dumpPath.
|
||||||
|
func (np *NodesProcessor) Dump() error {
|
||||||
|
ns := []Node{}
|
||||||
|
np.nodes.Range(func(key string, value Node) bool {
|
||||||
|
ns = append(ns, value)
|
||||||
|
return true
|
||||||
|
})
|
||||||
|
|
||||||
|
body, err := json.Marshal(ns)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to marshal nodes: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create directory if it doesn't exist.
|
||||||
|
if strings.Contains(np.dumpPath, "/") {
|
||||||
|
path := np.dumpPath[:strings.LastIndex(np.dumpPath, "/")]
|
||||||
|
|
||||||
|
if len(path) != 0 {
|
||||||
|
err = os.MkdirAll(path, 0755)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to create directory for nodes.json: %w", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
err = os.WriteFile(np.dumpPath, body, 0644)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to write nodes.json: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Loads content of np.dumpPath to np.nodes.
|
||||||
|
func (np *NodesProcessor) Load() error {
|
||||||
|
filePath := path.Join(np.dumpPath)
|
||||||
|
_, err := os.Stat(filePath)
|
||||||
|
if os.IsNotExist(err) {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
body, err := os.ReadFile(np.dumpPath)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to read nodes.json: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
np.Reset()
|
||||||
|
|
||||||
|
ns := []Node{}
|
||||||
|
|
||||||
|
err = json.Unmarshal(body, &ns)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to unmarshal nodes: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, n := range ns {
|
||||||
|
np.nodes.Store(n.HostName, n)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (np *NodesProcessor) Reset() {
|
||||||
|
np.nodes.Clear()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Returns string with Prometheus metrics
|
||||||
|
// TODO: finish this
|
||||||
|
func (np *NodesProcessor) GetMetrics() string {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
// List returns all nodes.
|
||||||
|
func (np *NodesProcessor) List() Nodes {
|
||||||
|
nodes := Nodes{}
|
||||||
|
|
||||||
|
np.nodes.Range(func(key string, value Node) bool {
|
||||||
|
nodes = append(nodes, value)
|
||||||
|
return true
|
||||||
|
})
|
||||||
|
|
||||||
|
return nodes
|
||||||
|
}
|
||||||
|
|
||||||
|
// Get returns a node by hostname.
|
||||||
|
func (np *NodesProcessor) Get(hostname string) (Node, bool) {
|
||||||
|
node, ok := np.nodes.Load(hostname)
|
||||||
|
return node, ok
|
||||||
|
}
|
||||||
|
|
||||||
|
// Refresh updates node with hostname and labels. If it doesn't exists it's created.
|
||||||
|
func (np *NodesProcessor) Refresh(hostname string, labels Labels, kv KV) {
|
||||||
|
node, ok := np.nodes.Load(hostname)
|
||||||
|
if !ok {
|
||||||
|
node = Node{}
|
||||||
|
}
|
||||||
|
node.LastUpdate = time.Now().Unix()
|
||||||
|
node.HostName = hostname
|
||||||
|
node.Labels = labels
|
||||||
|
node.KV = kv
|
||||||
|
np.nodes.Store(hostname, node)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Drop removes node with given hostname.
|
||||||
|
func (np *NodesProcessor) Drop(hostname string) {
|
||||||
|
np.nodes.Delete(hostname)
|
||||||
|
}
|
||||||
92
nodes/main_test.go
Normal file
92
nodes/main_test.go
Normal file
|
|
@ -0,0 +1,92 @@
|
||||||
|
package nodes
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
)
|
||||||
|
|
||||||
|
var np *NodesProcessor
|
||||||
|
|
||||||
|
const testDumpFile = "nodes.json"
|
||||||
|
|
||||||
|
func TestMain(m *testing.M) {
|
||||||
|
np = NewNodesProcessor(testDumpFile, 2)
|
||||||
|
|
||||||
|
os.Exit(m.Run())
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestNodesProcessor_GarbageCollect(t *testing.T) {
|
||||||
|
np.Refresh("test", Labels{"mylabel"}, KV{"mykey": "my value"})
|
||||||
|
assert.Equal(t, 1, len(np.List()))
|
||||||
|
time.Sleep(5 * time.Second)
|
||||||
|
np.garbageCollect()
|
||||||
|
assert.Equal(t, 0, len(np.List()))
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestNodesProcessor_Refresh(t *testing.T) {
|
||||||
|
np.Reset()
|
||||||
|
|
||||||
|
np.Refresh("test", Labels{"mylabel"}, KV{"mykey": "my value"})
|
||||||
|
np.Refresh("test2", Labels{"mylabel3"}, KV{"mykey3": "my value3"})
|
||||||
|
|
||||||
|
nodes := np.List()
|
||||||
|
assert.Equal(t, "test", nodes[0].HostName)
|
||||||
|
assert.Contains(t, nodes[0].Labels, "mylabel")
|
||||||
|
assert.Equal(t, "test2", nodes[1].HostName)
|
||||||
|
assert.Contains(t, nodes[1].Labels, "mylabel3")
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestNodesProcessor_DumpLoad(t *testing.T) {
|
||||||
|
np.Reset()
|
||||||
|
|
||||||
|
np.Refresh("test", Labels{"mylabel"}, KV{"mykey": "my value"})
|
||||||
|
np.Refresh("test2", Labels{"mylabel3"}, KV{"mykey3": "my value3"})
|
||||||
|
|
||||||
|
os.Remove(testDumpFile)
|
||||||
|
|
||||||
|
np.Dump()
|
||||||
|
np.Reset()
|
||||||
|
assert.Equal(t, 0, len(np.List()))
|
||||||
|
|
||||||
|
np.Load()
|
||||||
|
assert.Equal(t, 2, len(np.List()))
|
||||||
|
|
||||||
|
nodes := np.List()
|
||||||
|
assert.Len(t, nodes, 2)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestNodesProcessor_List(t *testing.T) {
|
||||||
|
np.Reset()
|
||||||
|
|
||||||
|
np.Refresh("test", Labels{"mylabel"}, KV{"mykey": "my value"})
|
||||||
|
nodes := np.List()
|
||||||
|
assert.Equal(t, "test", nodes[0].HostName)
|
||||||
|
assert.Contains(t, nodes[0].Labels, "mylabel")
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestNodesProcessor_Get(t *testing.T) {
|
||||||
|
np.Reset()
|
||||||
|
np.Refresh("test", Labels{"mylabel"}, KV{"mykey": "my value"})
|
||||||
|
node, _ := np.Get("test")
|
||||||
|
|
||||||
|
assert.Contains(t, node.Labels, "mylabel")
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestNodesProcessor_Drop(t *testing.T) {
|
||||||
|
np.Reset()
|
||||||
|
np.Refresh("test", Labels{"mylabel"}, KV{"mykey": "my value"})
|
||||||
|
node, _ := np.Get("test")
|
||||||
|
|
||||||
|
assert.Contains(t, node.Labels, "mylabel")
|
||||||
|
|
||||||
|
np.Drop("test")
|
||||||
|
assert.Zero(t, len(np.List()))
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO: TestNodesProcessor_GetMetrics
|
||||||
14
nodes/types.go
Normal file
14
nodes/types.go
Normal file
|
|
@ -0,0 +1,14 @@
|
||||||
|
package nodes
|
||||||
|
|
||||||
|
type KV map[string]string
|
||||||
|
type Labels []string
|
||||||
|
type JSONData map[string]interface{}
|
||||||
|
|
||||||
|
type Node struct {
|
||||||
|
LastUpdate int64 `json:"last_update,omitempty"`
|
||||||
|
HostName string `json:"hostname"`
|
||||||
|
Labels Labels `json:"labels,omitempty"`
|
||||||
|
KV KV `json:"kv,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type Nodes []Node
|
||||||
178
refresher/main.go
Normal file
178
refresher/main.go
Normal file
|
|
@ -0,0 +1,178 @@
|
||||||
|
package refresher
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"net/http"
|
||||||
|
"os"
|
||||||
|
"path"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"gitea.ceperka.net/rosti/lobby2/nodes"
|
||||||
|
)
|
||||||
|
|
||||||
|
const refreshIntervalSeconds = 15
|
||||||
|
const nodeFileName = "node.json"
|
||||||
|
|
||||||
|
// Refresher loads local node info and sends them to the master node.
|
||||||
|
type Refresher struct {
|
||||||
|
configPath string
|
||||||
|
nodeDirPath string
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewRefresher(nodeDirPath string, configPath string) *Refresher {
|
||||||
|
return &Refresher{
|
||||||
|
nodeDirPath: nodeDirPath,
|
||||||
|
configPath: configPath,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Run loop the process that updates node in the master node.
|
||||||
|
func (r *Refresher) Loop() {
|
||||||
|
log.Println("Start refresher loop")
|
||||||
|
|
||||||
|
for {
|
||||||
|
log.Println("Refreshing node")
|
||||||
|
err := r.Refresh()
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("failed to refresh node: %v\n", err)
|
||||||
|
}
|
||||||
|
time.Sleep(time.Second * refreshIntervalSeconds)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Refresh loads labels from the local filesystem and sends them to the master node.
|
||||||
|
func (r *Refresher) Refresh() error {
|
||||||
|
// Load config
|
||||||
|
cfg, err := r.getConfig()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Load labels
|
||||||
|
node, err := r.loadNode()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
nodeBytes, err := json.Marshal(node)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to marshal labels: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Send labels to master
|
||||||
|
req, err := http.NewRequest("POST", fmt.Sprintf("%s://%s:%d/nodes/%s", cfg.MasterProto, cfg.MasterHost, cfg.MasterPort, node.HostName), bytes.NewBuffer(nodeBytes))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
req.Header.Set("Content-Type", "application/json")
|
||||||
|
req.Header.Set("Authorization", "Bearer "+cfg.MasterToken)
|
||||||
|
|
||||||
|
client := &http.Client{}
|
||||||
|
resp, err := client.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
if resp.StatusCode != http.StatusNoContent {
|
||||||
|
return fmt.Errorf("unexpected status: %s", resp.Status)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
// Returns the node config.
|
||||||
|
func (r *Refresher) getConfig() (NodeConfig, error) {
|
||||||
|
cfg := NodeConfig{}
|
||||||
|
|
||||||
|
content, err := os.ReadFile(r.configPath)
|
||||||
|
if err != nil {
|
||||||
|
return cfg, fmt.Errorf("failed to read config file: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
err = json.Unmarshal(content, &cfg)
|
||||||
|
if err != nil {
|
||||||
|
return cfg, fmt.Errorf("failed to unmarshal config: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if cfg.MasterProto == "" {
|
||||||
|
cfg.MasterProto = "http"
|
||||||
|
}
|
||||||
|
|
||||||
|
if cfg.MasterPort == 0 {
|
||||||
|
cfg.MasterPort = 1352
|
||||||
|
}
|
||||||
|
|
||||||
|
return cfg, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO: rewrite this to load Node structure
|
||||||
|
func (r *Refresher) loadNode() (*nodes.Node, error) {
|
||||||
|
filePath := path.Join(r.nodeDirPath, nodeFileName)
|
||||||
|
_, err := os.Stat(filePath)
|
||||||
|
if os.IsNotExist(err) {
|
||||||
|
err = r.initNodeFile()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
content, err := os.ReadFile(path.Join(r.nodeDirPath, nodeFileName))
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to read file: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
node := &nodes.Node{}
|
||||||
|
err = json.Unmarshal(content, &node)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to unmarshal node: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return node, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *Refresher) initNodeFile() error {
|
||||||
|
err := r.createNodePath()
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to create node path: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
hostname, err := os.Hostname()
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to get hostname: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
node := &nodes.Node{
|
||||||
|
HostName: hostname,
|
||||||
|
}
|
||||||
|
|
||||||
|
nodeBytes, err := json.MarshalIndent(node, "", " ")
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to marshal node: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
err = os.WriteFile(path.Join(r.nodeDirPath, nodeFileName), nodeBytes, 0640)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to write node file: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *Refresher) createNodePath() error {
|
||||||
|
_, err := os.Stat(r.nodeDirPath)
|
||||||
|
if os.IsNotExist(err) {
|
||||||
|
err = os.MkdirAll(r.nodeDirPath, 0755)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to create node dir: %w", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
|
||||||
|
}
|
||||||
8
refresher/types.go
Normal file
8
refresher/types.go
Normal file
|
|
@ -0,0 +1,8 @@
|
||||||
|
package refresher
|
||||||
|
|
||||||
|
type NodeConfig struct {
|
||||||
|
MasterHost string `json:"master_host"`
|
||||||
|
MasterProto string `json:"master_proto"`
|
||||||
|
MasterToken string `json:"master_token"`
|
||||||
|
MasterPort int `json:"master_port"`
|
||||||
|
}
|
||||||
Loading…
Reference in a new issue