grpc: add start/end time to fix queryklines

SynapticDreams · Apr 12, 2022 · d836126 · d836126
1 parent 8705f38
commit d836126
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