schked 1.1.1 → 1.1.2
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +4 -4
 - data/lib/schked/redis_locker.rb +13 -2
 - data/lib/schked/version.rb +1 -1
 - data/lib/schked/worker.rb +10 -7
 - metadata +2 -2
 
    
        checksums.yaml
    CHANGED
    
    | 
         @@ -1,7 +1,7 @@ 
     | 
|
| 
       1 
1 
     | 
    
         
             
            ---
         
     | 
| 
       2 
2 
     | 
    
         
             
            SHA256:
         
     | 
| 
       3 
     | 
    
         
            -
              metadata.gz:  
     | 
| 
       4 
     | 
    
         
            -
              data.tar.gz:  
     | 
| 
      
 3 
     | 
    
         
            +
              metadata.gz: fe971fe49c1279ffd4cbc39948faea2fb82fa58434aa90814e1c1cbe2ebd99e2
         
     | 
| 
      
 4 
     | 
    
         
            +
              data.tar.gz: 2b8800aa8128347a42e1f7fe2d73930293baae1ff258d6f1ddd24f21dff297c4
         
     | 
| 
       5 
5 
     | 
    
         
             
            SHA512:
         
     | 
| 
       6 
     | 
    
         
            -
              metadata.gz:  
     | 
| 
       7 
     | 
    
         
            -
              data.tar.gz:  
     | 
| 
      
 6 
     | 
    
         
            +
              metadata.gz: a8dcdca1dad927793b3c0f6356a1f6222a9e4081b20baca7df9391c93ea044f8f2f9bf7fa2db05c330616a0692597abbf8e2807a07d5c6f80bff0687bfd90911
         
     | 
| 
      
 7 
     | 
    
         
            +
              data.tar.gz: 370cab83d6e000cbc0cb2c19be884738315434d3dd42d0bf718c1b094194231b77f8356d2782f6a7c773b98aa64f3eedc6de39f3a34206f8a3aa381645f5012e
         
     | 
    
        data/lib/schked/redis_locker.rb
    CHANGED
    
    | 
         @@ -4,22 +4,30 @@ module Schked 
     | 
|
| 
       4 
4 
     | 
    
         
             
              class RedisLocker
         
     | 
| 
       5 
5 
     | 
    
         
             
                attr_reader :lock_manager,
         
     | 
| 
       6 
6 
     | 
    
         
             
                  :lock_id,
         
     | 
| 
       7 
     | 
    
         
            -
                  :lock_ttl
         
     | 
| 
      
 7 
     | 
    
         
            +
                  :lock_ttl,
         
     | 
| 
      
 8 
     | 
    
         
            +
                  :logger
         
     | 
| 
       8 
9 
     | 
    
         | 
| 
       9 
10 
     | 
    
         
             
                LOCK_KEY = "schked:redis_locker"
         
     | 
| 
       10 
11 
     | 
    
         
             
                LOCK_TTL = 60_000 # ms
         
     | 
| 
       11 
12 
     | 
    
         | 
| 
       12 
     | 
    
         
            -
                def initialize(redis_servers, lock_ttl: LOCK_TTL)
         
     | 
| 
      
 13 
     | 
    
         
            +
                def initialize(redis_servers, lock_ttl: LOCK_TTL, logger: Logger.new($stdout))
         
     | 
| 
       13 
14 
     | 
    
         
             
                  @lock_manager = Redlock::Client.new(redis_servers, retry_count: 0)
         
     | 
| 
       14 
15 
     | 
    
         
             
                  @lock_ttl = lock_ttl
         
     | 
| 
      
 16 
     | 
    
         
            +
                  @logger = logger
         
     | 
| 
       15 
17 
     | 
    
         
             
                end
         
     | 
| 
       16 
18 
     | 
    
         | 
| 
       17 
19 
     | 
    
         
             
                def lock
         
     | 
| 
       18 
20 
     | 
    
         
             
                  valid_lock? || !!try_lock
         
     | 
| 
      
 21 
     | 
    
         
            +
                rescue => e
         
     | 
| 
      
 22 
     | 
    
         
            +
                  logger.error("Failed to acquire a lock with error: #{e.message}")
         
     | 
| 
      
 23 
     | 
    
         
            +
                  false
         
     | 
