simon987/imhashdb
1
0
Fork 0
mirror of https://github.com/simon987/imhashdb.git synced 2025-04-04 02:22:58 +00:00
Code Issues Projects Releases Wiki Activity

fix store flag

Browse Source
This commit is contained in:
simon987 2020-05-30 15:41:50 -04:00
parent a25b5072a4
commit 46081a8355
2 changed files with 12 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
cli/main.go
View File

@ -89,13 +89,6 @@ func main() {
EnvVars: []string{"IMHASHDB_QUERY_CONCURRENCY"},
Destination: &Conf.QueryConcurrency,
},
&cli.StringFlag{
Name: "store",
Value: "",
Usage: "If set, store downloaded images there",
EnvVars: []string{"IMHASHDB_STORE"},
Destination: &Conf.Store,
},
},
},
{
@ -183,6 +176,13 @@ func main() {
EnvVars: []string{"IMHASHDB_HASH_CONCURRENCY"},
Destination: &Conf.HasherConcurrency,
},
&cli.StringFlag{
Name: "store",
Value: "",
Usage: "If set, store downloaded images there",
EnvVars: []string{"IMHASHDB_STORE"},
Destination: &Conf.Store,
},
},
},
},

7
hasher/hasher.go
View File

@ -144,11 +144,14 @@ func trimUrl(link string) string {
return link
}
var StoreData = Conf.Store != ""
var DataPath = Conf.Store
var StoreData bool
var DataPath string
var Pattern = "imhash.*"
func Main() error {
StoreData = Conf.Store != ""
DataPath = Conf.Store
queue := make(chan []string)
for i := 0; i < Conf.HasherConcurrency; i++ {