feat: add support for entitlements events

We add a basic highwatermark cache to prevent recalculations multiple times. This will be more important when the ingest part is ready.
openmeterio · Jul 31, 2024 · 0c301b9 · 0c301b9
1 parent 7bad3e6
commit 0c301b9
Show file tree

Hide file tree

Showing 7 changed files with 246 additions and 228 deletions.
diff --git a/cmd/server/main.go b/cmd/server/main.go
@@ -12,7 +12,7 @@ import (
 	"syscall"
 	"time"
 
-	"entgo.io/ent/dialect/sql"
+	entDialectSQL "entgo.io/ent/dialect/sql"
 	health "github.com/AppsFlyer/go-sundheit"
 	healthhttp "github.com/AppsFlyer/go-sundheit/http"
 	"github.com/ClickHouse/clickhouse-go/v2"
@@ -298,6 +298,7 @@ func main() {
 	debugConnector := debug.NewDebugConnector(streamingConnector)
 	entitlementConnRegistry := &registry.Entitlement{}
 
+	// Initialize Postgres
 	if conf.Entitlements.Enabled {
 		pgClients, err := initPGClients(conf.Postgres)
 		if err != nil {
@@ -536,7 +537,7 @@ func initNamespace(config config.Configuration, namespaces ...namespace.Handler)
 }
 
 type pgClients struct {
-	driver *sql.Driver
+	driver *entDialectSQL.Driver
 	client *db.Client
 }
 

diff --git a/internal/cmd/README.md b/internal/cmd/README.md
diff --git a/internal/cmd/clickhouse/client.go b/internal/cmd/clickhouse/client.go
diff --git a/internal/cmd/postgres/client.go b/internal/cmd/postgres/client.go
diff --git a/internal/cmd/streaming/connector.go b/internal/cmd/streaming/connector.go
diff --git a/internal/common/entitlement/connectors.go b/internal/common/entitlement/connectors.go