Skip to content

Commit

Permalink
init it
Browse files Browse the repository at this point in the history
  • Loading branch information
palkan committed Sep 13, 2024
0 parents commit da8e81b
Show file tree
Hide file tree
Showing 17 changed files with 557 additions and 0 deletions.
1 change: 1 addition & 0 deletions .anycable-rails-revision
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
https://github.com/anycable/anycable-rails.git#feat/actioncable-v8
42 changes: 42 additions & 0 deletions .github/workflows/ci.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,42 @@
name: CI

on:
push:
branches:
- master
pull_request:
workflow_dispatch:

jobs:
wsdirector:
if: ${{ !contains(github.event.head_commit.message, '[ci skip]') }}
runs-on: ubuntu-latest
env:
BUNDLE_JOBS: 4
BUNDLE_RETRY: 3
CI: true
services:
redis:
image: redis:7.0-alpine
ports: ["6379:6379"]
options: --health-cmd="redis-cli ping" --health-interval 1s --health-timeout 3s --health-retries 30
strategy:
fail-fast: false
matrix:
server: ["puma", "anycable"]
steps:
- uses: actions/checkout@v4
- uses: ruby/setup-ruby@v1
with:
ruby-version: 3.3
bundler-cache: true
- name: Run server
run: |
bundle exec bento --${{ matrix.server }} &
bin/wait_tcp 8080
- name: Run echo scenario
run: |
SCENARIO=echo make wsdirector
- name: Run broadcast scenario
run: |
SCENARIO=broadcast make wsdirector
10 changes: 10 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
anycable-go-*
k6

.rails-path
.anycable-path

bin/dist
!bin/dist/.keep

Gemfile.lock
1 change: 1 addition & 0 deletions .rails-revision
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
https://github.com/palkan/rails.git#refactor/action-cable-server-adapterization
50 changes: 50 additions & 0 deletions Gemfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,50 @@
source "https://rubygems.org"

rails_path = File.file?(File.join(__dir__, ".rails-path")) ? File.read(File.join(__dir__, ".rails-path")).strip : File.join(__dir__, "../rails")

# Use local Rails copy if available
if File.directory?(rails_path)
gem "rails", group: :preload, path: rails_path
# Use Rails from a git repo
elsif File.file?(File.join(__dir__, ".rails-revision"))
git, branch = *File.read(File.join(__dir__, ".rails-revision")).strip.split("#", 2)
gem "rails", group: :preload, git:, branch:
else
gem "rails", "~> 8.0"
end

# Baseline setup: Puma + Redis pub/sub
gem "puma", "~> 6.4"
gem "redis", "~> 5.0", group: :preload

# Async setup
# TODO

# AnyCable setup
gem "grpc_kit" if ENV["ANYCABLE_GRPC_IMPL"] == "grpc_kit"
gem "grpc" unless ENV["ANYCABLE_GRPC_IMPL"] == "grpc_kit"

anycable_dir_path = File.file?(File.join(__dir__, ".anycable-path")) ? File.read(File.join(__dir__, ".anycable-path")).strip : File.join(__dir__, "..")

if File.file?(File.join(anycable_dir_path, "anycable/anycable-core.gemspec"))
gem "anycable-core", group: :preload, path: File.join(anycable_dir_path, "anycable")
elsif File.file?(File.join(__dir__, ".anycable-revision"))
git, branch = *File.read(File.join(__dir__, ".anycable-revision")).strip.split("#", 2)
gem "anycable-core", require: false, group: :preload, git:, branch:
else
gem "anycable-core"
end

if File.file?(File.join(anycable_dir_path, "anycable-rails/anycable-rails.gemspec"))
gem "anycable-rails", group: :preload, path: File.join(anycable_dir_path, "anycable-rails")
elsif File.file?(File.join(__dir__, ".anycable-rails-revision"))
git, branch = *File.read(File.join(__dir__, ".anycable-rails-revision")).strip.split("#", 2)
gem "anycable-rails", require: false, group: :preload, git:, branch:
else
gem "anycable-rails"
end

