LBRYCommunity/reflector.go
1
0
Fork 0
Code Issues 7 Pull requests Projects Releases 10 Packages Wiki Activity

make it easy to turn the db on and off

Browse source
This commit is contained in:
Alex Grintsvayg 2019-07-02 09:32:53 -04:00
parent c7643fb5da
commit c088a65041
No known key found for this signature in database
GPG key ID: AEB3F089F86A22B5
1 changed files with 27 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

41
cmd/reflector.go
View file

@ -7,10 +7,11 @@ import (
"syscall" "syscall"
"time" "time"
"github.com/lbryio/reflector.go/reflector"
"github.com/lbryio/reflector.go/db" "github.com/lbryio/reflector.go/db"
"github.com/lbryio/reflector.go/meta" "github.com/lbryio/reflector.go/meta"
"github.com/lbryio/reflector.go/peer" "github.com/lbryio/reflector.go/peer"
"github.com/lbryio/reflector.go/reflector"
"github.com/lbryio/reflector.go/store" "github.com/lbryio/reflector.go/store"
log "github.com/sirupsen/logrus" log "github.com/sirupsen/logrus"
@ -35,22 +36,32 @@ func reflectorCmd(cmd *cobra.Command, args []string) {
} }
s3 := store.NewS3BlobStore(globalConfig.AwsID, globalConfig.AwsSecret, globalConfig.BucketRegion, globalConfig.BucketName) s3 := store.NewS3BlobStore(globalConfig.AwsID, globalConfig.AwsSecret, globalConfig.BucketRegion, globalConfig.BucketName)
combo := store.NewDBBackedS3Store(s3, db)
reflectorServer := reflector.NewServer(combo) // flip this flag to false when doing db maintenance. uploads will not work (as reflector server wont be running)
reflectorServer.Timeout = 3 * time.Minute // but downloads will still work straight from s3
if globalConfig.SlackHookURL != "" { useDB := true
reflectorServer.StatLogger = log.StandardLogger()
reflectorServer.StatReportFrequency = 1 * time.Hour
}
reflectorServer.EnableBlocklist = true
err = reflectorServer.Start(":" + strconv.Itoa(reflector.DefaultPort)) var reflectorServer *reflector.Server
if err != nil { var blobStore store.BlobStore = s3
log.Fatal(err)
if useDB {
blobStore = store.NewDBBackedS3Store(s3, db)
reflectorServer = reflector.NewServer(blobStore)
reflectorServer.Timeout = 3 * time.Minute
if globalConfig.SlackHookURL != "" {
reflectorServer.StatLogger = log.StandardLogger()
reflectorServer.StatReportFrequency = 1 * time.Hour
}
reflectorServer.EnableBlocklist = true
err = reflectorServer.Start(":" + strconv.Itoa(reflector.DefaultPort))
if err != nil {
log.Fatal(err)
}
} }
peerServer := peer.NewServer(combo) peerServer := peer.NewServer(blobStore)
if globalConfig.SlackHookURL != "" { if globalConfig.SlackHookURL != "" {
peerServer.StatLogger = log.StandardLogger() peerServer.StatLogger = log.StandardLogger()
peerServer.StatReportFrequency = 1 * time.Hour peerServer.StatReportFrequency = 1 * time.Hour
@ -64,5 +75,7 @@ func reflectorCmd(cmd *cobra.Command, args []string) {
signal.Notify(interruptChan, os.Interrupt, syscall.SIGTERM) signal.Notify(interruptChan, os.Interrupt, syscall.SIGTERM)
<-interruptChan <-interruptChan
peerServer.Shutdown() peerServer.Shutdown()
reflectorServer.Shutdown() if reflectorServer != nil {
reflectorServer.Shutdown()
}
} }