riemann-client 1.0.0 → 1.1.0
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +4 -4
- data/.github/dependabot.yml +11 -0
- data/.github/workflows/ci.yml +19 -5
- data/.github/workflows/codeql-analysis.yml +72 -0
- data/.rspec +2 -0
- data/.rubocop.yml +20 -0
- data/CHANGELOG.md +28 -2
- data/Gemfile +2 -0
- data/README.markdown +19 -10
- data/Rakefile +5 -2
- data/SECURITY.md +42 -0
- data/lib/riemann/attribute.rb +2 -0
- data/lib/riemann/auto_state.rb +9 -3
- data/lib/riemann/client/ssl_socket.rb +22 -21
- data/lib/riemann/client/tcp.rb +38 -38
- data/lib/riemann/client/tcp_socket.rb +66 -51
- data/lib/riemann/client/udp.rb +8 -4
- data/lib/riemann/client.rb +97 -78
- data/lib/riemann/event.rb +45 -51
- data/lib/riemann/message.rb +2 -0
- data/lib/riemann/metric_thread.rb +59 -54
- data/lib/riemann/query.rb +4 -2
- data/lib/riemann/state.rb +8 -8
- data/lib/riemann/version.rb +3 -1
- data/lib/riemann.rb +2 -3
- data/riemann-client.gemspec +10 -5
- data/spec/client_spec.rb +66 -0
- data/spec/shared_examples.rb +531 -0
- data/spec/spec_helper.rb +38 -0
- metadata +47 -10
- data/spec/client.rb +0 -384
checksums.yaml
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
---
|
2
2
|
SHA256:
|
3
|
-
metadata.gz:
|
4
|
-
data.tar.gz:
|
3
|
+
metadata.gz: 33657c1dcec6dce903bd90154f17f3a31174b6e0fa146609941dc36960c2b6e2
|
4
|
+
data.tar.gz: 372a73585a2046e5af3a16ddd84254af64e6f6d17c6911183cf6c408eb22ac07
|
5
5
|
SHA512:
|
6
|
-
metadata.gz:
|
7
|
-
data.tar.gz:
|
6
|
+
metadata.gz: 694404b98fe75bb304f8d8da0903eb128d2600ecd83bc3b6c3d4e2b41eef2695a82e6df00c0f9c3a59115837e58314e6cdb88d56c1b10c959bd912d69ace479c
|
7
|
+
data.tar.gz: 874fdd45638f808e1bf9641616fe5342c01c3a51245f9d66542a9aec60751a093462e70c123e027a463dcedfcb608dfc98087ae4ad189fd02aa97210b8e90fdd
|
@@ -0,0 +1,11 @@
|
|
1
|
+
# To get started with Dependabot version updates, you'll need to specify which
|
2
|
+
# package ecosystems to update and where the package manifests are located.
|
3
|
+
# Please see the documentation for all configuration options:
|
4
|
+
# https://docs.github.com/github/administering-a-repository/configuration-options-for-dependency-updates
|
5
|
+
|
6
|
+
version: 2
|
7
|
+
updates:
|
8
|
+
- package-ecosystem: "bundler" # See documentation for possible values
|
9
|
+
directory: "/" # Location of package manifests
|
10
|
+
schedule:
|
11
|
+
interval: "daily"
|
data/.github/workflows/ci.yml
CHANGED
@@ -10,16 +10,30 @@ on:
|
|
10
10
|
- main
|
11
11
|
|
12
12
|
jobs:
|
13
|
+
lint:
|
14
|
+
runs-on: ubuntu-latest
|
15
|
+
steps:
|
16
|
+
- uses: actions/checkout@v3
|
17
|
+
- name: Setup ruby
|
18
|
+
uses: ruby/setup-ruby@v1
|
19
|
+
with:
|
20
|
+
ruby-version: '2.7'
|
21
|
+
bundler-cache: true
|
22
|
+
- name: Run rubocop
|
23
|
+
run: bundle exec rubocop
|
13
24
|
test:
|
25
|
+
needs: lint
|
14
26
|
runs-on: ubuntu-latest
|
15
27
|
strategy:
|
16
28
|
matrix:
|
17
29
|
ruby-version:
|
18
|
-
- 2.
|
19
|
-
-
|
20
|
-
- 3.
|
30
|
+
- '2.6'
|
31
|
+
- '2.7'
|
32
|
+
- '3.0'
|
33
|
+
- '3.1'
|
34
|
+
- '3.2'
|
21
35
|
steps:
|
22
|
-
- uses: actions/checkout@
|
36
|
+
- uses: actions/checkout@v3
|
23
37
|
- name: Setup Ruby
|
24
38
|
uses: ruby/setup-ruby@v1
|
25
39
|
with:
|
@@ -42,4 +56,4 @@ jobs:
|
|
42
56
|
|
43
57
|
while ! nc -z localhost 5555; do sleep 1; done
|
44
58
|
- name: Run the test suite
|
45
|
-
run: bundle exec
|
59
|
+
run: bundle exec rspec
|
@@ -0,0 +1,72 @@
|
|
1
|
+
# For most projects, this workflow file will not need changing; you simply need
|
2
|
+
# to commit it to your repository.
|
3
|
+
#
|
4
|
+
# You may wish to alter this file to override the set of languages analyzed,
|
5
|
+
# or to provide custom queries or build logic.
|
6
|
+
#
|
7
|
+
# ******** NOTE ********
|
8
|
+
# We have attempted to detect the languages in your repository. Please check
|
9
|
+
# the `language` matrix defined below to confirm you have the correct set of
|
10
|
+
# supported CodeQL languages.
|
11
|
+
#
|
12
|
+
name: "CodeQL"
|
13
|
+
|
14
|
+
on:
|
15
|
+
push:
|
16
|
+
branches: [ "main" ]
|
17
|
+
pull_request:
|
18
|
+
# The branches below must be a subset of the branches above
|
19
|
+
branches: [ "main" ]
|
20
|
+
schedule:
|
21
|
+
- cron: '33 17 * * 6'
|
22
|
+
|
23
|
+
jobs:
|
24
|
+
analyze:
|
25
|
+
name: Analyze
|
26
|
+
runs-on: ubuntu-latest
|
27
|
+
permissions:
|
28
|
+
actions: read
|
29
|
+
contents: read
|
30
|
+
security-events: write
|
31
|
+
|
32
|
+
strategy:
|
33
|
+
fail-fast: false
|
34
|
+
matrix:
|
35
|
+
language: [ 'ruby' ]
|
36
|
+
# CodeQL supports [ 'cpp', 'csharp', 'go', 'java', 'javascript', 'python', 'ruby' ]
|
37
|
+
# Learn more about CodeQL language support at https://aka.ms/codeql-docs/language-support
|
38
|
+
|
39
|
+
steps:
|
40
|
+
- name: Checkout repository
|
41
|
+
uses: actions/checkout@v3
|
42
|
+
|
43
|
+
# Initializes the CodeQL tools for scanning.
|
44
|
+
- name: Initialize CodeQL
|
45
|
+
uses: github/codeql-action/init@v2
|
46
|
+
with:
|
47
|
+
languages: ${{ matrix.language }}
|
48
|
+
# If you wish to specify custom queries, you can do so here or in a config file.
|
49
|
+
# By default, queries listed here will override any specified in a config file.
|
50
|
+
# Prefix the list here with "+" to use these queries and those in the config file.
|
51
|
+
|
52
|
+
# Details on CodeQL's query packs refer to : https://docs.github.com/en/code-security/code-scanning/automatically-scanning-your-code-for-vulnerabilities-and-errors/configuring-code-scanning#using-queries-in-ql-packs
|
53
|
+
# queries: security-extended,security-and-quality
|
54
|
+
|
55
|
+
|
56
|
+
# Autobuild attempts to build any compiled languages (C/C++, C#, or Java).
|
57
|
+
# If this step fails, then you should remove it and run the build manually (see below)
|
58
|
+
- name: Autobuild
|
59
|
+
uses: github/codeql-action/autobuild@v2
|
60
|
+
|
61
|
+
# ℹ️ Command-line programs to run using the OS shell.
|
62
|
+
# 📚 See https://docs.github.com/en/actions/using-workflows/workflow-syntax-for-github-actions#jobsjob_idstepsrun
|
63
|
+
|
64
|
+
# If the Autobuild fails above, remove it and uncomment the following three lines.
|
65
|
+
# modify them (or add more) to build your code if your project, please refer to the EXAMPLE below for guidance.
|
66
|
+
|
67
|
+
# - run: |
|
68
|
+
# echo "Run, Build Application using script"
|
69
|
+
# ./location_of_script_within_repo/buildscript.sh
|
70
|
+
|
71
|
+
- name: Perform CodeQL Analysis
|
72
|
+
uses: github/codeql-action/analyze@v2
|
data/.rspec
ADDED
data/.rubocop.yml
ADDED
@@ -0,0 +1,20 @@
|
|
1
|
+
---
|
2
|
+
require:
|
3
|
+
- rubocop-rspec
|
4
|
+
Metrics/AbcSize:
|
5
|
+
Enabled: false
|
6
|
+
Metrics/BlockLength:
|
7
|
+
Enabled: false
|
8
|
+
Metrics/ClassLength:
|
9
|
+
Enabled: false
|
10
|
+
Metrics/CyclomaticComplexity:
|
11
|
+
Enabled: false
|
12
|
+
Metrics/MethodLength:
|
13
|
+
Enabled: false
|
14
|
+
Metrics/PerceivedComplexity:
|
15
|
+
Enabled: false
|
16
|
+
Naming/VariableNumber:
|
17
|
+
AllowedIdentifiers:
|
18
|
+
- TLSv1_2
|
19
|
+
Style/Documentation:
|
20
|
+
Enabled: false
|
data/CHANGELOG.md
CHANGED
@@ -1,8 +1,34 @@
|
|
1
1
|
# Changelog
|
2
2
|
|
3
|
-
## [1.0
|
3
|
+
## [v1.1.0](https://github.com/riemann/riemann-ruby-client/tree/v1.1.0) (2023-01-23)
|
4
4
|
|
5
|
-
[Full Changelog](https://github.com/riemann/riemann-ruby-client/compare/0.
|
5
|
+
[Full Changelog](https://github.com/riemann/riemann-ruby-client/compare/v1.0.1...v1.1.0)
|
6
|
+
|
7
|
+
**Implemented enhancements:**
|
8
|
+
|
9
|
+
- Add support for sending events in bulk [\#44](https://github.com/riemann/riemann-ruby-client/pull/44) ([smortex](https://github.com/smortex))
|
10
|
+
|
11
|
+
**Fixed bugs:**
|
12
|
+
|
13
|
+
- Fix UDP fallback to TCP on large messages [\#46](https://github.com/riemann/riemann-ruby-client/pull/46) ([smortex](https://github.com/smortex))
|
14
|
+
|
15
|
+
**Merged pull requests:**
|
16
|
+
|
17
|
+
- Modernize unit tests [\#45](https://github.com/riemann/riemann-ruby-client/pull/45) ([smortex](https://github.com/smortex))
|
18
|
+
- Switch from Bacon to RSpec [\#43](https://github.com/riemann/riemann-ruby-client/pull/43) ([smortex](https://github.com/smortex))
|
19
|
+
- Create codeql-analysis.yml [\#40](https://github.com/riemann/riemann-ruby-client/pull/40) ([jamtur01](https://github.com/jamtur01))
|
20
|
+
|
21
|
+
## [v1.0.1](https://github.com/riemann/riemann-ruby-client/tree/v1.0.1) (2022-06-25)
|
22
|
+
|
23
|
+
[Full Changelog](https://github.com/riemann/riemann-ruby-client/compare/v1.0.0...v1.0.1)
|
24
|
+
|
25
|
+
**Merged pull requests:**
|
26
|
+
|
27
|
+
- Setup Rubocop and lower required Ruby version [\#37](https://github.com/riemann/riemann-ruby-client/pull/37) ([smortex](https://github.com/smortex))
|
28
|
+
|
29
|
+
## [v1.0.0](https://github.com/riemann/riemann-ruby-client/tree/v1.0.0) (2022-06-16)
|
30
|
+
|
31
|
+
[Full Changelog](https://github.com/riemann/riemann-ruby-client/compare/0.2.6...v1.0.0)
|
6
32
|
|
7
33
|
**Implemented enhancements:**
|
8
34
|
|
data/Gemfile
CHANGED
data/README.markdown
CHANGED
@@ -1,12 +1,16 @@
|
|
1
|
-
|
2
|
-
==========
|
1
|
+
# Riemann Ruby Client
|
3
2
|
|
4
|
-
|
3
|
+
[![CI](https://github.com/riemann/riemann-ruby-client/actions/workflows/ci.yml/badge.svg)](https://github.com/riemann/riemann-ruby-client/actions/workflows/ci.yml)
|
5
4
|
|
6
|
-
|
7
|
-
===
|
5
|
+
## Installing
|
8
6
|
|
9
|
-
```
|
7
|
+
```shell
|
8
|
+
gem install riemann-client
|
9
|
+
```
|
10
|
+
|
11
|
+
## Use
|
12
|
+
|
13
|
+
```ruby
|
10
14
|
require 'riemann/client'
|
11
15
|
|
12
16
|
# Create a client. Host, port and timeout are optional.
|
@@ -38,8 +42,8 @@ c['host =~ "%.dc1" and (state = "critical" or state = "warning")']
|
|
38
42
|
|
39
43
|
```
|
40
44
|
|
41
|
-
Transports
|
42
|
-
|
45
|
+
## Transports
|
46
|
+
|
43
47
|
|
44
48
|
Riemann::Client sends small events over UDP by default, and uses TCP for
|
45
49
|
queries and large events. UDP sends are essentially "shouting into the void".
|
@@ -54,8 +58,7 @@ c.tcp["true"] # => [#<Event ... >, ...]
|
|
54
58
|
c.udp["true"] # => raise Riemann::Client::Unsupported
|
55
59
|
```
|
56
60
|
|
57
|
-
Client state management
|
58
|
-
=======================
|
61
|
+
## Client state management
|
59
62
|
|
60
63
|
Riemann::Client provides some classes to make managing state updates easier.
|
61
64
|
|
@@ -64,3 +67,9 @@ be used to flush an accumulated value to ustate at regular intervals.
|
|
64
67
|
|
65
68
|
Riemann::AutoState bundles a state and a client together. Any changes to the
|
66
69
|
AutoState automatically send the new state to the client.
|
70
|
+
|
71
|
+
## License
|
72
|
+
|
73
|
+
The MIT License
|
74
|
+
|
75
|
+
Copyright (c) 2011-2022 Kyle Kingsbury
|
data/Rakefile
CHANGED
@@ -1,12 +1,15 @@
|
|
1
|
+
# frozen_string_literal: true
|
2
|
+
|
1
3
|
require 'riemann'
|
2
4
|
|
3
|
-
require
|
5
|
+
require 'bundler/gem_tasks'
|
4
6
|
|
5
7
|
require 'github_changelog_generator/task'
|
6
8
|
|
7
9
|
GitHubChangelogGenerator::RakeTask.new :changelog do |config|
|
8
10
|
config.user = 'riemann'
|
9
11
|
config.project = 'riemann-ruby-client'
|
10
|
-
config.
|
12
|
+
config.exclude_labels = ['skip-changelog']
|
13
|
+
config.future_release = "v#{Riemann::VERSION}"
|
11
14
|
config.add_issues_wo_labels = false
|
12
15
|
end
|
data/SECURITY.md
ADDED
@@ -0,0 +1,42 @@
|
|
1
|
+
# Riemann Security and Disclosure Information
|
2
|
+
This page describes Riemann security and disclosure information.
|
3
|
+
|
4
|
+
## Supported Versions
|
5
|
+
|
6
|
+
The currently supported version of Riemann for security-patching purposes is always the latest version.
|
7
|
+
|
8
|
+
## Security Announcements
|
9
|
+
|
10
|
+
Will be made on the [Riemann mailing list](https://groups.google.com/g/riemann-users?pli=1).
|
11
|
+
|
12
|
+
## Report a Vulnerability
|
13
|
+
|
14
|
+
We're extremely grateful for security researchers and users that report vulnerabilities to Riemann. All reports are thoroughly investigated by the maintainers.
|
15
|
+
|
16
|
+
To make a report, you should email the private security@riemann.io list with the details.
|
17
|
+
|
18
|
+
## When Should I Report a Vulnerability?
|
19
|
+
|
20
|
+
* You think you discovered a potential security vulnerability in Riemann.
|
21
|
+
* You are unsure how a vulnerability affects Riemann.
|
22
|
+
* You think you discovered a vulnerability in another project that Riemann depends on
|
23
|
+
|
24
|
+
For projects with their own vulnerability reporting and disclosure process, please report it directly there.
|
25
|
+
|
26
|
+
## When Should I NOT Report a Vulnerability?
|
27
|
+
|
28
|
+
* You need help tuning Riemann components for security
|
29
|
+
* You need help applying security related updates
|
30
|
+
* Your issue is not security related
|
31
|
+
|
32
|
+
## Security Vulnerability Response
|
33
|
+
|
34
|
+
Each report is acknowledged and analyzed within 5 working days.
|
35
|
+
|
36
|
+
Any vulnerability information shared stays within Riemann project and will not be disseminated to other projects unless it is necessary to get the issue fixed.
|
37
|
+
|
38
|
+
As the security issue moves from triage, to identified fix, to release planning we will keep the reporter updated.
|
39
|
+
|
40
|
+
## Public Disclosure Timing
|
41
|
+
|
42
|
+
A public disclosure date is negotiated by the Riemann maintainers nd the bug submitter. We prefer to fully disclose the bug as soon as possible once a user mitigation is available. It is reasonable to delay disclosure when the bug or the fix is not yet fully understood, the solution is not well-tested, or for vendor coordination. The timeframe for disclosure is from immediate (especially if it's already publicly known) to a few weeks. For a vulnerability with a straightforward mitigation, we expect report date to disclosure date to be on the order of 7 days. The Riemann maintainers hold the final say when setting a disclosure date.
|
data/lib/riemann/attribute.rb
CHANGED
data/lib/riemann/auto_state.rb
CHANGED
@@ -1,3 +1,5 @@
|
|
1
|
+
# frozen_string_literal: true
|
2
|
+
|
1
3
|
module Riemann
|
2
4
|
class AutoState
|
3
5
|
# Binds together a state hash and a Client. Any change made here
|
@@ -40,7 +42,7 @@ module Riemann
|
|
40
42
|
# @state.state = 'heavy lifting b'
|
41
43
|
# ...
|
42
44
|
# end
|
43
|
-
|
45
|
+
|
44
46
|
def initialize(client = Client.new, state = {})
|
45
47
|
@client = client
|
46
48
|
@state = state
|
@@ -95,7 +97,11 @@ module Riemann
|
|
95
97
|
def once(opts)
|
96
98
|
o = @state.merge opts
|
97
99
|
o[:time] = Time.now.to_i
|
98
|
-
o[:tags] =
|
100
|
+
o[:tags] = begin
|
101
|
+
(o[:tags] | ['once'])
|
102
|
+
rescue StandardError
|
103
|
+
['once']
|
104
|
+
end
|
99
105
|
@client << o
|
100
106
|
end
|
101
107
|
|
@@ -111,7 +117,7 @@ module Riemann
|
|
111
117
|
def service=(service)
|
112
118
|
@state[:service] = service
|
113
119
|
flush
|
114
|
-
end
|
120
|
+
end
|
115
121
|
|
116
122
|
def service
|
117
123
|
@state[:service]
|
@@ -1,11 +1,12 @@
|
|
1
|
+
# frozen_string_literal: true
|
2
|
+
|
1
3
|
require 'openssl'
|
2
4
|
require_relative 'tcp_socket'
|
3
5
|
|
4
6
|
module Riemann
|
5
7
|
class Client
|
6
|
-
|
8
|
+
# Socket: A specialized socket that has been configure
|
7
9
|
class SSLSocket < TcpSocket
|
8
|
-
|
9
10
|
def initialize(options = {})
|
10
11
|
super(options)
|
11
12
|
@key_file = options[:key_file]
|
@@ -16,11 +17,11 @@ module Riemann
|
|
16
17
|
|
17
18
|
def ssl_context
|
18
19
|
@ssl_context ||= OpenSSL::SSL::SSLContext.new.tap do |ctx|
|
19
|
-
|
20
|
-
|
21
|
-
|
22
|
-
|
23
|
-
|
20
|
+
ctx.key = OpenSSL::PKey::RSA.new(File.read(@key_file))
|
21
|
+
ctx.cert = OpenSSL::X509::Certificate.new(File.read(@cert_file))
|
22
|
+
ctx.ca_file = @ca_file if @ca_file
|
23
|
+
ctx.ssl_version = :TLSv1_2
|
24
|
+
ctx.verify_mode = OpenSSL::SSL::VERIFY_PEER if @ssl_verify
|
24
25
|
end
|
25
26
|
end
|
26
27
|
|
@@ -30,7 +31,7 @@ module Riemann
|
|
30
31
|
#
|
31
32
|
# Return the ::Socket when it is connected, or raise an Error if no
|
32
33
|
# connection was possible.
|
33
|
-
def connect_nonblock(
|
34
|
+
def connect_nonblock(addr, timeout)
|
34
35
|
sock = super(addr, timeout)
|
35
36
|
ssl_socket = OpenSSL::SSL::SSLSocket.new(sock, ssl_context)
|
36
37
|
ssl_socket.sync = true
|
@@ -38,17 +39,17 @@ module Riemann
|
|
38
39
|
begin
|
39
40
|
ssl_socket.connect_nonblock
|
40
41
|
rescue IO::WaitReadable
|
41
|
-
|
42
|
-
retry
|
43
|
-
else
|
42
|
+
unless IO.select([ssl_socket], nil, nil, timeout)
|
44
43
|
raise Timeout, "Could not read from #{host}:#{port} in #{timeout} seconds"
|
45
44
|
end
|
45
|
+
|
46
|
+
retry
|
46
47
|
rescue IO::WaitWritable
|
47
|
-
|
48
|
-
retry
|
49
|
-
else
|
48
|
+
unless IO.select(nil, [ssl_socket], nil, timeout)
|
50
49
|
raise Timeout, "Could not write to #{host}:#{port} in #{timeout} seconds"
|
51
50
|
end
|
51
|
+
|
52
|
+
retry
|
52
53
|
end
|
53
54
|
ssl_socket
|
54
55
|
end
|
@@ -60,13 +61,13 @@ module Riemann
|
|
60
61
|
#
|
61
62
|
# Returns the bytes read
|
62
63
|
def readpartial(maxlen, outbuf = nil)
|
63
|
-
|
64
|
+
super(maxlen, outbuf)
|
64
65
|
rescue OpenSSL::SSL::SSLErrorWaitReadable
|
65
|
-
|
66
|
-
retry
|
67
|
-
else
|
66
|
+
unless wait_readable(read_timeout)
|
68
67
|
raise Timeout, "Could not read from #{host}:#{port} in #{read_timeout} seconds"
|
69
68
|
end
|
69
|
+
|
70
|
+
retry
|
70
71
|
end
|
71
72
|
|
72
73
|
# Internal: Write the given data to the socket
|
@@ -80,11 +81,11 @@ module Riemann
|
|
80
81
|
def write(buf)
|
81
82
|
super(buf)
|
82
83
|
rescue OpenSSL::SSL::SSLErrorWaitWritable
|
83
|
-
|
84
|
-
retry
|
85
|
-
else
|
84
|
+
unless wait_writable(write_timeout)
|
86
85
|
raise Timeout, "Could not write to #{host}:#{port} in #{write_timeout} seconds"
|
87
86
|
end
|
87
|
+
|
88
|
+
retry
|
88
89
|
end
|
89
90
|
end
|
90
91
|
end
|
data/lib/riemann/client/tcp.rb
CHANGED
@@ -1,3 +1,5 @@
|
|
1
|
+
# frozen_string_literal: true
|
2
|
+
|
1
3
|
require 'monitor'
|
2
4
|
require 'riemann/client/tcp_socket'
|
3
5
|
require 'riemann/client/ssl_socket'
|
@@ -5,12 +7,12 @@ require 'riemann/client/ssl_socket'
|
|
5
7
|
module Riemann
|
6
8
|
class Client
|
7
9
|
class TCP < Client
|
8
|
-
attr_accessor :host, :port
|
10
|
+
attr_accessor :host, :port
|
9
11
|
|
10
12
|
# Public: Set a socket factory -- an object responding
|
11
13
|
# to #call(options) that returns a Socket object
|
12
|
-
|
13
|
-
|
14
|
+
class << self
|
15
|
+
attr_writer :socket_factory
|
14
16
|
end
|
15
17
|
|
16
18
|
# Public: Return a socket factory
|
@@ -24,16 +26,14 @@ module Riemann
|
|
24
26
|
}
|
25
27
|
end
|
26
28
|
|
27
|
-
def initialize(options = {})
|
29
|
+
def initialize(options = {}) # rubocop:disable Lint/MissingSuper
|
28
30
|
@options = options
|
29
31
|
@locket = Monitor.new
|
30
32
|
end
|
31
33
|
|
32
34
|
def socket
|
33
35
|
@locket.synchronize do
|
34
|
-
if @pid && @pid != Process.pid
|
35
|
-
close
|
36
|
-
end
|
36
|
+
close if @pid && @pid != Process.pid
|
37
37
|
|
38
38
|
return @socket if connected?
|
39
39
|
|
@@ -58,32 +58,32 @@ module Riemann
|
|
58
58
|
end
|
59
59
|
|
60
60
|
# Read a message from a stream
|
61
|
-
def read_message(
|
62
|
-
|
63
|
-
|
64
|
-
|
65
|
-
str = s.read length
|
66
|
-
message = Riemann::Message.decode str
|
67
|
-
rescue => e
|
68
|
-
puts "Message was #{str.inspect}"
|
69
|
-
raise
|
70
|
-
end
|
61
|
+
def read_message(socket)
|
62
|
+
unless (buffer = socket.read(4)) && (buffer.size == 4)
|
63
|
+
raise InvalidResponse, 'unexpected EOF'
|
64
|
+
end
|
71
65
|
|
72
|
-
|
73
|
-
|
74
|
-
|
75
|
-
|
66
|
+
length = buffer.unpack1('N')
|
67
|
+
begin
|
68
|
+
str = socket.read length
|
69
|
+
message = Riemann::Message.decode str
|
70
|
+
rescue StandardError
|
71
|
+
puts "Message was #{str.inspect}"
|
72
|
+
raise
|
73
|
+
end
|
76
74
|
|
77
|
-
|
78
|
-
|
79
|
-
raise
|
75
|
+
unless message.ok
|
76
|
+
puts 'Failed'
|
77
|
+
raise ServerError, message.error
|
80
78
|
end
|
79
|
+
|
80
|
+
message
|
81
81
|
end
|
82
82
|
|
83
83
|
def send_recv(message)
|
84
|
-
with_connection do |
|
85
|
-
|
86
|
-
read_message(
|
84
|
+
with_connection do |socket|
|
85
|
+
socket.write(message.encode_with_length)
|
86
|
+
read_message(socket)
|
87
87
|
end
|
88
88
|
end
|
89
89
|
|
@@ -94,17 +94,17 @@ module Riemann
|
|
94
94
|
tries = 0
|
95
95
|
|
96
96
|
@locket.synchronize do
|
97
|
-
|
98
|
-
|
99
|
-
|
100
|
-
|
101
|
-
|
102
|
-
|
103
|
-
|
104
|
-
|
105
|
-
|
106
|
-
|
107
|
-
|
97
|
+
tries += 1
|
98
|
+
yield(socket)
|
99
|
+
rescue IOError, Errno::EPIPE, Errno::ECONNREFUSED, InvalidResponse, Timeout::Error,
|
100
|
+
Riemann::Client::TcpSocket::Error
|
101
|
+
close
|
102
|
+
raise if tries > 3
|
103
|
+
|
104
|
+
retry
|
105
|
+
rescue StandardError
|
106
|
+
close
|
107
|
+
raise
|
108
108
|
end
|
109
109
|
end
|
110
110
|
end
|