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

make cache dir a cli flag

Browse source
This commit is contained in:
Alex Grintsvayg 2020-02-27 14:52:48 -05:00
parent 4a5a148843
commit dde93a1fe6
No known key found for this signature in database
GPG key ID: AEB3F089F86A22B5
1 changed files with 12 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
cmd/reflector.go
View file

@ -18,12 +18,15 @@ import (
"github.com/spf13/cobra" "github.com/spf13/cobra"
) )
var reflectorCmdCacheDir string
func init() { func init() {
var cmd = &cobra.Command{ var cmd = &cobra.Command{
Use: "reflector", Use: "reflector",
Short: "Run reflector server", Short: "Run reflector server",
Run: reflectorCmd, Run: reflectorCmd,
} }
cmd.Flags().StringVar(&reflectorCmdCacheDir, "cache", "", "Enable disk cache for blobs. Store them in this directory")
rootCmd.AddCommand(cmd) rootCmd.AddCommand(cmd)
} }
@ -35,7 +38,6 @@ func reflectorCmd(cmd *cobra.Command, args []string) {
useDB := true useDB := true
s3 := store.NewS3BlobStore(globalConfig.AwsID, globalConfig.AwsSecret, globalConfig.BucketRegion, globalConfig.BucketName) s3 := store.NewS3BlobStore(globalConfig.AwsID, globalConfig.AwsSecret, globalConfig.BucketRegion, globalConfig.BucketName)
diskStore := store.NewDiskBlobStore("/home/lbry/lbry_downloaded_blobs", 2)
var err error var err error
@ -60,9 +62,16 @@ func reflectorCmd(cmd *cobra.Command, args []string) {
log.Fatal(err) log.Fatal(err)
} }
} }
cacheStore := store.NewCachingBlobStore(blobStore, diskStore)
peerServer := peer.NewServer(cacheStore) if reflectorCmdCacheDir != "" {
err = os.MkdirAll(reflectorCmdCacheDir, os.ModePerm)
if err != nil {
log.Fatal(err)
}
blobStore = store.NewCachingBlobStore(blobStore, store.NewDiskBlobStore(reflectorCmdCacheDir, 2))
}
peerServer := peer.NewServer(blobStore)
err = peerServer.Start(":5567") err = peerServer.Start(":5567")
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)