Method UpdateMessage

pkg/session/store.go:288–313

UpdateMessage updates an existing message by its ID.

(_ context.Context, messageID int64, msg *Message)

// UpdateMessage updates an existing message by its ID.
func (s *InMemorySessionStore) UpdateMessage(_ context.Context, messageID int64, msg *Message) error {
	// Create a deep copy of the message to avoid mutating the caller's pointer,
	// which may be shared with another Session object.
	updated := cloneMessage(msg)
	updated.ID = messageID

	// For in-memory store, we need to find the message across all sessions
	var found bool
	s.sessions.Range(func(_ string, session *Session) bool {
		session.mu.Lock()
		defer session.mu.Unlock()
		for i := range session.Messages {
			if session.Messages[i].Message == nil || session.Messages[i].Message.ID != messageID {
				continue
			}
			session.Messages[i].Message = updated
			found = true
			return false
		}
		return true
	})
	if !found {
		return ErrNotFound
	}
	return nil
}

// AddSubSession creates a sub-session and links it to the parent.
func (s *InMemorySessionStore) AddSubSession(_ context.Context, parentSessionID string, subSession *Session) error {