diff options
Diffstat (limited to 'cmd/cache.go')
| -rw-r--r-- | cmd/cache.go | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/cmd/cache.go b/cmd/cache.go index c72af16..5e17fdd 100644 --- a/cmd/cache.go +++ b/cmd/cache.go @@ -2,7 +2,6 @@ package cmd import ( "dtool/prober" - "dtool/scheduler" "dtool/utils" "github.com/spf13/cobra" @@ -11,6 +10,8 @@ import ( var query_cnt int var inputfile string var outputfile string +var goroutine_num int +var controlled_domain string var cacheCmd = &cobra.Command{ Use: "cache", Short: "cache related test", @@ -24,14 +25,16 @@ func cache_test(cmd *cobra.Command, args []string) { prober.RecursiveCacheTest(args[0], query_cnt) } } else { - scheduler.CreateTask(prober.RecursiveCacheProbe, inputfile, outputfile, 10) + prober.CreateTask(prober.RecursiveCacheProbe, controlled_domain, inputfile, outputfile, goroutine_num) } } func init() { + cacheCmd.Flags().StringVarP(&controlled_domain, "domain", "d", "echodns.xyz", "controlled domain") cacheCmd.Flags().StringVarP(&inputfile, "input", "i", "", "input file(optional)") cacheCmd.Flags().StringVarP(&outputfile, "output", "o", "", "output file(optional)") cacheCmd.MarkFlagsRequiredTogether("input", "output") cacheCmd.Flags().IntVarP(&query_cnt, "num", "n", 20, "number of queries in one test") + cacheCmd.Flags().IntVarP(&goroutine_num, "concurrency", "t", 150, "number of goroutine") rootCmd.AddCommand(cacheCmd) } |