Add NewSession function

Signed-off-by: Michał Matczuk <michal@scylladb.com>
scylladb · May 21, 2020 · ae6d066 · ae6d066
1 parent a721151
commit ae6d066
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/session.go b/session.go
@@ -22,8 +22,19 @@ type Session struct {
 	Mapper *reflectx.Mapper
 }
 
+// NewSession wraps existing gocql.session.
+func NewSession(session *gocql.Session) Session {
+	return Session{
+		Session: session,
+		Mapper:  DefaultMapper,
+	}
+}
+
 // WrapSession should be called on CreateSession() gocql function to convert
 // the created session to gocqlx.Session.
+//
+// Example:
+//     session, err := gocqlx.WrapSession(cluster.CreateSession())
 func WrapSession(session *gocql.Session, err error) (Session, error) {
 	return Session{
 		Session: session,