From 90867a2b7bd05f7d6196ece7a11bf19a4dd20894 Mon Sep 17 00:00:00 2001
From: Michael Ackley <mackley@connamara.com>
Date: Wed, 25 Sep 2024 11:16:40 -0500
Subject: [PATCH] Adds convenience getters for session log and store

---
 registry.go | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/registry.go b/registry.go
index 145c710c3..5f8e69fc2 100644
--- a/registry.go
+++ b/registry.go
@@ -129,6 +129,24 @@ func GetExpectedTargetNum(sessionID SessionID) (int, error) {
 	return session.store.NextTargetMsgSeqNum(), nil
 }
 
+// GetMessageStore returns the MessageStore interface for session matching the session id.
+func GetMessageStore(sessionID SessionID) (MessageStore, error) {
+	session, ok := lookupSession(sessionID)
+	if !ok {
+		return nil, errUnknownSession
+	}
+	return session.store, nil
+}
+
+// GetLog returns the Log interface for session matching the session id.
+func GetLog(sessionID SessionID) (Log, error) {
+	session, ok := lookupSession(sessionID)
+	if !ok {
+		return nil, errUnknownSession
+	}
+	return session.log, nil
+}
+
 func registerSession(s *session) error {
 	sessionsLock.Lock()
 	defer sessionsLock.Unlock()