Serge Bazanski
8f7ec7e141
This improves the structure of the code, separating the data/control interface out and then implementing the http interface as a user of this interface.
83 lines
2.2 KiB
Go
83 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
|
|
"k8s.io/klog"
|
|
)
|
|
|
|
var (
|
|
flagFake bool
|
|
flagListenHTTP string
|
|
flagPressureThresholdRough = ScientificNotationValue(1e-1)
|
|
flagPressureThresholdHigh = ScientificNotationValue(1e-4)
|
|
)
|
|
|
|
func main() {
|
|
flag.BoolVar(&flagFake, "fake", false, "Enable fake mode which allows to run succd for tests outside the succbone")
|
|
flag.StringVar(&flagListenHTTP, "listen_http", ":8080", "Address at which to listen for HTTP requests")
|
|
flag.TextVar(&flagPressureThresholdRough, "pressure_threshold_rough", &flagPressureThresholdRough, "Threshold for opening up diffusion pump (mbar)")
|
|
flag.TextVar(&flagPressureThresholdHigh, "pressure_threshold_high", &flagPressureThresholdHigh, "Threshold for enabling high voltage circuits (mbar)")
|
|
flag.Parse()
|
|
|
|
ctx, _ := signal.NotifyContext(context.Background(), os.Interrupt)
|
|
|
|
d := daemon{}
|
|
d.daemonState.rpOn = true
|
|
|
|
d.aboveRough.threshold = float64(flagPressureThresholdRough)
|
|
d.aboveHigh.threshold = float64(flagPressureThresholdHigh)
|
|
if flagFake {
|
|
klog.Infof("Starting with fake peripherals")
|
|
d.adcPirani = &fakeADC{}
|
|
d.gpioRoughingPump = &fakeGPIO{desc: "rp"}
|
|
d.gpioDiffusionPump = &fakeGPIO{desc: "~dp"}
|
|
d.gpioBtnPumpDown = &fakeGPIO{desc: "~pd"}
|
|
d.gpioBtnVent = &fakeGPIO{desc: "~vent"}
|
|
d.gpioBelowRough = &fakeGPIO{desc: "~rough"}
|
|
d.gpioBelowHigh = &fakeGPIO{desc: "~high"}
|
|
} else {
|
|
adc, err := newBBADC(0)
|
|
if err != nil {
|
|
klog.Exitf("Failed to setup Pirani ADC: %v", err)
|
|
}
|
|
d.adcPirani = adc
|
|
|
|
for _, c := range []struct {
|
|
out *gpio
|
|
num int
|
|
}{
|
|
{&d.gpioRoughingPump, 115},
|
|
{&d.gpioDiffusionPump, 49},
|
|
{&d.gpioBtnPumpDown, 48},
|
|
{&d.gpioBtnVent, 60},
|
|
{&d.gpioBelowRough, 30},
|
|
{&d.gpioBelowHigh, 7},
|
|
} {
|
|
*c.out, err = newBBGPIO(c.num, true)
|
|
if err != nil {
|
|
klog.Exitf("Failed to setup GPIO: %v", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
httpServer := httpServer{
|
|
d: &d,
|
|
}
|
|
httpServer.setupViews()
|
|
|
|
klog.Infof("Listening for HTTP at %s", flagListenHTTP)
|
|
go func() {
|
|
if err := http.ListenAndServe(flagListenHTTP, nil); err != nil {
|
|
klog.Errorf("HTTP listen failed: %v", err)
|
|
}
|
|
}()
|
|
|
|
go d.process(ctx)
|
|
<-ctx.Done()
|
|
}
|