使用sshtunnel实现python公网连接阿里云mongo服务器

提示：扫一扫查出行
【扫一扫了解最新限行尾号】
复制提示

背景：

公司使用阿里云的云数据库MongoDB。基于安全原因考虑，阿里云MongoDB云数据库目前只支持从阿里云ECS上访问，无法通过公网直接访问，不方便用户在本地开发环境里直接进行测试。

阿里云官方推荐使用rinetd搭建跳板系统，实现公网访问云mongo。使用过程中，发现，python程序在使用过程中如果没有正确释放连接，即使python程序退出后，该mongo连接仍然被rinetd占用。曾出现调用该地址的python程序全部退出后，rinetd仍占用100多个连接，导致云mongo可用连接数不足。此时，需要重启rinetd方可释放被占用的连接。

最终使用本机ssh连接阿里云ECS，再从该主机使用私网连接到mongo的方法，实现云mongo公网访问。这种方法可以使开发环境中的robomongo连接到mongo。

python连接云mongo：

1 依赖

python -m pip install sshtunnel==0.1.2

2 demo

# -*- coding:utf-8 -*-
from __future__ import unicode_literalsimport pymongo
from sshtunnel import SSHTunnelForwarderclass AliyunMongoDebug(object):
def __init__(self):
self._record = {} def get_aliyun_mongo_client(self, only_read=True):
# 跳板机参数
ecs_host = "<ecs ip>"
ecs_user = "<ecs user>"
ecs_password = "<ecs password>" # 云mongo 配置
aliyun_mongo_master_host = "<mongo 副本集主服务器>"
aliyun_mongo_slave_host = "<mongo 副本集副本>"
aliyun_mongo_database = "<mongo 数据库>"
aliyun_mongo_account = "<mongo 账号>"
aliyun_mongo_password = "<mongo 密码>" host = aliyun_mongo_slave_host if only_read else aliyun_mongo_master_host
server = SSHTunnelForwarder(
(ecs_host, 22),
ssh_password=ecs_password,
ssh_username=ecs_user,
remote_bind_address=(host, 3717)) server.start() client = pymongo.MongoClient('127.0.0.1', server.local_bind_port)
mongo_database = client[aliyun_mongo_database]
mongo_database.authenticate(aliyun_mongo_account, aliyun_mongo_password) self._record[client] = server return client def return_aliyun_mongo_client(self, client):
if client in self._record:
server = self._record.pop(client)
else:
server = None client.close()
if server:
server.close()if __name__ == '__main__':
mongo_debug_manager = AliyunMongoDebug()
mongo_client = None
try:
mongo_client = mongo_debug_manager.get_aliyun_mongo_client(only_read=False)
mongo_client["Test"]["ssh_test"].insert({"msg": "Hello World!"})
finally:
if mongo_client:
mongo_debug_manager.return_aliyun_mongo_client(mongo_client)