| 
       19 
24 
     | 
    
         
             
                end
         
     | 
| 
       20 
25 
     | 
    
         | 
| 
       21 
26 
     | 
    
         
             
                def unlock
         
     | 
| 
       22 
27 
     | 
    
         
             
                  lock_manager.unlock(lock_id) if valid_lock?
         
     | 
| 
      
 28 
     | 
    
         
            +
                rescue => e
         
     | 
| 
      
 29 
     | 
    
         
            +
                  logger.error("Failed to release the lock with error: #{e.message}")
         
     | 
| 
      
 30 
     | 
    
         
            +
                  false
         
     | 
| 
       23 
31 
     | 
    
         
             
                end
         
     | 
| 
       24 
32 
     | 
    
         | 
| 
       25 
33 
     | 
    
         
             
                def extend_lock
         
     | 
| 
         @@ -28,6 +36,9 @@ module Schked 
     | 
|
| 
       28 
36 
     | 
    
         
             
                  @lock_id = lock_manager.lock(LOCK_KEY, lock_ttl, extend: lock_id, extend_only_if_locked: true)
         
     | 
| 
       29 
37 
     | 
    
         | 
| 
       30 
38 
     | 
    
         
             
                  !!@lock_id
         
     | 
| 
      
 39 
     | 
    
         
            +
                rescue => e
         
     | 
| 
      
 40 
     | 
    
         
            +
                  logger.error("Failed to extend the lock with error: #{e.message}")
         
     | 
| 
      
 41 
     | 
    
         
            +
                  false
         
     | 
| 
       31 
42 
     | 
    
         
             
                end
         
     | 
| 
       32 
43 
     | 
    
         | 
| 
       33 
44 
     | 
    
         
             
                def valid_lock?
         
     | 
    
        data/lib/schked/version.rb
    CHANGED
    
    
    
        data/lib/schked/worker.rb
    CHANGED
    
    | 
         @@ -7,7 +7,7 @@ module Schked 
     | 
|
| 
       7 
7 
     | 
    
         
             
                def initialize(config:)
         
     | 
| 
       8 
8 
     | 
    
         
             
                  @config = config
         
     | 
| 
       9 
9 
     | 
    
         | 
| 
       10 
     | 
    
         
            -
                  @locker = RedisLocker.new(config.redis_servers, lock_ttl: 40_000) unless config.standalone?
         
     | 
| 
      
 10 
     | 
    
         
            +
                  @locker = RedisLocker.new(config.redis_servers, lock_ttl: 40_000, logger: config.logger) unless config.standalone?
         
     | 
| 
       11 
11 
     | 
    
         
             
                  @scheduler = Rufus::Scheduler.new(trigger_lock: locker)
         
     | 
| 
       12 
12 
     | 
    
         | 
| 
       13 
13 
     | 
    
         
             
                  watch_signals
         
     | 
| 
         @@ -46,26 +46,29 @@ module Schked 
     | 
|
| 
       46 
46 
     | 
    
         
             
                def define_callbacks
         
     | 
| 
       47 
47 
     | 
    
         
             
                  cfg = config
         
     | 
| 
       48 
48 
     | 
    
         | 
