diff options
| -rw-r--r-- | backend/cmd/main.go | 4 | ||||
| -rw-r--r-- | backend/internal/leader/manager/ibd/scrape/scrape.go | 2 | ||||
| -rw-r--r-- | backend/internal/server2/idb/stock/v1/stock.go | 7 | ||||
| -rw-r--r-- | backend/internal/worker/auth/auth.go | 2 |
4 files changed, 9 insertions, 6 deletions
diff --git a/backend/cmd/main.go b/backend/cmd/main.go index 8040560..37ee970 100644 --- a/backend/cmd/main.go +++ b/backend/cmd/main.go @@ -53,7 +53,9 @@ func main() { // Connect to the database db, err := connectDB(logger, cfg) - defer db.Close() + defer func(db database.Database) { + _ = db.Close() + }(db) if err != nil { log.Fatal("Unable to connect to database: ", err) } diff --git a/backend/internal/leader/manager/ibd/scrape/scrape.go b/backend/internal/leader/manager/ibd/scrape/scrape.go index 5f2c2a7..a3d4226 100644 --- a/backend/internal/leader/manager/ibd/scrape/scrape.go +++ b/backend/internal/leader/manager/ibd/scrape/scrape.go @@ -79,7 +79,7 @@ func (m *Manager) Run(ctx context.Context) { case <-timer.C: nextExec := m.schedule.Next(nextTime) m.scrapeIBD50(ctx, nextExec) - case _ = <-ch: + case <-ch: nextExec := m.schedule.Next(time.Now()) m.scrapeIBD50(ctx, nextExec) case <-ctx.Done(): diff --git a/backend/internal/server2/idb/stock/v1/stock.go b/backend/internal/server2/idb/stock/v1/stock.go index 3a94c82..30f0991 100644 --- a/backend/internal/server2/idb/stock/v1/stock.go +++ b/backend/internal/server2/idb/stock/v1/stock.go @@ -30,7 +30,7 @@ func New(db database.StockStore, queue taskqueue.TaskQueue[scrape.TaskInfo]) *Se return &Server{db: db, queue: queue} } -func (s *Server) CreateStock(ctx context.Context, request *pb.CreateStockRequest) (*longrunningpb.Operation, error) { +func (s *Server) CreateStock(ctx context.Context, request *pb.CreateStockRequest) (*pb.CreateStockResponse, error) { task, err := s.queue.Enqueue(ctx, scrape.TaskInfo{Symbol: request.Symbol}) if err != nil { slog.ErrorContext(ctx, "failed to enqueue task", "err", err) @@ -50,11 +50,12 @@ func (s *Server) CreateStock(ctx context.Context, request *pb.CreateStockRequest slog.ErrorContext(ctx, "failed to marshal metadata", "err", err) return nil, status.New(codes.Internal, "failed to marshal metadata").Err() } - return op, nil + return &pb.CreateStockResponse{Operation: op}, nil } func (s *Server) GetStock(ctx context.Context, request *pb.GetStockRequest) (*pb.GetStockResponse, error) { - + //TODO implement me + panic("implement me") } func (s *Server) ListStocks(ctx context.Context, request *pb.ListStocksRequest) (*pb.ListStocksResponse, error) { diff --git a/backend/internal/worker/auth/auth.go b/backend/internal/worker/auth/auth.go index e1c6661..b748d07 100644 --- a/backend/internal/worker/auth/auth.go +++ b/backend/internal/worker/auth/auth.go @@ -153,7 +153,7 @@ func hasValidCookies(ctx context.Context, store database.CookieStore, user strin } // If the user has non-degraded cookies, return true - if cookies != nil && len(cookies) > 0 { + if len(cookies) > 0 { return true, nil } return false, nil |