File: write_buffer.rb

package info (click to toggle)
gitlab 17.6.5-19
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 629,368 kB
  • sloc: ruby: 1,915,304; javascript: 557,307; sql: 60,639; xml: 6,509; sh: 4,567; makefile: 1,239; python: 406
file content (29 lines) | stat: -rw-r--r-- 697 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
# frozen_string_literal: true

module ClickHouse
  module WriteBuffer
    BUFFER_KEY_PREFIX = 'clickhouse_write_buffer_'

    class << self
      def add(table_name, event_hash)
        Gitlab::Redis::SharedState.with do |redis|
          redis.rpush(buffer_key(table_name), event_hash.to_json)
        end
      end

      def pop(table_name, limit)
        Gitlab::Redis::SharedState.with do |redis|
          Array.wrap(redis.lpop(buffer_key(table_name), limit)).map do |hash|
            Gitlab::Json.parse(hash, symbolize_names: true)
          end
        end
      end

      private

      def buffer_key(table_name)
        "#{BUFFER_KEY_PREFIX}#{table_name}"
      end
    end
  end
end