| 
       49 
     | 
    
         
            -
                  scheduler.define_singleton_method(: 
     | 
| 
       50 
     | 
    
         
            -
                     
     | 
| 
      
 49 
     | 
    
         
            +
                  scheduler.define_singleton_method(:extract_job_name) do |job|
         
     | 
| 
      
 50 
     | 
    
         
            +
                    if job
         
     | 
| 
       51 
51 
     | 
    
         
             
                      job.opts[:as] || job.job_id
         
     | 
| 
       52 
52 
     | 
    
         
             
                    else
         
     | 
| 
       53 
53 
     | 
    
         
             
                      "unknown"
         
     | 
| 
       54 
54 
     | 
    
         
             
                    end
         
     | 
| 
       55 
     | 
    
         
            -
             
     | 
| 
      
 55 
     | 
    
         
            +
                  end
         
     | 
| 
      
 56 
     | 
    
         
            +
             
     | 
| 
      
 57 
     | 
    
         
            +
                  scheduler.define_singleton_method(:on_error) do |job, error|
         
     | 
| 
      
 58 
     | 
    
         
            +
                    cfg.logger.fatal("Task #{extract_job_name(job)} failed with error: #{error.message}")
         
     | 
| 
       56 
59 
     | 
    
         
             
                    cfg.logger.error(error.backtrace.join("\n")) if error.backtrace
         
     | 
| 
       57 
60 
     | 
    
         | 
| 
       58 
61 
     | 
    
         
             
                    cfg.fire_callback(:on_error, job, error)
         
     | 
| 
       59 
62 
     | 
    
         
             
                  end
         
     | 
| 
       60 
63 
     | 
    
         | 
| 
       61 
64 
     | 
    
         
             
                  scheduler.define_singleton_method(:on_pre_trigger) do |job, time|
         
     | 
| 
       62 
     | 
    
         
            -
                    cfg.logger.info("Started task: #{job 
     | 
| 
      
 65 
     | 
    
         
            +
                    cfg.logger.info("Started task: #{extract_job_name(job)}")
         
     | 
| 
       63 
66 
     | 
    
         | 
| 
       64 
67 
     | 
    
         
             
                    cfg.fire_callback(:before_start, job, time)
         
     | 
| 
       65 
68 
     | 
    
         
             
                  end
         
     | 
| 
       66 
69 
     | 
    
         | 
| 
       67 
70 
     | 
    
         
             
                  scheduler.define_singleton_method(:on_post_trigger) do |job, time|
         
     | 
| 
       68 
     | 
    
         
            -
                    cfg.logger.info("Finished task: #{job 
     | 
| 
      
 71 
     | 
    
         
            +
                    cfg.logger.info("Finished task: #{extract_job_name(job)}")
         
     | 
| 
       69 
72 
     | 
    
         | 
| 
       70 
73 
     | 
    
         
             
                    cfg.fire_callback(:after_finish, job, time)
         
     | 
| 
       71 
74 
     | 
    
         
             
                  end
         
     | 
| 
         @@ -85,7 +88,7 @@ module Schked 
     | 
|
| 
       85 
88 
     | 
    
         
             
                  Thread.new do
         
     | 
| 
       86 
89 
     | 
    
         
             
                    loop do
         
     | 
| 
       87 
90 
     | 
    
         
             
                      scheduler.shutdown(wait: 5) if @shutdown
         
     | 
| 
       88 
     | 
    
         
            -
                      sleep  
     | 
| 
      
 91 
     | 
    
         
            +
                      sleep 1
         
     | 
| 
       89 
92 
     | 
    
         
             
                    end
         
     | 
| 
       90 
93 
     | 
    
         
             
                  end
         
     | 
| 
       91 
94 
     | 
    
         
             
                end
         
     | 
    
        metadata
    CHANGED
    
    | 
         @@ -1,14 +1,14 @@ 
     | 
|
| 
       1 
1 
     | 
    
         
             
            --- !ruby/object:Gem::Specification
         
     | 
| 
       2 
2 
     | 
    
         
             
            name: schked
         
     | 
| 
       3 
3 
     | 
    
         
             
            version: !ruby/object:Gem::Version
         
     | 
| 
       4 
     | 
    
         
            -
              version: 1.1. 
     | 
| 
      
 4 
     | 
    
         
            +
              version: 1.1.2
         
     | 
| 
       5 
5 
     | 
    
         
             
            platform: ruby
         
     | 
| 
       6 
6 
     | 
    
         
             
            authors:
         
     | 
| 
       7 
7 
     | 
    
         
             
            - Misha Merkushin
         
     | 
| 
       8 
8 
     | 
    
         
             
            autorequire:
         
     | 
| 
       9 
9 
     | 
    
         
             
            bindir: exe
         
     | 
| 
       10 
10 
     | 
    
         
             
            cert_chain: []
         
     | 
| 
       11 
     | 
    
         
            -
            date: 2022- 
     | 
| 
      
 11 
     | 
    
         
            +
            date: 2022-12-16 00:00:00.000000000 Z
         
     | 
| 
       12 
12 
     | 
    
         
             
            dependencies:
         
     | 
| 
       13 
13 
     | 
    
         
             
            - !ruby/object:Gem::Dependency
         
     | 
| 
       14 
14 
     | 
    
         
             
              name: redlock
         
     |