Merge pull request c9s#519 from narumiruna/grpc/fix-query-klines-time

grpc: add start/end time to fix queryklines
SynapticDreams · Apr 12, 2022 · fad9449 · fad9449
2 parents 8705f38 + d836126
commit fad9449
Show file tree

Hide file tree

Showing 4 changed files with 184 additions and 160 deletions.
diff --git a/pkg/grpc/server.go b/pkg/grpc/server.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"fmt"
 	"net"
+	"time"
 
 	"github.com/pkg/errors"
 	log "github.com/sirupsen/logrus"
@@ -113,6 +114,17 @@ func (s *Server) QueryKLines(ctx context.Context, request *pb.QueryKLinesRequest
 				Limit: int(request.Limit),
 			}
 
+			endTime := time.Now()
+			if request.EndTime != 0 {
+				endTime = time.Unix(request.EndTime, 0)
+			}
+			options.EndTime = &endTime
+
+			if request.StartTime != 0 {
+				startTime := time.Unix(request.StartTime, 0)
+				options.StartTime = &startTime
+			}
+
 			klines, err := session.Exchange.QueryKLines(ctx, request.Symbol, types.Interval(request.Interval), options)
 			if err != nil {
 				return nil, err