3pp-mirror/pomerium
1
0
Fork 0
mirror of https://github.com/pomerium/pomerium.git synced 2025-05-18 03:27:16 +02:00
Code Issues Projects Releases Packages Wiki Activity

databroker server backend config (#1127)

Browse source 
* config,docs: add databroker storage backend configuration

* cache: allow configuring which backend storage to use

Currently supported types are "memory", "redis".
This commit is contained in:
Cuong Manh Le 2020-07-23 10:42:43 +07:00 • committed by GitHub
parent c9182f757e
commit 1640151bc1
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
9 changed files with 99 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

19
internal/databroker/server.go
View file

@ -21,6 +21,7 @@ import (
"github.com/pomerium/pomerium/pkg/grpc/databroker"
"github.com/pomerium/pomerium/pkg/storage"
"github.com/pomerium/pomerium/pkg/storage/inmemory"
"github.com/pomerium/pomerium/pkg/storage/redis"
)
const (
@ -278,10 +279,26 @@ func (srv *Server) getDB(recordType string) storage.Backend {
srv.mu.Lock()
db = srv.byType[recordType]
if db == nil {
db = inmemory.NewDB(recordType, srv.cfg.btreeDegree)
db = srv.newDB(recordType)
srv.byType[recordType] = db
}
srv.mu.Unlock()
}
return db
}
func (srv *Server) newDB(recordType string) storage.Backend {
switch srv.cfg.storageType {
case inmemory.Name:
return inmemory.NewDB(recordType, srv.cfg.btreeDegree)
case redis.Name:
db, err := redis.New(srv.cfg.storageConnectionString, recordType, int64(srv.cfg.deletePermanentlyAfter.Seconds()))
if err != nil {
srv.log.Error().Err(err).Msg("failed to create new redis storage")
return nil
}
return db
default:
return nil
}
}