ractor_dns 0.3.0 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: abb60858b1f6214730ce4d5ac169dc4b03151b94e610575df0b39469f3d81ff7
-  data.tar.gz: f7cad7e86501e3e194428284ecfa26f86033f3e63654a4cc4782b604d837d564
+  metadata.gz: '02339e90260e10f1dd8e76901a10e02e3435fa2e9b0cdde87fd0099a8cb250f4'
+  data.tar.gz: cd6d4423d563ce281960bf77659262ffe221e17b84b10f993f5b43ead544584b
 SHA512:
-  metadata.gz: b79a3b762b5996b967787f5fdfc1164d5f08cd110c5fc96418dcfd1500d3280d64501f2cea21c2a1c00e3191b351ba9834abe046e6a43459d5b4501379119dac
-  data.tar.gz: a627088137a11008814c51b95e665798937ca2aac1182d5f7f79ca5c85f2f5040075eb4203a54dddb65f55a4d73171b50fc5016cc2a0d01c5a8ff52975d88cf5
+  metadata.gz: 4628cf0965ad76b018466ef3ecba44ef9d1939d5a7c97513d4eb1947e8ef35240a7985afd65fb1169ff4bc31f8ece340ad34027712e0da346709976c4ae8825c
+  data.tar.gz: 02a18d6559b4133de18a2115d77cd131c9fb72be31dfec27d06fbe27599e164830da419a66029bd078c5a18497de73c443fb923dfe6caeb602ebafda913385d9

data/README.md

@@ -6,15 +6,51 @@ RactorDNS is a work-in-progress concurrent DNS server built with Ruby 3.0 Ractor
 
 Install the gem and add to the application's Gemfile by executing:
 
-    bundle add ractor_dns
+```bash
+bundle add ractor_dns
+```
 
 If bundler is not being used to manage dependencies, install the gem by executing:
 
-    gem install ractor_dns
+```bash
+gem install ractor_dns
+```
 
 ## Usage
 
-TODO: Write usage instructions here
+### Creating a server
+
+```ruby
+server = RactorDNS::Server.new(
+  port: 8053,
+  authority: ["ns.ractordns.local", 300, RRs::IN::A.new("1.1.1.1")], 
+  cpu_count: 12,
+  zones: {
+    "google.com" => [
+      {
+        name: "google.com",
+        ttl: 300,
+        rr: RRs::IN::A.new("8.8.8.8")
+      },
+      {
+        name: "test.google.com",
+        ttl: 300,
+        rr: RRs::IN::A.new("1.1.1.1")
+      }
+    ]
+  }
+)
+```
+
+### Updating zones
+
+```ruby
+transaction = RactorDNS::Transaction.new do |zones|
+  zones["google.com"][0][:rr] = RRs::IN::A.new("2.2.2.2")
+end
+
+server.zones.transact transaction
+```
 
 ## Development
 

data/lib/ractor_dns/active_zone.rb

@@ -1,6 +1,6 @@
 class RactorDNS::ActiveZone
   attr_accessor :server
-  def initialize(name: "ns.ractordns.local", ttl: 300, ip: "0.0.0.0", server: nil, port: 53, cpu_count: 8, zones:)
+  def initialize(name: "ns.ractordns.local", ttl: 300, ip: "0.0.0.0", server: nil, port: 8053, cpu_count: 8, zones:)
    @server = server.presence || RactorDNS::Server.new(
      port:,
      authority: [name, ttl, RRs::IN::A.new(ip)], 

data/lib/ractor_dns/server.rb

@@ -2,10 +2,10 @@ module RactorDNS
   class Server
     attr_reader :authority, :ractor, :zones, :cpu_count, :port
 
-    def initialize(authority:, cpu_count:, zones:, port: 53)
+    def initialize(authority:, cpu_count:, zones:, port: 8053)
       @authority = authority
       @cpu_count = cpu_count
-      @zones = zones
+      @zones = ZoneCollection.new(zones)
       @port = port
     end
 
@@ -23,10 +23,10 @@ module RactorDNS
         end
       end
 
-      @resolvers = @cpu_count.times.map do
+      @resolvers = (@cpu_count ^ 2).times.map do
         Ractor.new(@pipe, @authority) do |pipe, authority|
           loop do
-            data, zone_h = pipe.take
+            data, zone_h = Ractor.recv
 
             break if data == :stop && zone_h.nil?
 
@@ -49,16 +49,16 @@ module RactorDNS
           end
         end
       end
-
-      @endpoint = Async::IO::Endpoint.udp('0.0.0.0', @port)
       
       @zones.start
 
       if here
         start_main_loop
       else
-        @thread = Thread.new do
-          start_main_loop
+        @threads = @cpu_count.times.map do |i|
+          Thread.new do
+            start_main_loop
+          end
         end
       end
 
@@ -69,7 +69,9 @@ module RactorDNS
 
     def stop
       begin
-        @thread.kill
+        @threads.each do |t|
+          t.kill
+        end
         @pipe.send(:stop)
         @zones.stop
       rescue Ractor::ClosedError
@@ -77,14 +79,16 @@ module RactorDNS
       end
     end
     
-    def start_main_loop
+    def start_main_loop(i = rand(0...@resolvers.length))
       Async do
-        @endpoint.bind do |socket|
-          loop do
-            data, address = socket.recvfrom(1024)
-            @pipe.send([data, @zones.freeze])
-            _, encoded = Ractor.select(*@resolvers)
-            socket.send(encoded, 0, address)
+        @cpu_count.times do
+          Async::IO::Endpoint.udp('0.0.0.0', @port).bind do |socket|
+            loop do
+              data, address = socket.recvfrom(1024)
+              @resolvers[i].send([data, @zones.freeze])
+              encoded = @resolvers[i].take
+              socket.send(encoded, 0, address)
+            end
           end
         end
       end

data/lib/ractor_dns/version.rb

@@ -1,3 +1,3 @@
 module RactorDNS
-  VERSION = "0.3.0"
+  VERSION = "0.5.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ractor_dns
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.5.0
 platform: ruby
 authors:
 - reesericci
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-08-19 00:00:00.000000000 Z
+date: 2024-08-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport