Skip to content
Snippets Groups Projects
Commit d86d8dd6 authored by Travis Ralston's avatar Travis Ralston
Browse files

Fix import_synapse script, probably

parent a3a7ac5b
No related branches found
No related tags found
No related merge requests found
package main package main
import ( import (
"context"
"errors" "errors"
"flag" "flag"
"fmt" "fmt"
...@@ -17,6 +16,7 @@ import ( ...@@ -17,6 +16,7 @@ import (
"github.com/turt2live/matrix-media-repo/common" "github.com/turt2live/matrix-media-repo/common"
"github.com/turt2live/matrix-media-repo/common/config" "github.com/turt2live/matrix-media-repo/common/config"
"github.com/turt2live/matrix-media-repo/common/logging" "github.com/turt2live/matrix-media-repo/common/logging"
"github.com/turt2live/matrix-media-repo/common/rcontext"
"github.com/turt2live/matrix-media-repo/controllers/upload_controller" "github.com/turt2live/matrix-media-repo/controllers/upload_controller"
"github.com/turt2live/matrix-media-repo/storage" "github.com/turt2live/matrix-media-repo/storage"
"github.com/turt2live/matrix-media-repo/storage/datastore" "github.com/turt2live/matrix-media-repo/storage/datastore"
...@@ -64,7 +64,7 @@ func main() { ...@@ -64,7 +64,7 @@ func main() {
} }
logrus.Info("Preparing database...") logrus.Info("Preparing database...")
mediaStore := storage.GetDatabase().GetMediaStore(context.TODO(), &logrus.Entry{}) mediaStore := storage.GetDatabase().GetMediaStore(rcontext.Initial())
logrus.Info("Initializing datastores...") logrus.Info("Initializing datastores...")
enabledDatastores := 0 enabledDatastores := 0
...@@ -76,7 +76,7 @@ func main() { ...@@ -76,7 +76,7 @@ func main() {
enabledDatastores++ enabledDatastores++
uri := datastore.GetUriForDatastore(ds) uri := datastore.GetUriForDatastore(ds)
_, err := storage.GetOrCreateDatastoreOfType(context.TODO(), &logrus.Entry{}, ds.Type, uri) _, err := storage.GetOrCreateDatastoreOfType(rcontext.Initial(), ds.Type, uri)
if err != nil { if err != nil {
logrus.Fatal(err) logrus.Fatal(err)
} }
...@@ -174,10 +174,9 @@ func main() { ...@@ -174,10 +174,9 @@ func main() {
func fetchMedia(req interface{}) interface{} { func fetchMedia(req interface{}) interface{} {
payload := req.(*fetchRequest) payload := req.(*fetchRequest)
record := payload.media record := payload.media
ctx := context.TODO() ctx := rcontext.Initial()
log := logrus.WithFields(logrus.Fields{})
db := storage.GetDatabase().GetMediaStore(ctx, log) db := storage.GetDatabase().GetMediaStore(ctx)
_, err := db.Get(payload.serverName, record.MediaId) _, err := db.Get(payload.serverName, record.MediaId)
if err == nil { if err == nil {
...@@ -191,7 +190,7 @@ func fetchMedia(req interface{}) interface{} { ...@@ -191,7 +190,7 @@ func fetchMedia(req interface{}) interface{} {
return nil return nil
} }
_, err = upload_controller.StoreDirect(body, -1, record.ContentType, record.UploadName, record.UserId, payload.serverName, record.MediaId, common.KindLocalMedia, ctx, log) _, err = upload_controller.StoreDirect(body, -1, record.ContentType, record.UploadName, record.UserId, payload.serverName, record.MediaId, common.KindLocalMedia, ctx)
if err != nil { if err != nil {
logrus.Error(err.Error()) logrus.Error(err.Error())
return nil return nil
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment