Method _update_cache

src/environment/alpacaentry/bars.py:255–298

Update bars cache with a new bar record. Args: symbol: Symbol name new_bar: New bar data dictionary to add to cache

(self, symbol: str, new_bar: Dict)

            return False
    
    def _update_cache(self, symbol: str, new_bar: Dict) -> None:
        """Update bars cache with a new bar record.
        
        Args:
            symbol: Symbol name
            new_bar: New bar data dictionary to add to cache
        """
        try:
            # Convert new_bar to DataFrame row
            new_row = pd.DataFrame([new_bar])
            
            # Get current cache or create empty DataFrame
            if symbol in self._cache and not self._cache[symbol].empty:
                cached_df = self._cache[symbol]
                
                # Check if this bar already exists (by timestamp and id if available)
                if 'id' in new_bar and 'id' in cached_df.columns:
                    if new_bar['id'] in cached_df['id'].values:
                        # Bar already in cache, skip
                        return
                
                # Append new bar to cache
                cached_df = pd.concat([cached_df, new_row], ignore_index=True)
                
                # Sort by timestamp and id to maintain order
                if 'timestamp' in cached_df.columns:
                    cached_df = cached_df.sort_values('timestamp', ascending=True)
                    if 'id' in cached_df.columns:
                        cached_df = cached_df.sort_values(['timestamp', 'id'], ascending=True)
                
                # Remove oldest bars if cache exceeds limit
                if len(cached_df) > self._cache_limit:
                    cached_df = cached_df.tail(self._cache_limit).reset_index(drop=True)
                
                self._cache[symbol] = cached_df
            else:
                # Cache is empty, just set the new bar
                self._cache[symbol] = new_row
                
        except Exception as e:
            logger.warning(f"Failed to update bars cache for {symbol}: {e}")
            # On error, invalidate cache to force reload from database next time
            if symbol in self._cache:
                del self._cache[symbol]
    
    async def get_recent_bars(self, symbol: str, limit: int = 200) -> pd.DataFrame:
        """Get recent bars data for calculating technical indicators.