Flush data points out of query range to prevent using heap more than …

…need

main.go

@@ -140,7 +140,8 @@ func (c *cli) run(args []string) int {
 		return 1
 	}
 
-	s, err := storage.NewStorage()
+	// Data points out of query range get flushed to prevent using heap more than need.
+	s, err := storage.NewStorage(c.queryRange * 2)
 	if err != nil {
 		fmt.Fprintf(c.stderr, "failed to initialize time-series storage: %v\n", err)
 		c.usage()

storage/storage.go

@@ -42,9 +42,10 @@ type Result struct {
 	P99       time.Duration
 }
 
-func NewStorage() (Storage, error) {
+func NewStorage(partitionDuration time.Duration) (Storage, error) {
 	s, err := tstorage.NewStorage(
 		tstorage.WithLogger(log.Default()),
+		tstorage.WithPartitionDuration(partitionDuration),
 	)
 	if err != nil {
 		return nil, err