Add InitEventStore function

extract the method from the management cmd package
This commit is contained in:
Maycon Santos
2023-12-30 20:19:00 +01:00
parent c462587ae6
commit 92debdf072
+16
View File
@@ -3,7 +3,10 @@ package integrations
import (
"github.com/gorilla/mux"
"github.com/netbirdio/netbird/management/server"
"github.com/netbirdio/netbird/management/server/activity"
"github.com/netbirdio/netbird/management/server/activity/sqlite"
"github.com/netbirdio/netbird/management/server/jwtclaims"
log "github.com/sirupsen/logrus"
)
func RegisterHandlers(
@@ -13,3 +16,16 @@ func RegisterHandlers(
) *mux.Router {
return router
}
func InitEventStore(dataDir string, key string) (activity.Store, string, error) {
var err error
if key == "" {
log.Debugf("generate new activity store encryption key")
key, err = sqlite.GenerateKey()
if err != nil {
return nil, "", err
}
}
store, err := sqlite.NewSQLiteStore(dataDir, key)
return store, key, err
}