summaryrefslogtreecommitdiff
path: root/internal/trans
diff options
context:
space:
mode:
authorLibravatar R. Aidan Campbell <raidancampbell@users.noreply.github.com>2021-10-11 05:37:33 -0700
committerLibravatar GitHub <noreply@github.com>2021-10-11 14:37:33 +0200
commit083099a9575f8b2fac22c1d4a51a9dd0e2201243 (patch)
treed1787aa544679c433f797d2313ce532250fe574f /internal/trans
parentHandle forwarded messages (#273) (diff)
downloadgotosocial-083099a9575f8b2fac22c1d4a51a9dd0e2201243.tar.xz
reference global logrus (#274)
* reference logrus' global logger instead of passing and storing a logger reference everywhere * always directly use global logrus logger instead of referencing an instance * test suites should also directly use the global logrus logger * rename gin logging function to clarify that it's middleware * correct comments which erroneously referenced removed logger parameter * setting log level for tests now uses logrus' exported type instead of the string value, to guarantee error isn't possible
Diffstat (limited to 'internal/trans')
-rw-r--r--internal/trans/exporter.go7
-rw-r--r--internal/trans/exportminimal_test.go2
-rw-r--r--internal/trans/import.go19
-rw-r--r--internal/trans/import_test.go4
-rw-r--r--internal/trans/importer.go7
-rw-r--r--internal/trans/trans_test.go4
6 files changed, 18 insertions, 25 deletions
diff --git a/internal/trans/exporter.go b/internal/trans/exporter.go
index 3dc0558f5..d6d5678d0 100644
--- a/internal/trans/exporter.go
+++ b/internal/trans/exporter.go
@@ -21,7 +21,6 @@ package trans
import (
"context"
- "github.com/sirupsen/logrus"
"github.com/superseriousbusiness/gotosocial/internal/db"
)
@@ -32,15 +31,13 @@ type Exporter interface {
type exporter struct {
db db.DB
- log *logrus.Logger
writtenIDs map[string]bool
}
-// NewExporter returns a new Exporter that will use the given db and logger.
-func NewExporter(db db.DB, log *logrus.Logger) Exporter {
+// NewExporter returns a new Exporter that will use the given db.
+func NewExporter(db db.DB) Exporter {
return &exporter{
db: db,
- log: log,
writtenIDs: make(map[string]bool),
}
}
diff --git a/internal/trans/exportminimal_test.go b/internal/trans/exportminimal_test.go
index 2bffffcfe..e01793862 100644
--- a/internal/trans/exportminimal_test.go
+++ b/internal/trans/exportminimal_test.go
@@ -38,7 +38,7 @@ func (suite *ExportMinimalTestSuite) TestExportMinimalOK() {
tempFilePath := fmt.Sprintf("%s/%s", suite.T().TempDir(), uuid.NewString())
// export to the tempFilePath
- exporter := trans.NewExporter(suite.db, suite.log)
+ exporter := trans.NewExporter(suite.db)
err := exporter.ExportMinimal(context.Background(), tempFilePath)
suite.NoError(err)
diff --git a/internal/trans/import.go b/internal/trans/import.go
index 74624e540..f93b50f9e 100644
--- a/internal/trans/import.go
+++ b/internal/trans/import.go
@@ -23,6 +23,7 @@ import (
"encoding/json"
"errors"
"fmt"
+ "github.com/sirupsen/logrus"
"io"
"os"
@@ -47,7 +48,7 @@ func (i *importer) Import(ctx context.Context, path string) error {
err := decoder.Decode(&entry)
if err != nil {
if err == io.EOF {
- i.log.Infof("Import: reached end of file")
+ logrus.Infof("Import: reached end of file")
return neatClose(file)
}
return fmt.Errorf("Import: error decoding in readLoop: %s", err)
@@ -73,7 +74,7 @@ func (i *importer) inputEntry(ctx context.Context, entry transmodel.Entry) error
if err := i.putInDB(ctx, account); err != nil {
return fmt.Errorf("inputEntry: error adding account to database: %s", err)
}
- i.log.Infof("inputEntry: added account with id %s", account.ID)
+ logrus.Infof("inputEntry: added account with id %s", account.ID)
return nil
case transmodel.TransBlock:
block, err := i.blockDecode(entry)
@@ -83,7 +84,7 @@ func (i *importer) inputEntry(ctx context.Context, entry transmodel.Entry) error
if err := i.putInDB(ctx, block); err != nil {
return fmt.Errorf("inputEntry: error adding block to database: %s", err)
}
- i.log.Infof("inputEntry: added block with id %s", block.ID)
+ logrus.Infof("inputEntry: added block with id %s", block.ID)
return nil
case transmodel.TransDomainBlock:
block, err := i.domainBlockDecode(entry)
@@ -93,7 +94,7 @@ func (i *importer) inputEntry(ctx context.Context, entry transmodel.Entry) error
if err := i.putInDB(ctx, block); err != nil {
return fmt.Errorf("inputEntry: error adding domain block to database: %s", err)
}
- i.log.Infof("inputEntry: added domain block with id %s", block.ID)
+ logrus.Infof("inputEntry: added domain block with id %s", block.ID)
return nil
case transmodel.TransFollow:
follow, err := i.followDecode(entry)
@@ -103,7 +104,7 @@ func (i *importer) inputEntry(ctx context.Context, entry transmodel.Entry) error
if err := i.putInDB(ctx, follow); err != nil {
return fmt.Errorf("inputEntry: error adding follow to database: %s", err)
}
- i.log.Infof("inputEntry: added follow with id %s", follow.ID)
+ logrus.Infof("inputEntry: added follow with id %s", follow.ID)
return nil
case transmodel.TransFollowRequest:
fr, err := i.followRequestDecode(entry)
@@ -113,7 +114,7 @@ func (i *importer) inputEntry(ctx context.Context, entry transmodel.Entry) error
if err := i.putInDB(ctx, fr); err != nil {
return fmt.Errorf("inputEntry: error adding follow request to database: %s", err)
}
- i.log.Infof("inputEntry: added follow request with id %s", fr.ID)
+ logrus.Infof("inputEntry: added follow request with id %s", fr.ID)
return nil
case transmodel.TransInstance:
inst, err := i.instanceDecode(entry)
@@ -123,7 +124,7 @@ func (i *importer) inputEntry(ctx context.Context, entry transmodel.Entry) error
if err := i.putInDB(ctx, inst); err != nil {
return fmt.Errorf("inputEntry: error adding instance to database: %s", err)
}
- i.log.Infof("inputEntry: added instance with id %s", inst.ID)
+ logrus.Infof("inputEntry: added instance with id %s", inst.ID)
return nil
case transmodel.TransUser:
user, err := i.userDecode(entry)
@@ -133,11 +134,11 @@ func (i *importer) inputEntry(ctx context.Context, entry transmodel.Entry) error
if err := i.putInDB(ctx, user); err != nil {
return fmt.Errorf("inputEntry: error adding user to database: %s", err)
}
- i.log.Infof("inputEntry: added user with id %s", user.ID)
+ logrus.Infof("inputEntry: added user with id %s", user.ID)
return nil
}
- i.log.Errorf("inputEntry: didn't recognize transtype '%s', skipping it", t)
+ logrus.Errorf("inputEntry: didn't recognize transtype '%s', skipping it", t)
return nil
}
diff --git a/internal/trans/import_test.go b/internal/trans/import_test.go
index 137a5fae1..c8ceb614c 100644
--- a/internal/trans/import_test.go
+++ b/internal/trans/import_test.go
@@ -42,7 +42,7 @@ func (suite *ImportMinimalTestSuite) TestImportMinimalOK() {
tempFilePath := fmt.Sprintf("%s/%s", suite.T().TempDir(), uuid.NewString())
// export to the tempFilePath
- exporter := trans.NewExporter(suite.db, suite.log)
+ exporter := trans.NewExporter(suite.db)
err := exporter.ExportMinimal(ctx, tempFilePath)
suite.NoError(err)
@@ -57,7 +57,7 @@ func (suite *ImportMinimalTestSuite) TestImportMinimalOK() {
newDB := testrig.NewTestDB()
testrig.CreateTestTables(newDB)
- importer := trans.NewImporter(newDB, suite.log)
+ importer := trans.NewImporter(newDB)
err = importer.Import(ctx, tempFilePath)
suite.NoError(err)
diff --git a/internal/trans/importer.go b/internal/trans/importer.go
index ea8866f53..81029a098 100644
--- a/internal/trans/importer.go
+++ b/internal/trans/importer.go
@@ -21,7 +21,6 @@ package trans
import (
"context"
- "github.com/sirupsen/logrus"
"github.com/superseriousbusiness/gotosocial/internal/db"
)
@@ -32,13 +31,11 @@ type Importer interface {
type importer struct {
db db.DB
- log *logrus.Logger
}
-// NewImporter returns a new Importer interface that uses the given db and logger.
-func NewImporter(db db.DB, log *logrus.Logger) Importer {
+// NewImporter returns a new Importer interface that uses the given db.
+func NewImporter(db db.DB) Importer {
return &importer{
db: db,
- log: log,
}
}
diff --git a/internal/trans/trans_test.go b/internal/trans/trans_test.go
index 4a231486b..3e3dfe4a3 100644
--- a/internal/trans/trans_test.go
+++ b/internal/trans/trans_test.go
@@ -19,7 +19,6 @@
package trans_test
import (
- "github.com/sirupsen/logrus"
"github.com/stretchr/testify/suite"
"github.com/superseriousbusiness/gotosocial/internal/db"
"github.com/superseriousbusiness/gotosocial/testrig"
@@ -28,12 +27,11 @@ import (
type TransTestSuite struct {
suite.Suite
db db.DB
- log *logrus.Logger
}
func (suite *TransTestSuite) SetupTest() {
suite.db = testrig.NewTestDB()
- suite.log = testrig.NewTestLog()
+ testrig.InitTestLog()
testrig.StandardDBSetup(suite.db, nil)
}