aboutsummaryrefslogtreecommitdiff
path: root/cgi/servers.go
diff options
context:
space:
mode:
Diffstat (limited to 'cgi/servers.go')
-rw-r--r--cgi/servers.go39
1 files changed, 28 insertions, 11 deletions
diff --git a/cgi/servers.go b/cgi/servers.go
index 0f23aac..84d6ffe 100644
--- a/cgi/servers.go
+++ b/cgi/servers.go
@@ -3,7 +3,7 @@ package cgi
import (
"context"
"fmt"
- "io"
+ "io/ioutil"
"net"
"net/http"
"net/http/cgi"
@@ -12,12 +12,17 @@ import (
"time"
"git.capotej.com/capotej/communique/config"
+ "github.com/dgraph-io/badger/v3"
+ "go.uber.org/zap"
)
-type Servers struct{}
+type Servers struct {
+ log *zap.SugaredLogger
+ db *badger.DB
+}
-func NewServers() *Servers {
- return &Servers{}
+func NewServers(log *zap.SugaredLogger, db *badger.DB) *Servers {
+ return &Servers{log: log, db: db}
}
// Start iterates over all Handlers and starts an internal CGI server for each one
@@ -33,7 +38,7 @@ func (s *Servers) Start(cfg config.Config) {
}(handler)
go func(aHandler config.Handler) {
defer wg.Done()
- startTicker(aHandler)
+ startTicker(aHandler, s.db, s.log)
}(handler)
}
wg.Wait()
@@ -55,23 +60,32 @@ func startCGIServer(h config.Handler) {
server.Serve(unixListener)
}
-func startTicker(h config.Handler) {
+func startTicker(h config.Handler, db *badger.DB, log *zap.SugaredLogger) {
// TODO add config for this
- ticker := time.NewTicker(5 * time.Second)
+ ticker := time.NewTicker(h.Interval)
done := make(chan bool)
func() {
for {
select {
case <-done:
return
- case _ = <-ticker.C:
- tick(h)
+ case t := <-ticker.C:
+ output := tick(h)
+ keyName := fmt.Sprintf("outbox:%s:%d", h.Name, t.Unix())
+ err := db.Update(func(txn *badger.Txn) error {
+ e := badger.NewEntry([]byte(keyName), output)
+ log.With("name", "ticker").Infof("writing '%s' to %s", output, keyName)
+ return txn.SetEntry(e)
+ })
+ if err != nil {
+ log.Error(err)
+ }
}
}
}()
}
-func tick(h config.Handler) {
+func tick(h config.Handler) []byte {
sock := fmt.Sprintf("%s.sock", h.Name)
httpc := http.Client{
Transport: &http.Transport{
@@ -86,5 +100,8 @@ func tick(h config.Handler) {
if err != nil {
panic(err)
}
- io.Copy(os.Stdout, response.Body)
+
+ body, err := ioutil.ReadAll(response.Body)
+ response.Body.Close()
+ return body
}