Docs 13739 fix go connection (#651)

* DOCS-13739: fix go connection snippet

Author: Chris Cho

source/go.txt

@@ -50,20 +50,12 @@ Connect to MongoDB Atlas
 
 .. include:: /includes/atlas-connect-blurb.rst
 
-.. code-block:: go
-
-   import "go.mongodb.org/mongo-driver/mongo"
-
-   ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
-   client, err := mongo.Connect(ctx, options.Client().ApplyURI(
-      "mongodb+srv://<username>:<password>@<cluster-address>/test?w=majority"
-   ))
-   if err != nil { log.Fatal(err) }
+.. literalinclude:: /includes/connection-snippets/go-connection.go
+   :language: go
 
 See `Usage <https://github.com/mongodb/mongo-go-driver#usage>`__
 for more detail.
 
-
 Compatibility
 -------------
 

source/includes/connection-snippets/go-connection.go

@@ -0,0 +1,38 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"log"
+	"time"
+
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+	"go.mongodb.org/mongo-driver/mongo/readpref"
+)
+
+func main() {
+
+	// Replace the uri string with your MongoDB deployment's connection string.
+	uri := "mongodb+srv://<username>:<password>@<cluster-address>/test?w=majority"
+	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+	defer cancel()
+
+	client, err := mongo.Connect(ctx, options.Client().ApplyURI(uri))
+	if err != nil {
+		panic(err)
+	}
+
+	defer func() {
+		if err = client.Disconnect(ctx); err != nil {
+			panic(err)
+		}
+	}()
+
+	// Ping the primary
+	if err := client.Ping(ctx, readpref.Primary()); err != nil {
+		panic(err)
+	}
+
+	fmt.Println("Successfully connected and pinged.")
+}