From bc9e1250f33b06422ebb0e3ed4605fa9aded0c86 Mon Sep 17 00:00:00 2001
From: zachmann <gabriel.zachmann@kit.edu>
Date: Mon, 16 Nov 2020 09:51:35 +0100
Subject: [PATCH] add db garbage collector
---
cmd/dbGarbageCollector/main.go | 29 +++++++++++++++++++++++++++++
cmd/mytoken/main.go | 7 +++----
2 files changed, 32 insertions(+), 4 deletions(-)
create mode 100644 cmd/dbGarbageCollector/main.go
diff --git a/cmd/dbGarbageCollector/main.go b/cmd/dbGarbageCollector/main.go
new file mode 100644
index 00000000..5c5f9774
--- /dev/null
+++ b/cmd/dbGarbageCollector/main.go
@@ -0,0 +1,29 @@
+package main
+
+import (
+ "log"
+
+ "github.com/zachmann/mytoken/internal/config"
+ "github.com/zachmann/mytoken/internal/db"
+)
+
+func main() {
+ config.Load()
+ if err := db.Connect(); err != nil {
+ log.Fatal(err)
+ }
+ deleteExpiredPollingCodes()
+ deleteExpiredAuthInfo()
+}
+
+func deleteExpiredPollingCodes() {
+ if _, err := db.DB().Exec(`DELETE FROM PollingCodes WHERE expires_at < CURRENT_TIMESTAMP()`); err != nil {
+ log.Printf("%s", err)
+ }
+}
+
+func deleteExpiredAuthInfo() {
+ if _, err := db.DB().Exec(`DELETE FROM AuthInfo WHERE expires_at < CURRENT_TIMESTAMP()`); err != nil {
+ log.Printf("%s", err)
+ }
+}
diff --git a/cmd/mytoken/main.go b/cmd/mytoken/main.go
index 687e662d..4e6009ef 100644
--- a/cmd/mytoken/main.go
+++ b/cmd/mytoken/main.go
@@ -3,7 +3,7 @@ package main
import (
"github.com/zachmann/mytoken/internal/config"
"github.com/zachmann/mytoken/internal/db"
- "github.com/zachmann/mytoken/internal/endpoints/configuration"
+ configurationEndpoint "github.com/zachmann/mytoken/internal/endpoints/configuration"
"github.com/zachmann/mytoken/internal/jws"
"github.com/zachmann/mytoken/internal/oidc/authcode"
"github.com/zachmann/mytoken/internal/server"
@@ -11,10 +11,9 @@ import (
func main() {
config.Load()
- configuration.Init()
+ configurationEndpoint.Init()
authcode.Init()
- err := db.Connect()
- if err != nil {
+ if err := db.Connect(); err != nil {
panic(err)
}
jws.LoadKey()
--
GitLab