summaryrefslogtreecommitdiff
path: root/bin/background_jobs_sk_cluster
blob: d48b5484fcea7a62a681d56a76d6d79b1732a05f (plain)
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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
#!/usr/bin/env bash

cd $(dirname $0)/..
app_root=$(pwd)
sidekiq_pidfile="$app_root/tmp/pids/sidekiq-cluster.pid"
sidekiq_logfile="$app_root/log/sidekiq.log"
gitlab_user=$(ls -l config.ru | awk '{print $3}')

warn()
{
  echo "$@" 1>&2
}

get_sidekiq_pid()
{
  if [ ! -f $sidekiq_pidfile ]; then
    warn "No pidfile found at $sidekiq_pidfile; is Sidekiq running?"
    return
  fi

  cat $sidekiq_pidfile
}

stop()
{
  sidekiq_pid=$(get_sidekiq_pid)

  if [ $sidekiq_pid ]; then
    kill -TERM $sidekiq_pid
  fi
}

restart()
{
  if [ -f $sidekiq_pidfile ]; then
    stop
  fi

  warn "Sidekiq output will be written to $sidekiq_logfile"
  start_sidekiq "$@" >> $sidekiq_logfile 2>&1
}

start_sidekiq()
{
  cmd="exec"
  chpst=$(command -v chpst)

  if [ -n "$chpst" ]; then
    cmd="${cmd} ${chpst} -P"
  fi

  # sidekiq-cluster expects '*' '*' arguments (one wildcard for each process).
  for (( i=1; i<=$SIDEKIQ_WORKERS; i++ ))
  do
    processes_args+=("*")
  done

  ${cmd} bin/sidekiq-cluster "${processes_args[@]}" -P $sidekiq_pidfile -e $RAILS_ENV "$@"
}

case "$1" in
  stop)
    stop
    ;;
  start)
    restart "$@" &
    ;;
  start_foreground)
    start_sidekiq "$@"
    ;;
  restart)
    restart "$@" &
    ;;
  *)
    echo "Usage: RAILS_ENV=<env> SIDEKIQ_WORKERS=<n> $0 {stop|start|start_foreground|restart}"
esac