Go语言通过SSH连接MongoDB：mgo是否有类似mysql.RegisterDial的函数？

Great question! Unlike the MySQL driver which has mysql.RegisterDial to override the default dialer, mgo doesn't have an exact equivalent function—but it does offer a flexible way to achieve the same SSH tunneling behavior using custom dial functions with mgo.DialWithInfo.

Here's how to adapt your existing SSH setup to work with mgo:

1. Establish your SSH connection first (just like you did for MySQL):

sshcon, err := ssh.Dial("tcp", fmt.Sprintf("%s:%d", sshHost, sshPort), sshConfig)
if err != nil {
    // Handle your error here (log, return, etc.)
    panic(err)
}
defer sshcon.Close()

2. Create a custom dial function that routes connections through your SSH tunnel:

// This function will use the existing SSH connection to dial the MongoDB server
customDial := func(addr string) (net.Conn, error) {
    return sshcon.Dial("tcp", addr)
}

3. Use mgo.DialWithInfo to connect and pass your custom dial function:

// Configure your MongoDB connection details
dialInfo := &mgo.DialInfo{
    Addrs:    []string{"your-mongo-host:27017"}, // e.g., "localhost:27017" if MongoDB runs locally on the remote server
    Username: "your-mongo-username", // Omit if you don't use authentication
    Password: "your-mongo-password", // Omit if you don't use authentication
    Dial:     customDial, // This tells mgo to use our SSH tunneled dialer
}

// Establish the MongoDB session
session, err := mgo.DialWithInfo(dialInfo)
if err != nil {
    // Handle connection error
    panic(err)
}
defer session.Close()

// Now you can use the session to interact with MongoDB
// Example: session.DB("your-db").C("your-collection").Find(nil)

How this works: #

Instead of registering a global dialer (like mysql.RegisterDial does), mgo lets you specify a dial function directly in the DialInfo struct. Every time mgo needs to establish a connection to MongoDB, it will use your customDial function—which routes the traffic through your active SSH tunnel, just like your MySQL setup did.

This approach is more explicit than the MySQL driver's global registration, but it achieves exactly the same result: all MongoDB connections are tunneled over SSH.

内容的提问来源于stack exchange，提问作者Vladimir Loct