# Tools
gem "wsdirector-cli", require: false

gem "debug" unless ENV["CI"]
24 changes: 24 additions & 0 deletions Makefile
Original file line number Diff line number Diff line change
@@ -0,0 +1,24 @@
SCENARIO ?= echo
SCALE ?= 10

websocket-bench:
websocket-bench broadcast ws://localhost:8080/cable \
--concurrent 8 \
--sample-size 100 \
--step-size 200 \
--payload-padding 200 \
--total-steps 10 \
--origin http://0.0.0.0 \
--server-type=actioncable

build-k6:
@test -x ./k6 || \
xk6 build v0.42.0 --with github.com/anycable/xk6-cable@latest

k6: build-k6
./k6 run scripts/k6/benchmark.js

wsdirector:
@bundle exec wsdirector -f scripts/wsdirector/$(SCENARIO).yml -u ws://localhost:8080/cable -s $(SCALE)

.PHONY: websocket-bench
68 changes: 68 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,68 @@
# Action Cable 8 playground

This is a support project for the Action Server adapterization happening here: [rails/rails#50979][the-pr].

This project contains examples, tests and benchmarks aiming to ensure that the refactoring goes well and brings the promised benefits.

## Usage

Install the dependencies:

```sh
bundle install
```

NOTE: You can use local dependencies (Rails, AnyCable) either by following path conventions or by providing `.xxx-path` files (see the Gemfile).

Now you should be able to run a minimal Action Cable application via one of the supported web servers. By default, Puma is used:

```sh
$ bundle exec bento

⚡️ Running Action Cable via puma

[18378] Puma starting in cluster mode...

...
```

Other servers:

```sh
# AnyCable

$ bundle exec bento --anycable

⚡️ Running Action Cable via anycable

2024-09-13 10:07:54.272 INF Starting AnyCable 1.5.3-56288f0 (pid: 37818, open file limit: 122880, gomaxprocs: 8) nodeid=NFoelH

...
```

## Tests

You can run basic smoke tests via [wsdirector][] as follows (NOTE: the server must be running):

```sh
# The default scenario is echo
$ make wsdirector

10 clients, 0 failures

# Run broadcast scenario
$ SCENARIO=broadcast make wsdirector

Group publisher: 10 clients, 0 failures
Group listener: 20 clients, 0 failures

# You can specify the scale factor, too (default: 10)
$ SCALE=20 SCENARIO=broadcast make wsdirector

Group publisher: 20 clients, 0 failures
Group listener: 40 clients, 0 failures

```

[the-pr]: https://github.com/rails/rails/pull/50979
[wsdirector]: https://github.com/palkan/wsdirector
31 changes: 31 additions & 0 deletions bento
Original file line number Diff line number Diff line change
@@ -0,0 +1,31 @@
#!/usr/bin/env ruby

require "bundler/setup"

$stdout.puts "YJIT is enabled 🎉" if RUBY_DESCRIPTION =~ /\+YJIT/

require "optparse"

$benchmark_server = :puma

OptionParser.new do |opts|
opts.banner = "Usage: bento [options]"

opts.on('--puma', 'Run Rails via Puma server') do |v|
$benchmark_server = :puma
end

opts.on('--anycable', 'Run AnyCable RPC server') do |v|
$benchmark_server = :anycable
end

opts.on('--falcon', 'Run Falcon/Async server') do |v|
$benchmark_server = :falcon
end
end.parse!

require_relative "lib/application"

$stdout.puts "⚡️ Running Action Cable via #{$benchmark_server}"

BenchmarkServer.run!
22 changes: 22 additions & 0 deletions bin/anycable-go
Original file line number Diff line number Diff line change
@@ -0,0 +1,22 @@
#!/bin/bash

cd $(dirname $0)/..

# It's recommended to use the exact version of AnyCable-Go here
version="1.5.3"

if [ ! -f ./bin/dist/anycable-go ]; then
echo "AnyCable server is not installed, downloading..."
./bin/rails g anycable:download --version=$version --bin-path=$(pwd)/bin/dist
fi

curVersion=$(./bin/dist/anycable-go -v)

if [[ "$version" != "latest" ]]; then
if [[ "$curVersion" != "$version"* ]]; then
echo "AnyCable server version is not $version, downloading a new one..."
./bin/rails g anycable:download --version=$version --bin-path=$(pwd)/bin/dist
fi
fi

./bin/dist/anycable-go $@
11 changes: 11 additions & 0 deletions bin/rails
Original file line number Diff line number Diff line change
@@ -0,0 +1,11 @@
#!/usr/bin/env ruby

require "bundler/setup"

APP_PATH = File.expand_path("../lib/application", __dir__)

require_relative APP_PATH

Dir.chdir(File.dirname(APP_PATH))

require "rails/commands"
24 changes: 24 additions & 0 deletions bin/wait_tcp
Original file line number Diff line number Diff line change
@@ -0,0 +1,24 @@
#!/usr/bin/env ruby

require "socket"

host = "127.0.0.1"
port = ARGV[0]
timeout = ARGV[1]&.to_f || 5.0 # seconds

$stdout.puts "Waiting for TCP server to start at #{port}"

while timeout > 0
begin
Socket.tcp(host, port, connect_timeout: 1).close
$stdout.puts "TCP server is listening at #{port}"
return
rescue Errno::ECONNREFUSED, Errno::EHOSTUNREACH, SocketError
end

sleep 0.5
timeout -= 0.5
end

$stderr.puts "No server is listening at #{port}"
exit(1)
68 changes: 68 additions & 0 deletions lib/application.rb
Original file line number Diff line number Diff line change
@@ -0,0 +1,68 @@
# frozen_string_literal: true

require "rails"
require "global_id"

require "action_controller/railtie"
require "action_view/railtie"
require "action_cable/engine"

require "debug" unless ENV["CI"]

# config/application.rb
class App < Rails::Application
config.root = __dir__
config.eager_load = false
config.consider_all_requests_local = true
config.action_dispatch.show_exceptions = false
config.secret_key_base = "i_am_a_secret"

config.hosts = []

config.logger = ActiveSupport::Logger.new((ENV["LOG"] == "1") ? $stdout : IO::NULL)
config.log_level = (ENV["LOG"] == "1") ? :debug : :fatal

routes.append do
# Add routes here if needed
end
end

ActionCable.server.config.connection_class = -> { ApplicationCable::Connection }
ActionCable.server.config.disable_request_forgery_protection = true
ActionCable.server.config.logger = Rails.logger

# Load server configuration
require_relative "servers/#{$benchmark_server}" if defined?($benchmark_server)

Rails.application.initialize!

module ApplicationCable
class Connection < ActionCable::Connection::Base
identified_by :uid
end

class Channel < ActionCable::Channel::Base
end
end

class BenchmarkChannel < ApplicationCable::Channel
def subscribed
stream_from "all#{stream_id}"
end

def echo(data)
transmit data
end

def broadcast(data)
ActionCable.server.broadcast "all#{stream_id}", data
data["action"] = "broadcastResult"
transmit data
end

private

def stream_id
params[:id] || ""
end
end
21 changes: 21 additions & 0 deletions lib/servers/anycable.rb
Original file line number Diff line number Diff line change
@@ -0,0 +1,21 @@
# frozen_string_literal: true

require "redis"
require "anycable-rails"

ActionCable.server.config.cable = {
"adapter" => $benchmark_server == :anycable ? "any_cable" : "redis",
"url" => ENV["REDIS_URL"]
}

class BenchmarkServer
def self.run!
require "anycable/cli"
cli = AnyCable::CLI.new
# We're already within the app context
cli.define_singleton_method(:boot_app!) { }

anycable_server_path = Rails.root.join("../bin/anycable-go")
cli.run(["--server-command", "#{anycable_server_path} --host 0.0.0.0"])
end
end
Loading

0 comments on commit da8e81b

Please sign in to comment.