mirror of
https://github.com/tim-beatham/smegmesh.git
synced 2024-12-14 02:21:30 +01:00
Build error forgot to add query.go
This commit is contained in:
parent
5183edc592
commit
8d8a13d6ff
84
pkg/query/query.go
Normal file
84
pkg/query/query.go
Normal file
@ -0,0 +1,84 @@
|
|||||||
|
package query
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"github.com/jmespath/go-jmespath"
|
||||||
|
"github.com/tim-beatham/wgmesh/pkg/lib"
|
||||||
|
"github.com/tim-beatham/wgmesh/pkg/mesh"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Querier queries a data store for the given data
|
||||||
|
// and returns data in the corresponding encoding
|
||||||
|
type Querier interface {
|
||||||
|
Query(meshId string, queryParams string) ([]byte, error)
|
||||||
|
}
|
||||||
|
|
||||||
|
type JmesQuerier struct {
|
||||||
|
manager *mesh.MeshManager
|
||||||
|
}
|
||||||
|
|
||||||
|
type QueryError struct {
|
||||||
|
msg string
|
||||||
|
}
|
||||||
|
|
||||||
|
type QueryNode struct {
|
||||||
|
HostEndpoint string `json:"hostEndpoint"`
|
||||||
|
PublicKey string `json:"publicKey"`
|
||||||
|
WgEndpoint string `json:"wgEndpoint"`
|
||||||
|
WgHost string `json:"wgHost"`
|
||||||
|
Timestamp int64 `json:"timestmap"`
|
||||||
|
Description string `json:"description"`
|
||||||
|
Routes []string `json:"routes"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *QueryError) Error() string {
|
||||||
|
return m.msg
|
||||||
|
}
|
||||||
|
|
||||||
|
// Query: queries the data
|
||||||
|
func (j *JmesQuerier) Query(meshId, queryParams string) ([]byte, error) {
|
||||||
|
mesh, ok := j.manager.Meshes[meshId]
|
||||||
|
|
||||||
|
if !ok {
|
||||||
|
return nil, &QueryError{msg: fmt.Sprintf("%s does not exist", meshId)}
|
||||||
|
}
|
||||||
|
|
||||||
|
snapshot, err := mesh.GetMesh()
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
nodes := lib.Map(lib.MapValues(snapshot.GetNodes()), meshNodeToQueryNode)
|
||||||
|
|
||||||
|
result, err := jmespath.Search(queryParams, nodes)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
bytes, err := json.Marshal(result)
|
||||||
|
return bytes, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func meshNodeToQueryNode(node mesh.MeshNode) *QueryNode {
|
||||||
|
queryNode := new(QueryNode)
|
||||||
|
queryNode.HostEndpoint = node.GetHostEndpoint()
|
||||||
|
pubKey, _ := node.GetPublicKey()
|
||||||
|
|
||||||
|
queryNode.PublicKey = pubKey.String()
|
||||||
|
|
||||||
|
queryNode.WgEndpoint = node.GetWgEndpoint()
|
||||||
|
queryNode.WgHost = node.GetWgHost().String()
|
||||||
|
|
||||||
|
queryNode.Timestamp = node.GetTimeStamp()
|
||||||
|
queryNode.Routes = node.GetRoutes()
|
||||||
|
queryNode.Description = node.GetDescription()
|
||||||
|
return queryNode
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewJmesQuerier(manager *mesh.MeshManager) Querier {
|
||||||
|
return &JmesQuerier{manager: manager}
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user