mongo 1.12.5 → 2.0.0.beta

Sign up to get free protection for your applications and to get access to all the features.
Files changed (437) hide show
  1. checksums.yaml +4 -4
  2. checksums.yaml.gz.sig +0 -0
  3. data.tar.gz.sig +0 -0
  4. data/CONTRIBUTING.md +64 -0
  5. data/LICENSE +1 -1
  6. data/README.md +23 -125
  7. data/Rakefile +26 -21
  8. data/bin/mongo_console +6 -38
  9. data/lib/mongo.rb +23 -82
  10. data/lib/mongo/address.rb +111 -0
  11. data/lib/mongo/address/ipv4.rb +85 -0
  12. data/lib/mongo/address/ipv6.rb +85 -0
  13. data/lib/mongo/address/unix.rb +76 -0
  14. data/lib/mongo/auth.rb +108 -0
  15. data/lib/mongo/auth/cr.rb +44 -0
  16. data/lib/mongo/auth/cr/conversation.rb +119 -0
  17. data/lib/mongo/auth/executable.rb +52 -0
  18. data/lib/mongo/auth/ldap.rb +48 -0
  19. data/lib/mongo/auth/ldap/conversation.rb +92 -0
  20. data/lib/mongo/auth/roles.rb +104 -0
  21. data/lib/mongo/auth/scram.rb +53 -0
  22. data/lib/mongo/auth/scram/conversation.rb +450 -0
  23. data/lib/mongo/auth/user.rb +159 -0
  24. data/lib/mongo/auth/user/view.rb +102 -0
  25. data/lib/mongo/auth/x509.rb +48 -0
  26. data/lib/mongo/auth/x509/conversation.rb +92 -0
  27. data/lib/mongo/{gridfs.rb → bulk.rb} +2 -5
  28. data/lib/mongo/bulk/bulk_write.rb +307 -0
  29. data/lib/mongo/client.rb +233 -0
  30. data/lib/mongo/cluster.rb +203 -0
  31. data/lib/mongo/cluster/topology.rb +60 -0
  32. data/lib/mongo/cluster/topology/replica_set.rb +160 -0
  33. data/lib/mongo/cluster/topology/sharded.rb +132 -0
  34. data/lib/mongo/cluster/topology/standalone.rb +132 -0
  35. data/lib/mongo/cluster/topology/unknown.rb +155 -0
  36. data/lib/mongo/collection.rb +130 -1101
  37. data/lib/mongo/collection/view.rb +169 -0
  38. data/lib/mongo/collection/view/aggregation.rb +108 -0
  39. data/lib/mongo/collection/view/explainable.rb +49 -0
  40. data/lib/mongo/collection/view/immutable.rb +43 -0
  41. data/lib/mongo/collection/view/iterable.rb +48 -0
  42. data/lib/mongo/collection/view/map_reduce.rb +191 -0
  43. data/lib/mongo/collection/view/readable.rb +363 -0
  44. data/lib/mongo/collection/view/writable.rb +169 -0
  45. data/lib/mongo/cursor.rb +79 -680
  46. data/lib/mongo/database.rb +224 -0
  47. data/lib/mongo/database/view.rb +101 -0
  48. data/lib/mongo/error.rb +81 -0
  49. data/lib/mongo/error/bulk_write_failure.rb +41 -0
  50. data/lib/mongo/{utils/thread_local_variable_manager.rb → error/empty_batch.rb} +22 -8
  51. data/{test/functional/db_connection_test.rb → lib/mongo/error/invalid_bulk_operation.rb} +19 -8
  52. data/lib/mongo/error/invalid_collection_name.rb +39 -0
  53. data/lib/mongo/error/invalid_database_name.rb +39 -0
  54. data/{test/replica_set/ssl_test.rb → lib/mongo/error/invalid_document.rb} +21 -14
  55. data/lib/mongo/error/invalid_file.rb +38 -0
  56. data/lib/mongo/error/invalid_nonce.rb +46 -0
  57. data/lib/mongo/error/invalid_replacement_document.rb +39 -0
  58. data/lib/mongo/error/invalid_signature.rb +47 -0
  59. data/{test/functional/ssl_test.rb → lib/mongo/error/invalid_update_document.rb} +22 -12
  60. data/lib/mongo/error/max_bson_size.rb +40 -0
  61. data/lib/mongo/error/max_message_size.rb +42 -0
  62. data/lib/mongo/{utils.rb → error/need_primary_server.rb} +10 -6
  63. data/lib/mongo/{connection.rb → error/operation_failure.rb} +10 -6
  64. data/lib/mongo/error/parser.rb +77 -0
  65. data/lib/mongo/{connection/socket.rb → error/socket_error.rb} +10 -5
  66. data/lib/mongo/error/socket_timeout_error.rb +23 -0
  67. data/lib/mongo/error/unsupported_features.rb +43 -0
  68. data/lib/mongo/event.rb +40 -0
  69. data/lib/mongo/event/listeners.rb +63 -0
  70. data/lib/mongo/event/primary_elected.rb +53 -0
  71. data/lib/mongo/event/publisher.rb +42 -0
  72. data/lib/mongo/event/server_added.rb +53 -0
  73. data/lib/mongo/event/server_removed.rb +53 -0
  74. data/lib/mongo/event/subscriber.rb +41 -0
  75. data/lib/mongo/grid.rb +16 -0
  76. data/lib/mongo/grid/file.rb +94 -0
  77. data/lib/mongo/grid/file/chunk.rb +184 -0
  78. data/lib/mongo/grid/file/metadata.rb +223 -0
  79. data/lib/mongo/grid/fs.rb +149 -0
  80. data/lib/mongo/index.rb +64 -0
  81. data/lib/mongo/index/view.rb +205 -0
  82. data/lib/mongo/loggable.rb +126 -0
  83. data/lib/mongo/logger.rb +132 -0
  84. data/lib/mongo/operation.rb +26 -0
  85. data/lib/mongo/operation/aggregate.rb +100 -0
  86. data/lib/mongo/operation/aggregate/result.rb +84 -0
  87. data/lib/mongo/operation/batchable.rb +103 -0
  88. data/lib/mongo/operation/bulk_delete/result.rb +197 -0
  89. data/lib/mongo/operation/bulk_insert/result.rb +195 -0
  90. data/lib/mongo/operation/bulk_update/result.rb +295 -0
  91. data/lib/mongo/operation/command.rb +62 -0
  92. data/lib/mongo/operation/executable.rb +105 -0
  93. data/lib/mongo/operation/kill_cursors.rb +39 -0
  94. data/lib/mongo/operation/limited.rb +37 -0
  95. data/lib/mongo/operation/list_collections/result.rb +116 -0
  96. data/lib/mongo/operation/list_indexes/result.rb +118 -0
  97. data/lib/mongo/operation/map_reduce.rb +96 -0
  98. data/lib/mongo/operation/map_reduce/result.rb +122 -0
  99. data/lib/mongo/{functional.rb → operation/read.rb} +7 -7
  100. data/lib/mongo/operation/read/collections_info.rb +67 -0
  101. data/lib/mongo/operation/read/get_more.rb +71 -0
  102. data/lib/mongo/operation/read/indexes.rb +68 -0
  103. data/lib/mongo/operation/read/list_collections.rb +75 -0
  104. data/lib/mongo/operation/read/list_indexes.rb +77 -0
  105. data/lib/mongo/operation/read/query.rb +71 -0
  106. data/lib/mongo/operation/read_preferrable.rb +34 -0
  107. data/lib/mongo/operation/result.rb +259 -0
  108. data/lib/mongo/operation/specifiable.rb +380 -0
  109. data/lib/mongo/operation/write.rb +25 -0
  110. data/lib/mongo/operation/write/bulk_delete.rb +158 -0
  111. data/lib/mongo/operation/write/bulk_insert.rb +160 -0
  112. data/lib/mongo/operation/write/bulk_update.rb +167 -0
  113. data/lib/mongo/{connection/socket/socket_util.rb → operation/write/command.rb} +9 -24
  114. data/lib/mongo/operation/write/command/create_user.rb +43 -0
  115. data/lib/mongo/operation/write/command/delete.rb +56 -0
  116. data/lib/mongo/operation/write/command/drop_index.rb +51 -0
  117. data/lib/mongo/operation/write/command/ensure_index.rb +55 -0
  118. data/lib/mongo/operation/write/command/insert.rb +55 -0
  119. data/lib/mongo/operation/write/command/remove_user.rb +42 -0
  120. data/lib/mongo/operation/write/command/update.rb +60 -0
  121. data/lib/mongo/operation/write/command/writable.rb +61 -0
  122. data/lib/mongo/operation/write/create_index.rb +84 -0
  123. data/lib/mongo/operation/write/create_user.rb +75 -0
  124. data/lib/mongo/operation/write/delete.rb +91 -0
  125. data/lib/mongo/operation/write/drop_index.rb +62 -0
  126. data/lib/mongo/operation/write/insert.rb +88 -0
  127. data/lib/mongo/operation/write/remove_user.rb +70 -0
  128. data/lib/mongo/operation/write/update.rb +98 -0
  129. data/lib/mongo/protocol.rb +15 -0
  130. data/lib/mongo/protocol/bit_vector.rb +61 -0
  131. data/lib/mongo/protocol/delete.rb +94 -0
  132. data/lib/mongo/protocol/get_more.rb +99 -0
  133. data/lib/mongo/protocol/insert.rb +99 -0
  134. data/lib/mongo/protocol/kill_cursors.rb +74 -0
  135. data/lib/mongo/protocol/message.rb +252 -0
  136. data/lib/mongo/protocol/query.rb +147 -0
  137. data/lib/mongo/protocol/reply.rb +72 -0
  138. data/lib/mongo/protocol/serializers.rb +180 -0
  139. data/lib/mongo/protocol/update.rb +111 -0
  140. data/lib/mongo/server.rb +163 -0
  141. data/lib/mongo/server/connectable.rb +99 -0
  142. data/lib/mongo/server/connection.rb +133 -0
  143. data/lib/mongo/server/connection_pool.rb +141 -0
  144. data/lib/mongo/server/connection_pool/queue.rb +182 -0
  145. data/lib/mongo/server/context.rb +66 -0
  146. data/lib/mongo/server/description.rb +450 -0
  147. data/lib/mongo/server/description/features.rb +85 -0
  148. data/lib/mongo/server/description/inspector.rb +79 -0
  149. data/lib/mongo/server/description/inspector/primary_elected.rb +58 -0
  150. data/lib/mongo/server/description/inspector/server_added.rb +59 -0
  151. data/lib/mongo/server/description/inspector/server_removed.rb +59 -0
  152. data/lib/mongo/server/monitor.rb +160 -0
  153. data/lib/mongo/server/monitor/connection.rb +88 -0
  154. data/lib/mongo/server_selector.rb +81 -0
  155. data/lib/mongo/server_selector/nearest.rb +94 -0
  156. data/lib/mongo/server_selector/primary.rb +88 -0
  157. data/lib/mongo/server_selector/primary_preferred.rb +94 -0
  158. data/lib/mongo/server_selector/secondary.rb +91 -0
  159. data/lib/mongo/server_selector/secondary_preferred.rb +96 -0
  160. data/lib/mongo/server_selector/selectable.rb +209 -0
  161. data/lib/mongo/socket.rb +179 -0
  162. data/lib/mongo/socket/ssl.rb +108 -0
  163. data/lib/mongo/socket/tcp.rb +69 -0
  164. data/lib/mongo/socket/unix.rb +66 -0
  165. data/lib/mongo/uri.rb +504 -0
  166. data/lib/mongo/version.rb +21 -0
  167. data/lib/mongo/write_concern.rb +99 -0
  168. data/lib/mongo/write_concern/acknowledged.rb +38 -0
  169. data/lib/mongo/write_concern/normalizable.rb +73 -0
  170. data/lib/mongo/write_concern/unacknowledged.rb +43 -0
  171. data/mongo.gemspec +17 -14
  172. data/spec/mongo/address/ipv4_spec.rb +74 -0
  173. data/spec/mongo/address/ipv6_spec.rb +74 -0
  174. data/spec/mongo/address/unix_spec.rb +30 -0
  175. data/spec/mongo/address_spec.rb +206 -0
  176. data/spec/mongo/auth/cr_spec.rb +59 -0
  177. data/spec/mongo/auth/ldap_spec.rb +40 -0
  178. data/spec/mongo/auth/scram/conversation_spec.rb +197 -0
  179. data/spec/mongo/auth/scram_spec.rb +55 -0
  180. data/spec/mongo/auth/user/view_spec.rb +76 -0
  181. data/spec/mongo/auth/user_spec.rb +190 -0
  182. data/spec/mongo/auth/x509_spec.rb +40 -0
  183. data/spec/mongo/auth_spec.rb +65 -0
  184. data/spec/mongo/bulk/bulk_write_spec.rb +175 -0
  185. data/spec/mongo/client_spec.rb +564 -0
  186. data/spec/mongo/cluster/topology/replica_set_spec.rb +101 -0
  187. data/spec/mongo/cluster/topology/sharded_spec.rb +74 -0
  188. data/spec/mongo/cluster/topology/standalone_spec.rb +79 -0
  189. data/spec/mongo/cluster/topology_spec.rb +65 -0
  190. data/spec/mongo/cluster_spec.rb +129 -0
  191. data/spec/mongo/collection/view/aggregation_spec.rb +135 -0
  192. data/spec/mongo/collection/view/explainable_spec.rb +32 -0
  193. data/spec/mongo/collection/view/map_reduce_spec.rb +242 -0
  194. data/spec/mongo/collection/view/readable_spec.rb +603 -0
  195. data/spec/mongo/collection/view/writable_spec.rb +504 -0
  196. data/spec/mongo/collection/view_spec.rb +521 -0
  197. data/spec/mongo/collection_spec.rb +362 -0
  198. data/spec/mongo/cursor_spec.rb +295 -0
  199. data/spec/mongo/database_spec.rb +306 -0
  200. data/spec/mongo/error/parser_spec.rb +119 -0
  201. data/spec/mongo/event/publisher_spec.rb +50 -0
  202. data/spec/mongo/event/subscriber_spec.rb +34 -0
  203. data/spec/mongo/grid/file/chunk_spec.rb +226 -0
  204. data/spec/mongo/grid/file/metadata_spec.rb +69 -0
  205. data/spec/mongo/grid/file_spec.rb +138 -0
  206. data/spec/mongo/grid/fs_spec.rb +129 -0
  207. data/spec/mongo/index/view_spec.rb +226 -0
  208. data/spec/mongo/loggable_spec.rb +62 -0
  209. data/spec/mongo/logger_spec.rb +97 -0
  210. data/spec/mongo/operation/aggregate/result_spec.rb +80 -0
  211. data/spec/mongo/operation/aggregate_spec.rb +135 -0
  212. data/spec/mongo/operation/command_spec.rb +106 -0
  213. data/spec/mongo/operation/kill_cursors_spec.rb +66 -0
  214. data/spec/mongo/operation/limited_spec.rb +50 -0
  215. data/spec/mongo/operation/map_reduce_spec.rb +143 -0
  216. data/spec/mongo/operation/read/collections_info_spec.rb +40 -0
  217. data/spec/mongo/operation/read/get_more_spec.rb +81 -0
  218. data/spec/mongo/operation/read/indexes_spec.rb +31 -0
  219. data/spec/mongo/operation/read/query_spec.rb +84 -0
  220. data/spec/mongo/operation/result_spec.rb +275 -0
  221. data/spec/mongo/operation/specifiable_spec.rb +53 -0
  222. data/spec/mongo/operation/write/bulk_delete_spec.rb +473 -0
  223. data/spec/mongo/operation/write/bulk_insert_spec.rb +466 -0
  224. data/spec/mongo/operation/write/bulk_update_spec.rb +524 -0
  225. data/spec/mongo/operation/write/command/delete_spec.rb +116 -0
  226. data/spec/mongo/operation/write/command/insert_spec.rb +117 -0
  227. data/spec/mongo/operation/write/command/update_spec.rb +123 -0
  228. data/spec/mongo/operation/write/create_user_spec.rb +44 -0
  229. data/spec/mongo/operation/write/delete_spec.rb +178 -0
  230. data/spec/mongo/operation/write/drop_index_spec.rb +51 -0
  231. data/spec/mongo/operation/write/ensure_index_spec.rb +81 -0
  232. data/spec/mongo/operation/write/insert_spec.rb +231 -0
  233. data/spec/mongo/operation/write/remove_user_spec.rb +46 -0
  234. data/spec/mongo/operation/write/response_spec.rb +85 -0
  235. data/spec/mongo/operation/write/update_spec.rb +177 -0
  236. data/spec/mongo/protocol/delete_spec.rb +167 -0
  237. data/spec/mongo/protocol/get_more_spec.rb +146 -0
  238. data/spec/mongo/protocol/insert_spec.rb +161 -0
  239. data/spec/mongo/protocol/kill_cursors_spec.rb +101 -0
  240. data/spec/mongo/protocol/query_spec.rb +285 -0
  241. data/spec/mongo/protocol/reply_spec.rb +157 -0
  242. data/spec/mongo/protocol/update_spec.rb +186 -0
  243. data/spec/mongo/server/connection_pool/queue_spec.rb +170 -0
  244. data/spec/mongo/server/connection_pool_spec.rb +120 -0
  245. data/spec/mongo/server/connection_spec.rb +289 -0
  246. data/spec/mongo/server/description/features_spec.rb +138 -0
  247. data/spec/mongo/server/description/inspector/primary_elected_spec.rb +94 -0
  248. data/spec/mongo/server/description/inspector/server_added_spec.rb +92 -0
  249. data/spec/mongo/server/description/inspector/server_removed_spec.rb +95 -0
  250. data/spec/mongo/server/description_spec.rb +510 -0
  251. data/spec/mongo/server/monitor_spec.rb +130 -0
  252. data/spec/mongo/server_discovery_and_monitoring_spec.rb +103 -0
  253. data/spec/mongo/server_selection_rtt_spec.rb +104 -0
  254. data/spec/mongo/server_selection_spec.rb +89 -0
  255. data/spec/mongo/server_selector/nearest_spec.rb +250 -0
  256. data/spec/mongo/server_selector/primary_preferred_spec.rb +290 -0
  257. data/spec/mongo/server_selector/primary_spec.rb +114 -0
  258. data/spec/mongo/server_selector/secondary_preferred_spec.rb +252 -0
  259. data/spec/mongo/server_selector/secondary_spec.rb +196 -0
  260. data/spec/mongo/server_selector_spec.rb +101 -0
  261. data/spec/mongo/server_spec.rb +131 -0
  262. data/spec/mongo/uri_spec.rb +517 -0
  263. data/spec/mongo/write_concern/acknowledged_spec.rb +44 -0
  264. data/spec/mongo/write_concern/unacknowledged_spec.rb +15 -0
  265. data/spec/mongo_orchestration_spec.rb +70 -0
  266. data/spec/spec_helper.rb +148 -0
  267. data/spec/support/authorization.rb +245 -0
  268. data/spec/support/helpers.rb +140 -0
  269. data/spec/support/matchers.rb +37 -0
  270. data/spec/support/mongo_orchestration.rb +61 -0
  271. data/spec/support/mongo_orchestration/requestable.rb +109 -0
  272. data/spec/support/mongo_orchestration/standalone.rb +57 -0
  273. data/spec/support/sdam/rs/discover_arbiters.yml +41 -0
  274. data/spec/support/sdam/rs/discover_passives.yml +41 -0
  275. data/spec/support/sdam/rs/discover_primary.yml +40 -0
  276. data/spec/support/sdam/rs/discover_secondary.yml +41 -0
  277. data/spec/support/sdam/rs/discovery.yml +195 -0
  278. data/spec/support/sdam/rs/ghost_discovered.yml +39 -0
  279. data/spec/support/sdam/rs/hosts_differ_from_seeds.yml +34 -0
  280. data/spec/support/sdam/rs/member_reconfig.yml +68 -0
  281. data/spec/support/sdam/rs/member_standalone.yml +60 -0
  282. data/spec/support/sdam/rs/new_primary.yml +74 -0
  283. data/spec/support/sdam/rs/new_primary_wrong_set_name.yml +71 -0
  284. data/spec/support/sdam/rs/non_rs_member.yml +31 -0
  285. data/spec/support/sdam/rs/normalize_case.yml +49 -0
  286. data/spec/support/sdam/rs/primary_becomes_standalone.yml +52 -0
  287. data/spec/support/sdam/rs/primary_changes_set_name.yml +57 -0
  288. data/spec/support/sdam/rs/primary_disconnect.yml +56 -0
  289. data/spec/support/sdam/rs/primary_wrong_set_name.yml +27 -0
  290. data/spec/support/sdam/rs/response_from_removed.yml +63 -0
  291. data/spec/support/sdam/rs/rsother_discovered.yml +41 -0
  292. data/spec/support/sdam/rs/sec_not_auth.yml +49 -0
  293. data/spec/support/sdam/rs/secondary_wrong_set_name.yml +28 -0
  294. data/spec/support/sdam/rs/secondary_wrong_set_name_with_primary.yml +69 -0
  295. data/spec/support/sdam/rs/unexpected_mongos.yml +26 -0
  296. data/spec/support/sdam/rs/wrong_set_name.yml +35 -0
  297. data/spec/support/sdam/sharded/multiple_mongoses.yml +46 -0
  298. data/spec/support/sdam/sharded/non_mongos_removed.yml +41 -0
  299. data/spec/support/sdam/sharded/normalize_uri_case.yml +32 -0
  300. data/spec/support/sdam/single/direct_connection_external_ip.yml +34 -0
  301. data/spec/support/sdam/single/direct_connection_mongos.yml +33 -0
  302. data/spec/support/sdam/single/direct_connection_rsarbiter.yml +35 -0
  303. data/spec/support/sdam/single/direct_connection_rsprimary.yml +34 -0
  304. data/spec/support/sdam/single/direct_connection_rssecondary.yml +35 -0
  305. data/spec/support/sdam/single/direct_connection_slave.yml +32 -0
  306. data/spec/support/sdam/single/direct_connection_standalone.yml +32 -0
  307. data/spec/support/sdam/single/not_ok_response.yml +39 -0
  308. data/spec/support/sdam/single/standalone_removed.yml +32 -0
  309. data/spec/support/sdam/single/unavailable_seed.yml +28 -0
  310. data/spec/support/server_discovery_and_monitoring.rb +167 -0
  311. data/spec/support/server_selection.rb +140 -0
  312. data/spec/support/server_selection/rtt/first_value.yml +4 -0
  313. data/spec/support/server_selection/rtt/first_value_zero.yml +4 -0
  314. data/spec/support/server_selection/rtt/value_test_1.yml +4 -0
  315. data/spec/support/server_selection/rtt/value_test_2.yml +4 -0
  316. data/spec/support/server_selection/rtt/value_test_3.yml +4 -0
  317. data/spec/support/server_selection/rtt/value_test_4.yml +4 -0
  318. data/spec/support/server_selection/rtt/value_test_5.yml +4 -0
  319. data/spec/support/server_selection/selection/ReplicaSetNoPrimary/read/Nearest.yml +32 -0
  320. data/spec/support/server_selection/selection/ReplicaSetNoPrimary/read/Nearest_non_matching.yml +27 -0
  321. data/spec/support/server_selection/selection/ReplicaSetNoPrimary/read/Primary.yml +23 -0
  322. data/spec/support/server_selection/selection/ReplicaSetNoPrimary/read/PrimaryPreferred.yml +32 -0
  323. data/spec/support/server_selection/selection/ReplicaSetNoPrimary/read/PrimaryPreferred_non_matching.yml +27 -0
  324. data/spec/support/server_selection/selection/ReplicaSetNoPrimary/read/Secondary.yml +32 -0
  325. data/spec/support/server_selection/selection/ReplicaSetNoPrimary/read/SecondaryPreferred.yml +32 -0
  326. data/spec/support/server_selection/selection/ReplicaSetNoPrimary/read/SecondaryPreferred_non_matching.yml +27 -0
  327. data/spec/support/server_selection/selection/ReplicaSetNoPrimary/read/Secondary_non_matching.yml +27 -0
  328. data/spec/support/server_selection/selection/ReplicaSetWithPrimary/read/Nearest.yml +41 -0
  329. data/spec/support/server_selection/selection/ReplicaSetWithPrimary/read/Nearest_non_matching.yml +34 -0
  330. data/spec/support/server_selection/selection/ReplicaSetWithPrimary/read/Primary.yml +33 -0
  331. data/spec/support/server_selection/selection/ReplicaSetWithPrimary/read/PrimaryPreferred.yml +39 -0
  332. data/spec/support/server_selection/selection/ReplicaSetWithPrimary/read/PrimaryPreferred_non_matching.yml +36 -0
  333. data/spec/support/server_selection/selection/Sharded/read/SecondaryPreferred.yml +32 -0
  334. data/spec/support/server_selection/selection/Single/read/SecondaryPreferred.yml +23 -0
  335. data/spec/support/server_selection/selection/Unknown/read/SecondaryPreferred.yml +13 -0
  336. data/spec/support/server_selection_rtt.rb +41 -0
  337. data/spec/support/shared/bulk_write.rb +498 -0
  338. data/spec/support/shared/cursor.rb +38 -0
  339. data/spec/support/shared/operation.rb +77 -0
  340. data/spec/support/shared/protocol.rb +31 -0
  341. data/spec/support/shared/server_selector.rb +111 -0
  342. data/spec/support/shared/socket.rb +82 -0
  343. data/spec/support/travis.rb +14 -0
  344. metadata +523 -189
  345. metadata.gz.sig +0 -0
  346. data/VERSION +0 -1
  347. data/lib/mongo/bulk_write_collection_view.rb +0 -387
  348. data/lib/mongo/collection_writer.rb +0 -364
  349. data/lib/mongo/connection/node.rb +0 -249
  350. data/lib/mongo/connection/pool.rb +0 -340
  351. data/lib/mongo/connection/pool_manager.rb +0 -320
  352. data/lib/mongo/connection/sharding_pool_manager.rb +0 -67
  353. data/lib/mongo/connection/socket/ssl_socket.rb +0 -95
  354. data/lib/mongo/connection/socket/tcp_socket.rb +0 -87
  355. data/lib/mongo/connection/socket/unix_socket.rb +0 -39
  356. data/lib/mongo/db.rb +0 -808
  357. data/lib/mongo/exception.rb +0 -145
  358. data/lib/mongo/functional/authentication.rb +0 -455
  359. data/lib/mongo/functional/logging.rb +0 -85
  360. data/lib/mongo/functional/read_preference.rb +0 -183
  361. data/lib/mongo/functional/scram.rb +0 -556
  362. data/lib/mongo/functional/uri_parser.rb +0 -409
  363. data/lib/mongo/functional/write_concern.rb +0 -66
  364. data/lib/mongo/gridfs/grid.rb +0 -112
  365. data/lib/mongo/gridfs/grid_ext.rb +0 -53
  366. data/lib/mongo/gridfs/grid_file_system.rb +0 -163
  367. data/lib/mongo/gridfs/grid_io.rb +0 -484
  368. data/lib/mongo/legacy.rb +0 -140
  369. data/lib/mongo/mongo_client.rb +0 -697
  370. data/lib/mongo/mongo_replica_set_client.rb +0 -535
  371. data/lib/mongo/mongo_sharded_client.rb +0 -159
  372. data/lib/mongo/networking.rb +0 -372
  373. data/lib/mongo/utils/conversions.rb +0 -110
  374. data/lib/mongo/utils/core_ext.rb +0 -70
  375. data/lib/mongo/utils/server_version.rb +0 -69
  376. data/lib/mongo/utils/support.rb +0 -80
  377. data/test/functional/authentication_test.rb +0 -39
  378. data/test/functional/bulk_api_stress_test.rb +0 -133
  379. data/test/functional/bulk_write_collection_view_test.rb +0 -1198
  380. data/test/functional/client_test.rb +0 -627
  381. data/test/functional/collection_test.rb +0 -2175
  382. data/test/functional/collection_writer_test.rb +0 -83
  383. data/test/functional/conversions_test.rb +0 -163
  384. data/test/functional/cursor_fail_test.rb +0 -57
  385. data/test/functional/cursor_message_test.rb +0 -56
  386. data/test/functional/cursor_test.rb +0 -683
  387. data/test/functional/db_api_test.rb +0 -835
  388. data/test/functional/db_test.rb +0 -348
  389. data/test/functional/grid_file_system_test.rb +0 -285
  390. data/test/functional/grid_io_test.rb +0 -252
  391. data/test/functional/grid_test.rb +0 -273
  392. data/test/functional/pool_test.rb +0 -136
  393. data/test/functional/safe_test.rb +0 -98
  394. data/test/functional/support_test.rb +0 -62
  395. data/test/functional/timeout_test.rb +0 -60
  396. data/test/functional/uri_test.rb +0 -446
  397. data/test/functional/write_concern_test.rb +0 -118
  398. data/test/helpers/general.rb +0 -50
  399. data/test/helpers/test_unit.rb +0 -476
  400. data/test/replica_set/authentication_test.rb +0 -37
  401. data/test/replica_set/basic_test.rb +0 -189
  402. data/test/replica_set/client_test.rb +0 -393
  403. data/test/replica_set/connection_test.rb +0 -138
  404. data/test/replica_set/count_test.rb +0 -66
  405. data/test/replica_set/cursor_test.rb +0 -220
  406. data/test/replica_set/insert_test.rb +0 -157
  407. data/test/replica_set/max_values_test.rb +0 -151
  408. data/test/replica_set/pinning_test.rb +0 -105
  409. data/test/replica_set/query_test.rb +0 -73
  410. data/test/replica_set/read_preference_test.rb +0 -219
  411. data/test/replica_set/refresh_test.rb +0 -211
  412. data/test/replica_set/replication_ack_test.rb +0 -95
  413. data/test/sharded_cluster/basic_test.rb +0 -203
  414. data/test/shared/authentication/basic_auth_shared.rb +0 -260
  415. data/test/shared/authentication/bulk_api_auth_shared.rb +0 -249
  416. data/test/shared/authentication/gssapi_shared.rb +0 -176
  417. data/test/shared/authentication/sasl_plain_shared.rb +0 -96
  418. data/test/shared/authentication/scram_shared.rb +0 -92
  419. data/test/shared/ssl_shared.rb +0 -235
  420. data/test/test_helper.rb +0 -61
  421. data/test/threading/basic_test.rb +0 -120
  422. data/test/tools/mongo_config.rb +0 -708
  423. data/test/tools/mongo_config_test.rb +0 -160
  424. data/test/unit/client_test.rb +0 -381
  425. data/test/unit/collection_test.rb +0 -166
  426. data/test/unit/connection_test.rb +0 -335
  427. data/test/unit/cursor_test.rb +0 -307
  428. data/test/unit/db_test.rb +0 -136
  429. data/test/unit/grid_test.rb +0 -76
  430. data/test/unit/mongo_sharded_client_test.rb +0 -48
  431. data/test/unit/node_test.rb +0 -93
  432. data/test/unit/pool_manager_test.rb +0 -111
  433. data/test/unit/read_pref_test.rb +0 -406
  434. data/test/unit/read_test.rb +0 -159
  435. data/test/unit/safe_test.rb +0 -158
  436. data/test/unit/sharding_pool_manager_test.rb +0 -84
  437. data/test/unit/write_concern_test.rb +0 -175
