forked from hswaw/hscloud
personal/q3k: door^Wship stuck
Change-Id: I189fc13971d46790634804c3fa1b54e2c4788273master
parent
7967ca177b
commit
3d116b2952
|
|
@ -0,0 +1,49 @@
|
|||
local mirko = import "../../kube/mirko.libsonnet";
|
||||
|
||||
{
|
||||
local top = self,
|
||||
shipstuck:: {
|
||||
cfg:: {
|
||||
image: "registry.k0.hswaw.net/q3k/shipstuck:315532800-a7282b5aa2952e5eb66a1c3ecf7cdafef8335aba",
|
||||
domain: error "domain must be set",
|
||||
},
|
||||
component(cfg, env): mirko.Component(env, "shipstuck") {
|
||||
local shipstuck = self,
|
||||
cfg+: {
|
||||
image: cfg.image,
|
||||
container: shipstuck.GoContainer("main", "/personal/q3k/shipstuck") {
|
||||
command+: [
|
||||
"-public_address", "0.0.0.0:8080",
|
||||
],
|
||||
},
|
||||
ports+: {
|
||||
publicHTTP: {
|
||||
public: {
|
||||
port: 8080,
|
||||
dns: cfg.domain,
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
|
||||
env(name):: mirko.Environment(name) {
|
||||
local env = self,
|
||||
local cfg = self.cfg,
|
||||
cfg+: {
|
||||
shipstuck: top.shipstuck.cfg,
|
||||
},
|
||||
components: {
|
||||
shipstuck: top.shipstuck.component(cfg.shipstuck, env),
|
||||
},
|
||||
},
|
||||
|
||||
prod: top.env("personal-q3k") {
|
||||
cfg+: {
|
||||
shipstuck+: {
|
||||
domain: "shipstuck.q3k.org",
|
||||
},
|
||||
},
|
||||
},
|
||||
}
|
||||
|
|
@ -0,0 +1,47 @@
|
|||
load("@io_bazel_rules_docker//container:container.bzl", "container_image", "container_layer", "container_push")
|
||||
load("@io_bazel_rules_go//go:def.bzl", "go_binary", "go_library")
|
||||
|
||||
go_library(
|
||||
name = "go_default_library",
|
||||
srcs = ["main.go"],
|
||||
importpath = "code.hackerspace.pl/hscloud/personal/q3k/shipstuck",
|
||||
visibility = ["//visibility:private"],
|
||||
deps = [
|
||||
"//go/mirko:go_default_library",
|
||||
"//personal/q3k/shipstuck/proto:go_default_library",
|
||||
"@com_github_golang_glog//:go_default_library",
|
||||
"@com_github_grpc_ecosystem_grpc_gateway//runtime:go_default_library",
|
||||
],
|
||||
)
|
||||
|
||||
go_binary(
|
||||
name = "shipstuck",
|
||||
embed = [":go_default_library"],
|
||||
visibility = ["//visibility:public"],
|
||||
)
|
||||
|
||||
container_layer(
|
||||
name = "layer_bin",
|
||||
files = [
|
||||
":shipstuck",
|
||||
],
|
||||
directory = "/personal/q3k/",
|
||||
)
|
||||
|
||||
container_image(
|
||||
name = "runtime",
|
||||
base = "@prodimage-bionic//image",
|
||||
layers = [
|
||||
":layer_bin",
|
||||
],
|
||||
)
|
||||
|
||||
container_push(
|
||||
name = "push",
|
||||
image = ":runtime",
|
||||
format = "Docker",
|
||||
registry = "registry.k0.hswaw.net",
|
||||
repository = "q3k/shipstuck",
|
||||
tag = "{BUILD_TIMESTAMP}-{STABLE_GIT_COMMIT}",
|
||||
)
|
||||
|
||||
|
|
@ -0,0 +1,191 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"flag"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"code.hackerspace.pl/hscloud/go/mirko"
|
||||
"github.com/golang/glog"
|
||||
"github.com/grpc-ecosystem/grpc-gateway/runtime"
|
||||
|
||||
pb "code.hackerspace.pl/hscloud/personal/q3k/shipstuck/proto"
|
||||
)
|
||||
|
||||
type vessel struct {
|
||||
// No idea what these fields are, but they seem to be related to
|
||||
// latitude/longitude. Use these to detect the stuckness of the ship.
|
||||
GT int64 `json:"gt"`
|
||||
DW int64 `json:"dw"`
|
||||
}
|
||||
|
||||
// get retrieves the current status of the ship - returns true if stack, false
|
||||
// otherwise.
|
||||
func get(ctx context.Context) (bool, error) {
|
||||
// Sorry vesselfinder, if you made it easy to set up an account I would
|
||||
// gladly pay for the API instead of doing this. Limiting requests to once
|
||||
// every 15 minutes though, that seems fair enough.
|
||||
req, err := http.NewRequestWithContext(ctx, "GET", "https://www.vesselfinder.com/api/pub/click/353136000", nil)
|
||||
if err != nil {
|
||||
return false, fmt.Errorf("NewRequestWithContext: %w", err)
|
||||
}
|
||||
req.Header.Set("User-Agent", "Mozilla/5.0 (X11; Linux x86_64; rv:86.0) Gecko/20100101 Firefox/86.0")
|
||||
|
||||
res, err := http.DefaultClient.Do(req)
|
||||
if err != nil {
|
||||
return false, fmt.Errorf("Do: %w", err)
|
||||
}
|
||||
|
||||
defer res.Body.Close()
|
||||
|
||||
v := &vessel{}
|
||||
err = json.NewDecoder(res.Body).Decode(v)
|
||||
if err != nil {
|
||||
return false, fmt.Errorf("Decode: %w", err)
|
||||
}
|
||||
|
||||
if v.GT == 219079 && v.DW == 199489 {
|
||||
return true, nil
|
||||
} else {
|
||||
glog.Infof("Freed! %+v", v)
|
||||
return false, nil
|
||||
}
|
||||
}
|
||||
|
||||
type shipState string
|
||||
|
||||
const (
|
||||
shipStateUnknown shipState = "UNKNOWN"
|
||||
shipStateStuck shipState = "STUCK"
|
||||
shipStateFreed shipState = "FREED"
|
||||
)
|
||||
|
||||
type service struct {
|
||||
lastStateMu sync.RWMutex
|
||||
lastState shipState
|
||||
lastStateTime time.Time
|
||||
}
|
||||
|
||||
func (s *service) worker(ctx context.Context) {
|
||||
update := func() {
|
||||
state := shipStateUnknown
|
||||
// shitty back off, good enough.
|
||||
retries := 10
|
||||
for {
|
||||
stuck, err := get(ctx)
|
||||
if err != nil {
|
||||
glog.Warningf("get: %v", err)
|
||||
if retries > 0 {
|
||||
time.Sleep(60 * time.Second)
|
||||
retries -= 1
|
||||
} else {
|
||||
glog.Errorf("giving up on get")
|
||||
break
|
||||
}
|
||||
} else {
|
||||
if stuck {
|
||||
state = shipStateStuck
|
||||
} else {
|
||||
state = shipStateFreed
|
||||
}
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
glog.Infof("New state: %v", state)
|
||||
s.lastStateMu.Lock()
|
||||
s.lastState = state
|
||||
s.lastStateTime = time.Now()
|
||||
s.lastStateMu.Unlock()
|
||||
}
|
||||
|
||||
update()
|
||||
ticker := time.NewTicker(15 * 60 * time.Second)
|
||||
for {
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
return
|
||||
case <-ticker.C:
|
||||
update()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func timeMust(t time.Time, err error) time.Time {
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
return t
|
||||
}
|
||||
|
||||
var (
|
||||
timeStuck = timeMust(time.Parse(
|
||||
"At 15:04 Eastern European Time (MST) on 2 January 2006",
|
||||
"At 07:40 Eastern European Time (UTC) on 23 March 2021",
|
||||
))
|
||||
)
|
||||
|
||||
func (s *service) Status(ctx context.Context, req *pb.StatusRequest) (*pb.StatusResponse, error) {
|
||||
s.lastStateMu.RLock()
|
||||
state := s.lastState
|
||||
lastChecked := s.lastStateTime
|
||||
s.lastStateMu.RUnlock()
|
||||
|
||||
res := &pb.StatusResponse{
|
||||
LastChecked: lastChecked.UnixNano(),
|
||||
}
|
||||
switch state {
|
||||
case shipStateUnknown:
|
||||
res.Current = pb.StatusResponse_STUCKNESS_UNKNOWN
|
||||
case shipStateStuck:
|
||||
res.Current = pb.StatusResponse_STUCKNESS_STUCK
|
||||
res.Elapsed = time.Since(timeStuck).Nanoseconds()
|
||||
case shipStateFreed:
|
||||
res.Current = pb.StatusResponse_STUCKNESS_FREE
|
||||
}
|
||||
|
||||
return res, nil
|
||||
}
|
||||
|
||||
var (
|
||||
flagPublicAddress string
|
||||
)
|
||||
|
||||
func main() {
|
||||
flag.StringVar(&flagPublicAddress, "public_address", "127.0.0.1:8080", "Public HTTP/JSON listen address")
|
||||
flag.Parse()
|
||||
m := mirko.New()
|
||||
if err := m.Listen(); err != nil {
|
||||
glog.Exitf("Listen(): %v", err)
|
||||
}
|
||||
|
||||
s := &service{}
|
||||
pb.RegisterShipStuckServer(m.GRPC(), s)
|
||||
|
||||
publicMux := runtime.NewServeMux()
|
||||
publicSrv := http.Server{
|
||||
Addr: flagPublicAddress,
|
||||
Handler: publicMux,
|
||||
}
|
||||
go func() {
|
||||
glog.Infof("REST listening on %s", flagPublicAddress)
|
||||
if err := publicSrv.ListenAndServe(); err != nil {
|
||||
glog.Exitf("public ListenAndServe: %v", err)
|
||||
}
|
||||
}()
|
||||
if err := pb.RegisterShipStuckHandlerServer(m.Context(), publicMux, s); err != nil {
|
||||
glog.Exitf("RegisterShipStuckHandlerSerever: %v", err)
|
||||
}
|
||||
|
||||
go s.worker(m.Context())
|
||||
|
||||
if err := m.Serve(); err != nil {
|
||||
glog.Exitf("Serve(): %v", err)
|
||||
}
|
||||
|
||||
<-m.Done()
|
||||
}
|
||||
|
|
@ -0,0 +1,31 @@
|
|||
load("@rules_proto//proto:defs.bzl", "proto_library")
|
||||
load("@io_bazel_rules_go//go:def.bzl", "go_library")
|
||||
load("@io_bazel_rules_go//proto:def.bzl", "go_proto_library")
|
||||
|
||||
proto_library(
|
||||
name = "proto_proto",
|
||||
srcs = ["shipstuck.proto"],
|
||||
visibility = ["//visibility:public"],
|
||||
deps = ["@go_googleapis//google/api:annotations_proto"],
|
||||
)
|
||||
|
||||
go_proto_library(
|
||||
name = "proto_go_proto",
|
||||
compilers = [
|
||||
"@com_github_grpc_ecosystem_grpc_gateway//protoc-gen-grpc-gateway:go_gen_grpc_gateway", # keep
|
||||
"@io_bazel_rules_go//proto:go_grpc",
|
||||
],
|
||||
importpath = "code.hackerspace.pl/hscloud/personal/q3k/shipstuck/proto",
|
||||
proto = ":proto_proto",
|
||||
visibility = ["//visibility:public"],
|
||||
deps = [
|
||||
"@go_googleapis//google/api:annotations_go_proto",
|
||||
],
|
||||
)
|
||||
|
||||
go_library(
|
||||
name = "go_default_library",
|
||||
embed = [":proto_go_proto"],
|
||||
importpath = "code.hackerspace.pl/hscloud/personal/q3k/shipstuck/proto",
|
||||
visibility = ["//visibility:public"],
|
||||
)
|
||||
|
|
@ -0,0 +1,30 @@
|
|||
syntax = "proto3";
|
||||
package proto;
|
||||
option go_package = "code.hackerspace.pl/hscloud/personal/q3k/shipstuck/proto";
|
||||
|
||||
import "google/api/annotations.proto";
|
||||
|
||||
service ShipStuck {
|
||||
rpc Status(StatusRequest) returns (StatusResponse) {
|
||||
option (google.api.http) = {
|
||||
get: "/v1/shipstuck/status"
|
||||
};
|
||||
};
|
||||
}
|
||||
|
||||
message StatusRequest {
|
||||
}
|
||||
|
||||
message StatusResponse {
|
||||
// Timestamp (nanos from epoch) of last check.
|
||||
int64 last_checked = 1;
|
||||
enum Stuckness {
|
||||
STUCKNESS_INVALID = 0;
|
||||
STUCKNESS_STUCK = 1;
|
||||
STUCKNESS_FREE = 2;
|
||||
STUCKNESS_UNKNOWN = 3;
|
||||
};
|
||||
Stuckness current = 2;
|
||||
// If STUCK, how many nanoseconds have elapsed since the whoopsie?
|
||||
int64 elapsed = 3;
|
||||
}
|
||||
Loading…
Reference in New Issue