data/lib/mongo/legacy.rb DELETED
@@ -1,140 +0,0 @@
1
- # Copyright (C) 2009-2013 MongoDB, Inc.
2
- #
3
- # Licensed under the Apache License, Version 2.0 (the "License");
4
- # you may not use this file except in compliance with the License.
5
- # You may obtain a copy of the License at
6
- #
7
- # http://www.apache.org/licenses/LICENSE-2.0
8
- #
9
- # Unless required by applicable law or agreed to in writing, software
10
- # distributed under the License is distributed on an "AS IS" BASIS,
11
- # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
- # See the License for the specific language governing permissions and
13
- # limitations under the License.
14
-
15
- module Mongo
16
- module LegacyWriteConcern
17
- @legacy_write_concern = true
18
-
19
- def safe=(value)
20
- @write_concern = value
21
- end
22
-
23
- def safe
24
- if @write_concern[:w] == 0
25
- return false
26
- elsif @write_concern[:w] == 1
27
- return true
28
- else
29
- return @write_concern
30
- end
31
- end
32
-
33
- def self.from_uri(uri = ENV['MONGODB_URI'], extra_opts={})
34
- parser = URIParser.new uri
35
- parser.connection(extra_opts, true)
36
- end
37
- end
38
- end
39
-
40
- module Mongo
41
- # @deprecated Use Mongo::MongoClient instead. Support will be removed after
42
- # v2.0. Please see old documentation for the Connection class.
43
- class Connection < MongoClient
44
- include Mongo::LegacyWriteConcern
45
-
46
- def initialize(*args)
47
- if args.last.is_a?(Hash)
48
- opts = args.pop
49
- write_concern_from_legacy(opts)
50
- args.push(opts)
51
- end
52
- super
53
- end
54
- end
55
-
56
- # @deprecated Use Mongo::MongoReplicaSetClient instead. Support will be
57
- # removed after v2.0. Please see old documentation for the
58
- # ReplSetConnection class.
59
- class ReplSetConnection < MongoReplicaSetClient
60
- include Mongo::LegacyWriteConcern
61
-
62
- def initialize(*args)
63
- if args.last.is_a?(Hash)
64
- opts = args.pop
65
- write_concern_from_legacy(opts)
66
- args.push(opts)
67
- end
68
- super
69
- end
70
- end
71
-
72
- # @deprecated Use Mongo::MongoShardedClient instead. Support will be removed
73
- # after v2.0. Please see old documentation for the ShardedConnection class.
74
- class ShardedConnection < MongoShardedClient
75
- include Mongo::LegacyWriteConcern
76
-
77
- def initialize(*args)
78
- if args.last.is_a?(Hash)
79
- opts = args.pop
80
- write_concern_from_legacy(opts)
81
- args.push(opts)
82
- end
83
- super
84
- end
85
- end
86
-
87
- class MongoClient
88
- # @deprecated This method is no longer in use and never needs to be called
89
- # directly. Support will be removed after v2.0
90
- # Authentication of sockets is handled upon checkout and checkin.
91
- def authenticate_pools
92
- end
93
-
94
- # @deprecated This method is no longer in use and never needs to be called
95
- # directly. Support will be removed after v2.0
96
- # Authentication of sockets is handled upon checkout and checkin.
97
- def logout_pools(database)
98
- end
99
-
100
- # @deprecated This method is no longer in use and never needs to be called
101
- # directly. Support will be removed after v2.0
102
- def apply_saved_authentication
103
- true
104
- end
105
- end
106
-
107
- class MongoReplicaSetClient
108
- # @deprecated This method is no longer in use and never needs to be called
109
- # directly. Support will be removed after v2.0
110
- # Authentication of sockets is handled upon checkout and checkin.
111
- def authenticate_pools
112
- end
113
-
114
- # @deprecated This method is no longer in use and never needs to be called
115
- # directly. Support will be removed after v2.0
116
- # Authentication of sockets is handled upon checkout and checkin.
117
- def logout_pools(database)
118
- end
119
- end
120
-
121
- class DB
122
- # @deprecated Please use MongoClient#issue_authentication instead. Support
123
- # will be removed after v2.0
124
- def issue_authentication(username, password, save_auth=true, opts={})
125
- auth = Authentication.validate_credentials({
126
- :db_name => self.name,
127
- :username => username,
128
- :password => password
129
- })
130
- opts[:save_auth] = save_auth
131
- @client.issue_authentication(auth, opts)
132
- end
133
-
134
- # @deprecated Please use MongoClient#issue_logout instead. Support will be
135
- # removed after v2.0
136
- def issue_logout(opts={})
137
- @client.issue_logout(self.name, opts)
138
- end
139
- end
140
- end
@@ -1,697 +0,0 @@
1
- # Copyright (C) 2009-2013 MongoDB, Inc.
2
- #
3
- # Licensed under the Apache License, Version 2.0 (the "License");
4
- # you may not use this file except in compliance with the License.
5
- # You may obtain a copy of the License at
6
- #
7
- # http://www.apache.org/licenses/LICENSE-2.0
8
- #
9
- # Unless required by applicable law or agreed to in writing, software
10
- # distributed under the License is distributed on an "AS IS" BASIS,
11
- # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
- # See the License for the specific language governing permissions and
13
- # limitations under the License.
14
-
15
- module Mongo
16
-
17
- # Instantiates and manages self.connections to MongoDB.
18
- class MongoClient
19
- include Mongo::Logging
20
- include Mongo::Networking
21
- include Mongo::WriteConcern
22
- include Mongo::Authentication
23
-
24
- # Wire version
25
- RELEASE_2_4_AND_BEFORE = 0 # Everything before we started tracking.
26
- AGG_RETURNS_CURSORS = 1 # The aggregation command may now be requested to return cursors.
27
- BATCH_COMMANDS = 2 # insert, update, and delete batch command
28
- MONGODB_3_0 = 3 # listCollections and listIndexes commands, SCRAM-SHA-1 auth mechanism
29
- MAX_WIRE_VERSION = MONGODB_3_0 # supported by this client implementation
30
- MIN_WIRE_VERSION = RELEASE_2_4_AND_BEFORE # supported by this client implementation
31
-
32
- # Server command headroom
33
- COMMAND_HEADROOM = 16_384
34
- APPEND_HEADROOM = COMMAND_HEADROOM / 2
35
- SERIALIZE_HEADROOM = APPEND_HEADROOM / 2
36
-
37
- DEFAULT_MAX_WRITE_BATCH_SIZE = 1000
38
-
39
- Mutex = ::Mutex
40
- ConditionVariable = ::ConditionVariable
41
-
42
- DEFAULT_HOST = 'localhost'
43
- DEFAULT_PORT = 27017
44
- DEFAULT_DB_NAME = 'test'
45
- DEFAULT_OP_TIMEOUT = 20
46
- GENERIC_OPTS = [:auths, :logger, :connect, :db_name]
47
- TIMEOUT_OPTS = [:timeout, :op_timeout, :connect_timeout]
48
- SSL_OPTS = [:ssl, :ssl_key, :ssl_cert, :ssl_verify, :ssl_ca_cert, :ssl_key_pass_phrase]
49
- POOL_OPTS = [:pool_size, :pool_timeout]
50
- READ_PREFERENCE_OPTS = [:read, :tag_sets, :secondary_acceptable_latency_ms]
51
- WRITE_CONCERN_OPTS = [:w, :j, :fsync, :wtimeout]
52
- CLIENT_ONLY_OPTS = [:slave_ok]
53
-
54
- mongo_thread_local_accessor :connections
55
-
56
- attr_reader :logger,
57
- :size,
58
- :auths,
59
- :primary,
60
- :write_concern,
61
- :host_to_try,
62
- :pool_size,
63
- :connect_timeout,
64
- :pool_timeout,
65
- :primary_pool,
66
- :socket_class,
67
- :socket_opts,
68
- :op_timeout,
69
- :tag_sets,
70
- :acceptable_latency,
71
- :read,
72
- :max_wire_version,
73
- :min_wire_version,
74
- :max_write_batch_size
75
-
76
- # Create a connection to single MongoDB instance.
77
- #
78
- # If no args are provided, it will check <code>ENV["MONGODB_URI"]</code>.
79
- #
80
- # You may specify whether connection to slave is permitted.
81
- # In all cases, the default host is "localhost" and the default port is 27017.
82
- #
83
- # If you're connecting to a replica set, you'll need to use MongoReplicaSetClient.new instead.
84
- #
85
- # Once connected to a replica set, you can find out which nodes are primary, secondary, and
86
- # arbiters with the corresponding accessors: MongoClient#primary, MongoClient#secondaries, and
87
- # MongoClient#arbiters. This is useful if your application needs to connect manually to nodes other
88
- # than the primary.
89
- #
90
- # @overload initialize(host, port, opts={})
91
- # @param [String] host hostname for the target MongoDB server.
92
- # @param [Integer] port specify a port number here if only one host is being specified.
93
- # @param [Hash] opts hash of optional settings and configuration values.
94
- #
95
- # @option opts [String, Integer, Symbol] :w (1) Set default number of nodes to which a write
96
- # should be acknowledged.
97
- # @option opts [Integer] :wtimeout (nil) Set replica set acknowledgement timeout.
98
- # @option opts [Boolean] :j (false) If true, block until write operations have been committed
99
- # to the journal. Cannot be used in combination with 'fsync'. Prior to MongoDB 2.6 this option was
100
- # ignored if the server was running without journaling. Starting with MongoDB 2.6, write operations will
101
- # fail with an exception if this option is used when the server is running without journaling.
102
- # @option opts [Boolean] :fsync (false) If true, and the server is running without journaling, blocks until
103
- # the server has synced all data files to disk. If the server is running with journaling, this acts the same as
104
- # the 'j' option, blocking until write operations have been committed to the journal.
105
- # Cannot be used in combination with 'j'.
106
- #
107
- # Notes about Write-Concern Options:
108
- # Write concern options are propagated to objects instantiated from this MongoClient.
109
- # These defaults can be overridden upon instantiation of any object by explicitly setting an options hash
110
- # on initialization.
111
- #
112
- # @option opts [Boolean] :ssl (false) If true, create the connection to the server using SSL.
113
- # @option opts [String] :ssl_cert (nil) The certificate file used to identify the local connection against MongoDB.
114
- # @option opts [String] :ssl_key (nil) The private keyfile used to identify the local connection against MongoDB.
115
- # Note that even if the key is stored in the same file as the certificate, both need to be explicitly specified.
116
- # @option opts [String] :ssl_key_pass_phrase (nil) A passphrase for the private key.
117
- # @option opts [Boolean] :ssl_verify (nil) Specifies whether or not peer certification validation should occur.
118
- # @option opts [String] :ssl_ca_cert (nil) The ca_certs file contains a set of concatenated "certification authority"
119
- # certificates, which are used to validate certificates passed from the other end of the connection.
120
- # Required for :ssl_verify.
121
- # @option opts [Boolean] :slave_ok (false) Must be set to +true+ when connecting
122
- # to a single, slave node.
123
- # @option opts [Logger, #debug] :logger (nil) A Logger instance for debugging driver ops. Note that
124
- # logging negatively impacts performance; therefore, it should not be used for high-performance apps.
125
- # @option opts [Integer] :pool_size (1) The maximum number of socket self.connections allowed per
126
- # connection pool. Note: this setting is relevant only for multi-threaded applications.
127
- # @option opts [Float] :pool_timeout (5.0) When all of the self.connections a pool are checked out,
128
- # this is the number of seconds to wait for a new connection to be released before throwing an exception.
129
- # Note: this setting is relevant only for multi-threaded applications.
130
- # @option opts [Float] :op_timeout (DEFAULT_OP_TIMEOUT) The number of seconds to wait for a read operation to time out.
131
- # Set to DEFAULT_OP_TIMEOUT (20) by default. A value of nil may be specified explicitly.
132
- # @option opts [Float] :connect_timeout (nil) The number of seconds to wait before timing out a
133
- # connection attempt.
134
- #
135
- # @example localhost, 27017 (or <code>ENV["MONGODB_URI"]</code> if available)
136
- # MongoClient.new
137
- #
138
- # @example localhost, 27017
139
- # MongoClient.new("localhost")
140
- #
141
- # @example localhost, 3000, max 5 self.connections, with max 5 seconds of wait time.
142
- # MongoClient.new("localhost", 3000, :pool_size => 5, :pool_timeout => 5)
143
- #
144
- # @example localhost, 3000, where this node may be a slave
145
- # MongoClient.new("localhost", 3000, :slave_ok => true)
146
- #
147
- # @example Unix Domain Socket
148
- # MongoClient.new("/var/run/mongodb.sock")
149
- #
150
- # @see http://api.mongodb.org/ruby/current/file.REPLICA_SETS.html Replica sets in Ruby
151
- #
152
- # @raise [ReplicaSetConnectionError] This is raised if a replica set name is specified and the
153
- # driver fails to connect to a replica set with that name.
154
- #
155
- # @raise [MongoArgumentError] If called with no arguments and <code>ENV["MONGODB_URI"]</code> implies a replica set.
156
- def initialize(*args)
157
- opts = args.last.is_a?(Hash) ? args.pop : {}
158
- @host, @port = parse_init(args[0], args[1], opts)
159
-
160
- # Lock for request ids.
161
- @id_lock = Mutex.new
162
-
163
- # Connection pool for primary node
164
- @primary = nil
165
- @primary_pool = nil
166
- @mongos = false
167
-
168
- # Not set for direct connection
169
- @tag_sets = []
170
- @acceptable_latency = 15
171
-
172
- @max_bson_size = nil
173
- @max_message_size = nil
174
- @max_wire_version = nil
175
- @min_wire_version = nil
176
- @max_write_batch_size = nil
177
-
178
- check_opts(opts)
179
- setup(opts.dup)
180
- end
181
-
182
- # DEPRECATED
183
- #
184
- # Initialize a connection to a MongoDB replica set using an array of seed nodes.
185
- #
186
- # The seed nodes specified will be used on the initial connection to the replica set, but note
187
- # that this list of nodes will be replaced by the list of canonical nodes returned by running the
188
- # is_master command on the replica set.
189
- #
190
- # @param nodes [Array] An array of arrays, each of which specifies a host and port.
191
- # @param opts [Hash] Any of the available options that can be passed to MongoClient.new.
192
- #
193
- # @option opts [String] :rs_name (nil) The name of the replica set to connect to. An exception will be
194
- # raised if unable to connect to a replica set with this name.
195
- # @option opts [Boolean] :read_secondary (false) When true, this connection object will pick a random slave
196
- # to send reads to.
197
- #
198
- # @example
199
- # Mongo::MongoClient.multi([["db1.example.com", 27017], ["db2.example.com", 27017]])
200
- #
201
- # @example This connection will read from a random secondary node.
202
- # Mongo::MongoClient.multi([["db1.example.com", 27017], ["db2.example.com", 27017], ["db3.example.com", 27017]],
203
- # :read_secondary => true)
204
- #
205
- # @return [Mongo::MongoClient]
206
- #
207
- # @deprecated
208
- def self.multi(nodes, opts={})
209
- warn 'MongoClient.multi is now deprecated and will be removed in v2.0. Please use MongoReplicaSetClient.new instead.'
210
- MongoReplicaSetClient.new(nodes, opts)
211
- end
212
-
213
- # Initialize a connection to MongoDB using the MongoDB URI spec.
214
- #
215
- # Since MongoClient.new cannot be used with any <code>ENV["MONGODB_URI"]</code> that has multiple hosts (implying a replicaset),
216
- # you may use this when the type of your connection varies by environment and should be determined solely from <code>ENV["MONGODB_URI"]</code>.
217
- #
218
- # @param uri [String]
219
- # A string of the format mongodb://[username:password@]host1[:port1][,host2[:port2],...[,hostN[:portN]]][/database]
220
- #
221
- # @param [Hash] extra_opts Any of the options available for MongoClient.new
222
- #
223
- # @return [Mongo::MongoClient, Mongo::MongoReplicaSetClient]
224
- def self.from_uri(uri = ENV['MONGODB_URI'], extra_opts={})
225
- parser = URIParser.new(uri)
226
- parser.connection(extra_opts)
227
- end
228
-
229
- # The host name used for this connection.
230
- #
231
- # @return [String]
232
- def host
233
- @primary_pool.host
234
- end
235
-
236
- # The port used for this connection.
237
- #
238
- # @return [Integer]
239
- def port
240
- @primary_pool.port
241
- end
242
-
243
- def host_port
244
- [@host, @port]
245
- end
246
-
247
- # Flush all pending writes to datafiles.
248
- #
249
- # @return [BSON::OrderedHash] the command response
250
- def lock!
251
- cmd = BSON::OrderedHash.new
252
- cmd[:fsync] = 1
253
- cmd[:lock] = true
254
- self['admin'].command(cmd)
255
- end
256
-
257
- # Is this database locked against writes?
258
- #
259
- # @return [Boolean]
260
- def locked?
261
- [1, true].include? self['admin']['$cmd.sys.inprog'].find_one['fsyncLock']
262
- end
263
-
264
- # Unlock a previously fsync-locked mongod process.
265
- #
266
- # @return [BSON::OrderedHash] command response
267
- def unlock!
268
- self['admin']['$cmd.sys.unlock'].find_one
269
- end
270
-
271
- # Return a hash with all database names
272
- # and their respective sizes on disk.
273
- #
274
- # @return [Hash]
275
- def database_info
276
- doc = self['admin'].command({:listDatabases => 1})
277
- doc['databases'].inject({}) do |info, db|
278
- info[db['name']] = db['sizeOnDisk'].to_i
279
- info
280
- end
281
- end
282
-
283
- # Return an array of database names.
284
- #
285
- # @return [Array]
286
- def database_names
287
- database_info.keys
288
- end
289
-
290
- # Return a database with the given name.
291
- # See DB#new for valid options hash parameters.
292
- #
293
- # @param name [String] The name of the database.
294
- # @param opts [Hash] A hash of options to be passed to the DB constructor.
295
- #
296
- # @return [DB] The DB instance.
297
- def db(name = nil, opts = {})
298
- DB.new(name || @db_name || DEFAULT_DB_NAME, self, opts)
299
- end
300
-
301
- # Shortcut for returning a database. Use MongoClient#db to accept options.
302
- #
303
- # @param name [String] The name of the database.
304
- #
305
- # @return [DB] The DB instance.
306
- def [](name)
307
- DB.new(name, self)
308
- end
309
-
310
- def refresh; end
311
-
312
- def pinned_pool
313
- @primary_pool
314
- end
315
-
316
- def pin_pool(pool, read_prefs); end
317
-
318
- def unpin_pool; end
319
-
320
- # Drop a database.
321
- #
322
- # @param database [String] name of an existing database.
323
- def drop_database(database)
324
- self[database].command(:dropDatabase => 1)
325
- end
326
-
327
- # Copy the database +from+ to +to+ on localhost. The +from+ database is
328
- # assumed to be on localhost, but an alternate host can be specified.
329
- #
330
- # @param from [String] name of the database to copy from.
331
- # @param to [String] name of the database to copy to.
332
- # @param from_host [String] host of the 'from' database.
333
- # @param username [String] username (applies to 'from' db)
334
- # @param password [String] password (applies to 'from' db)
335
- #
336
- # @note This command only supports the MONGODB-CR authentication mechanism.
337
- def copy_database(
338
- from,
339
- to,
340
- from_host = DEFAULT_HOST,
341
- username = nil,
342
- password = nil,
343
- mechanism = 'SCRAM-SHA-1'
344
- )
345
- if wire_version_feature?(MONGODB_3_0) && mechanism == 'SCRAM-SHA-1'
346
- copy_db_scram(username, password, from_host, from, to)
347
- else
348
- copy_db_mongodb_cr(username, password, from_host, from, to)
349
- end
350
- end
351
-
352
- # Checks if a server is alive. This command will return immediately
353
- # even if the server is in a lock.
354
- #
355
- # @return [Hash]
356
- def ping
357
- self['admin'].command({:ping => 1})
358
- end
359
-
360
- # Get the build information for the current connection.
361
- #
362
- # @return [Hash]
363
- def server_info
364
- self['admin'].command({:buildinfo => 1})
365
- end
366
-
367
- # Get the build version of the current server.
368
- #
369
- # @return [Mongo::ServerVersion]
370
- # object allowing easy comparability of version.
371
- def server_version
372
- ServerVersion.new(server_info['version'])
373
- end
374
-
375
- # Is it okay to connect to a slave?
376
- #
377
- # @return [Boolean]
378
- def slave_ok?
379
- @slave_ok
380
- end
381
-
382
- def mongos?
383
- @mongos
384
- end
385
-
386
- # Create a new socket and attempt to connect to master.
387
- # If successful, sets host and port to master and returns the socket.
388
- #
389
- # If connecting to a replica set, this method will replace the
390
- # initially-provided seed list with any nodes known to the set.
391
- #
392
- # @raise [ConnectionFailure] if unable to connect to any host or port.
393
- def connect
394
- close
395
- config = check_is_master(host_port)
396
- if config
397
- if config['ismaster'] == 1 || config['ismaster'] == true
398
- @read_primary = true
399
- elsif @slave_ok
400
- @read_primary = false
401
- end
402
-
403
- if config.has_key?('msg') && config['msg'] == 'isdbgrid'
404
- @mongos = true
405
- end
406
-
407
- @max_bson_size = config['maxBsonObjectSize']
408
- @max_message_size = config['maxMessageSizeBytes']
409
- @max_wire_version = config['maxWireVersion']
410
- @min_wire_version = config['minWireVersion']
411
- @max_write_batch_size = config['maxWriteBatchSize']
412
- check_wire_version_in_range
413
- set_primary(host_port)
414
- end
415
-
416
- unless connected?
417
- raise ConnectionFailure,
418
- "Failed to connect to a master node at #{host_port.join(":")}"
419
- end
420
-
421
- true
422
- end
423
- alias :reconnect :connect
424
-
425
- # It's possible that we defined connected as all nodes being connected???
426
- # NOTE: Do check if this needs to be more stringent.
427
- # Probably not since if any node raises a connection failure, all nodes will be closed.
428
- def connected?
429
- !!(@primary_pool && !@primary_pool.closed?)
430
- end
431
-
432
- # Determine if the connection is active. In a normal case the *server_info* operation
433
- # will be performed without issues, but if the connection was dropped by the server or
434
- # for some reason the sockets are unsynchronized, a ConnectionFailure will be raised and
435
- # the return will be false.
436
- #
437
- # @return [Boolean]
438
- def active?
439
- return false unless connected?
440
-
441
- ping
442
- true
443
-
444
- rescue ConnectionFailure, OperationTimeout
445
- false
446
- end
447
-
448
- # Determine whether we're reading from a primary node. If false,
449
- # this connection connects to a secondary node and @slave_ok is true.
450
- #
451
- # @return [Boolean]
452
- def read_primary?
453
- @read_primary
454
- end
455
- alias :primary? :read_primary?
456
-
457
- # The socket pool that this connection reads from.
458
- #
459
- # @return [Mongo::Pool]
460
- def read_pool
461
- @primary_pool
462
- end
463
-
464
- # Close the connection to the database.
465
- def close
466
- @primary_pool.close if @primary_pool
467
- @primary_pool = nil
468
- @primary = nil
469
- end
470
-
471
- # Returns the maximum BSON object size as returned by the core server.
472
- # Use the 4MB default when the server doesn't report this.
473
- #
474
- # @return [Integer]
475
- def max_bson_size
476
- @max_bson_size || DEFAULT_MAX_BSON_SIZE
477
- end
478
-
479
- def max_message_size
480
- @max_message_size || max_bson_size * MESSAGE_SIZE_FACTOR
481
- end
482
-
483
- def max_wire_version
484
- @max_wire_version || 0
485
- end
486
-
487
- def min_wire_version
488
- @min_wire_version || 0
489
- end
490
-
491
- def max_write_batch_size
492
- @max_write_batch_size || DEFAULT_MAX_WRITE_BATCH_SIZE
493
- end
494
-
495
- def wire_version_feature?(feature)
496
- min_wire_version <= feature && feature <= max_wire_version
497
- end
498
-
499
- def primary_wire_version_feature?(feature)
500
- min_wire_version <= feature && feature <= max_wire_version
501
- end
502
-
503
- def use_write_command?(write_concern)
504
- write_concern[:w] != 0 && primary_wire_version_feature?(Mongo::MongoClient::BATCH_COMMANDS)
505
- end
506
-
507
- # Checkout a socket for reading (i.e., a secondary node).
508
- # Note: this is overridden in MongoReplicaSetClient.
509
- def checkout_reader(read_preference)
510
- connect unless connected?
511
- @primary_pool.checkout
512
- end
513
-
514
- # Checkout a socket for writing (i.e., a primary node).
515
- # Note: this is overridden in MongoReplicaSetClient.
516
- def checkout_writer
517
- connect unless connected?
518
- @primary_pool.checkout
519
- end
520
-
521
- # Check a socket back into its pool.
522
- # Note: this is overridden in MongoReplicaSetClient.
523
- def checkin(socket)
524
- if @primary_pool && socket && socket.pool
525
- socket.checkin
526
- end
527
- end
528
-
529
- # Internal method for checking isMaster() on a given node.
530
- #
531
- # @param node [Array] Port and host for the target node
532
- # @return [Hash] Response from isMaster()
533
- #
534
- # @private
535
- def check_is_master(node)
536
- begin
537
- host, port = *node
538
- config = nil
539
- socket = @socket_class.new(host, port, @op_timeout, @connect_timeout, @socket_opts)
540
- if @connect_timeout
541
- Timeout::timeout(@connect_timeout, OperationTimeout) do
542
- config = self['admin'].command({:isMaster => 1}, :socket => socket)
543
- end
544
- else
545
- config = self['admin'].command({:isMaster => 1}, :socket => socket)
546
- end
547
- rescue OperationFailure, SocketError, SystemCallError, IOError
548
- close
549
- ensure
550
- socket.close unless socket.nil? || socket.closed?
551
- end
552
- config
553
- end
554
-
555
- protected
556
-
557
- def valid_opts
558
- GENERIC_OPTS +
559
- CLIENT_ONLY_OPTS +
560
- POOL_OPTS +
561
- READ_PREFERENCE_OPTS +
562
- WRITE_CONCERN_OPTS +
563
- TIMEOUT_OPTS +
564
- SSL_OPTS
565
- end
566
-
567
- def check_opts(opts)
568
- bad_opts = opts.keys.reject { |opt| valid_opts.include?(opt) }
569
-
570
- unless bad_opts.empty?
571
- bad_opts.each {|opt| warn "#{opt} is not a valid option for #{self.class}"}
572
- end
573
- end
574
-
575
- # Parse option hash
576
- def setup(opts)
577
- @slave_ok = opts.delete(:slave_ok)
578
- @ssl = opts.delete(:ssl)
579
- @unix = @host ? @host.end_with?('.sock') : false
580
-
581
- # if ssl options are present, but ssl is nil/false raise for misconfig
582
- ssl_opts = opts.keys.select { |k| k.to_s.start_with?('ssl') }
583
- if ssl_opts.size > 0 && !@ssl
584
- raise MongoArgumentError, "SSL has not been enabled (:ssl=false) " +
585
- "but the following SSL related options were " +
586
- "specified: #{ssl_opts.join(', ')}"
587
- end
588
-
589
- @socket_opts = {}
590
- if @ssl
591
- # construct ssl socket opts
592
- @socket_opts[:key] = opts.delete(:ssl_key)
593
- @socket_opts[:cert] = opts.delete(:ssl_cert)
594
- @socket_opts[:verify] = opts.delete(:ssl_verify)
595
- @socket_opts[:ca_cert] = opts.delete(:ssl_ca_cert)
596
- @socket_opts[:key_pass_phrase] = opts.delete(:ssl_key_pass_phrase)
597
-
598
- # verify peer requires ca_cert, raise if only one is present
599
- if @socket_opts[:verify] && !@socket_opts[:ca_cert]
600
- raise MongoArgumentError,
601
- 'If :ssl_verify_mode has been specified, then you must include ' +
602
- ':ssl_ca_cert in order to perform server validation.'
603
- end
604
-
605
- # if we have a keyfile passphrase but no key file, raise
606
- if @socket_opts[:key_pass_phrase] && !@socket_opts[:key]
607
- raise MongoArgumentError,
608
- 'If :ssl_key_pass_phrase has been specified, then you must include ' +
609
- ':ssl_key, the passphrase-protected keyfile.'
610
- end
611
-
612
- @socket_class = Mongo::SSLSocket
613
- elsif @unix
614
- @socket_class = Mongo::UNIXSocket
615
- else
616
- @socket_class = Mongo::TCPSocket
617
- end
618
-
619
- @db_name = opts.delete(:db_name)
620
- @auths = opts.delete(:auths) || Set.new
621
-
622
- # Pool size and timeout.
623
- @pool_size = opts.delete(:pool_size) || 1
624
- if opts[:timeout]
625
- warn 'The :timeout option has been deprecated ' +
626
- 'and will be removed in the 2.0 release. ' +
627
- 'Use :pool_timeout instead.'
628
- end
629
- @pool_timeout = opts.delete(:pool_timeout) || opts.delete(:timeout) || 5.0
630
-
631
- # Timeout on socket read operation.
632
- @op_timeout = opts.key?(:op_timeout) ? opts.delete(:op_timeout) : DEFAULT_OP_TIMEOUT
633
-
634
- # Timeout on socket connect.
635
- @connect_timeout = opts.delete(:connect_timeout) || 30
636
-
637
- @logger = opts.delete(:logger)
638
- if @logger
639
- write_logging_startup_message
640
- end
641
-
642
- # Determine read preference
643
- if defined?(@slave_ok) && (@slave_ok) || defined?(@read_secondary) && @read_secondary
644
- @read = :secondary_preferred
645
- else
646
- @read = opts.delete(:read) || :primary
647
- end
648
- Mongo::ReadPreference::validate(@read)
649
-
650
- @tag_sets = opts.delete(:tag_sets) || []
651
- @acceptable_latency = opts.delete(:secondary_acceptable_latency_ms) || 15
652
-
653
- # Connection level write concern options.
654
- @write_concern = get_write_concern(opts)
655
-
656
- connect if opts.fetch(:connect, true)
657
- end
658
-
659
- private
660
-
661
- # Parses client initialization info from MONGODB_URI env variable
662
- def parse_init(host, port, opts)
663
- if host.nil? && port.nil? && ENV.has_key?('MONGODB_URI')
664
- parser = URIParser.new(ENV['MONGODB_URI'])
665
- if parser.replicaset?
666
- raise MongoArgumentError,
667
- 'ENV[\'MONGODB_URI\'] implies a replica set.'
668
- end
669
- opts.merge!(parser.connection_options)
670
- [parser.host, parser.port]
671
- else
672
- host = host[1...-1] if host && host[0,1] == '[' # ipv6 support
673
- [host || DEFAULT_HOST, port || DEFAULT_PORT]
674
- end
675
- end
676
-
677
- # Set the specified node as primary
678
- def set_primary(node)
679
- host, port = *node
680
- @primary = [host, port]
681
- @primary_pool = Pool.new(self, host, port, :size => @pool_size, :timeout => @pool_timeout)
682
- end
683
-
684
- # calculate wire version in range
685
- def check_wire_version_in_range
686
- unless MIN_WIRE_VERSION <= max_wire_version &&
687
- MAX_WIRE_VERSION >= min_wire_version
688
- close
689
- raise ConnectionFailure,
690
- "Client wire-version range #{MIN_WIRE_VERSION} to " +
691
- "#{MAX_WIRE_VERSION} does not support server range " +
692
- "#{min_wire_version} to #{max_wire_version}, please update " +
693
- "clients or servers"
694
- end
695
- end
696
- end
697
- end