Initial attempt at porting server.rkt to fastrouting

This commit is contained in:
Tony Garnock-Jones 2014-05-28 16:36:52 -04:00
parent 95701d39c5
commit 461f96d5d4
1 changed files with 26 additions and 46 deletions

View File

@ -1,11 +1,11 @@
#lang minimart #lang minimart
;; Generic broker for WebSockets-based minimart/marketplace communication. ;; Generic broker for WebSockets-based minimart/marketplace communication.
(require racket/set)
(require net/rfc6455) (require net/rfc6455)
(require minimart/drivers/timer) (require minimart/drivers/timer)
(require minimart/drivers/websocket) (require minimart/drivers/websocket)
(require minimart/demand-matcher) (require minimart/demand-matcher)
(require minimart/pattern)
(require json) (require json)
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
@ -20,13 +20,10 @@
(define (spawn-server-listener port ssl-options) (define (spawn-server-listener port ssl-options)
(define server-id (websocket-local-server port ssl-options)) (define server-id (websocket-local-server port ssl-options))
(spawn-demand-matcher (websocket-message (websocket-remote-client ?) server-id ?) (spawn-demand-matcher (websocket-message (?! (websocket-remote-client ?)) server-id ?)
#:meta-level 1 #:meta-level 1
#:demand-is-subscription? #f #:demand-is-subscription? #f
(match-lambda ;; arrived-demand-route, i.e. new connection publisher (lambda (c) (spawn-connection-handler c server-id))))
[(route _ (websocket-message c _ _) 1 _)
(spawn-connection-handler c server-id)]
[_ '()])))
(spawn-world (spawn-world
(spawn-server-listener 8000 #f) (spawn-server-listener 8000 #f)
@ -35,36 +32,18 @@
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;; Wire protocol representation of events and actions ;; Wire protocol representation of events and actions
(define (drop-json-pattern p)
(pattern-subst p (hasheq '__ "__") ?))
(define (drop-json-route r)
(match r
[`(,pub-or-sub ,pattern ,meta-level ,level)
(route (match pub-or-sub ["sub" #t] ["pub" #f])
(drop-json-pattern pattern)
meta-level
level)]))
(define (drop-json-action j) (define (drop-json-action j)
(match j (match j
["ping" 'ping] ["ping" 'ping]
["pong" 'pong] ["pong" 'pong]
[`("routes" ,routes) (routing-update (map drop-json-route routes))] [`("routes" ,gj) (routing-update (jsexpr->gestalt gj (lambda (v) (set 'peer))))]
[`("message" ,body ,meta-level ,feedback?) (message body meta-level feedback?)])) [`("message" ,body ,meta-level ,feedback?) (message body meta-level feedback?)]))
(define (lift-json-pattern p)
(pattern-subst p ? (hasheq '__ "__")))
(define (lift-json-route r)
(match r
[(route sub? p ml l) `(,(if sub? "sub" "pub") ,(lift-json-pattern p) ,ml ,l)]))
(define (lift-json-event j) (define (lift-json-event j)
(match j (match j
['ping "ping"] ['ping "ping"]
['pong "pong"] ['pong "pong"]
[(routing-update rs) `("routes" ,(map lift-json-route rs))] [(routing-update g) `("routes" ,(gestalt->jsexpr g (lambda (v) #t)))]
[(message body meta-level feedback?) `("message" ,body ,meta-level ,feedback?)])) [(message body meta-level feedback?) `("message" ,body ,meta-level ,feedback?)]))
(require racket/trace) (require racket/trace)
@ -74,7 +53,7 @@
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;; Connections ;; Connections
(struct connection-state (client-id tunnelled-routes) #:transparent) (struct connection-state (client-id tunnelled-gestalt) #:transparent)
(define (spawn-connection-handler c server-id) (define (spawn-connection-handler c server-id)
(define (send-event e s) (define (send-event e s)
@ -82,38 +61,39 @@
(connection-state-client-id s) (connection-state-client-id s)
(jsexpr->string (lift-json-event e))) (jsexpr->string (lift-json-event e)))
#:meta-level 1)) #:meta-level 1))
(define ((handle-connection-routing-change rs) s) (define ((handle-connection-routing-change g) s)
(match rs (if (gestalt-empty? g)
['() (transition s (quit))] ;; websocket connection closed (transition s (quit)) ;; websocket connection closed
[_ (transition s '())])) #f))
(define ((handle-tunnelled-routing-change rs) s) (define ((handle-tunnelled-routing-change g) s)
(transition s (send-event (routing-update rs) s))) (transition s (send-event (routing-update g) s)))
(define ((handle-tunnellable-message m) s) (define ((handle-tunnellable-message m) s)
(if (ormap (lambda (r) (route-accepts? r m)) (connection-state-tunnelled-routes s)) (if (gestalt-accepts? (connection-state-tunnelled-gestalt s) m)
(transition s (send-event m s)) (transition s (send-event m s))
(transition s '()))) (transition s '())))
(define relay-connections (define relay-connections
(list (sub (timer-expired c ?) #:meta-level 1) (gestalt-union (sub (timer-expired c ?) #:meta-level 1)
(sub (websocket-message c server-id ?) #:meta-level 1) (sub (websocket-message c server-id ?) #:meta-level 1)
(sub (websocket-message c server-id ?) #:meta-level 1 #:level 1) (sub (websocket-message c server-id ?) #:meta-level 1 #:level 1)
(pub (websocket-message server-id c ?) #:meta-level 1))) (pub (websocket-message server-id c ?) #:meta-level 1)))
(define (connection-handler e s) (define (connection-handler e s)
(match e (match e
[(routing-update rs) [(routing-update g)
(sequence-transitions (sequence-transitions
(transition s '()) (transition s '())
(handle-connection-routing-change (intersect-routes rs relay-connections)) (handle-connection-routing-change (gestalt-filter g relay-connections))
(handle-tunnelled-routing-change (handle-tunnelled-routing-change
(intersect-routes rs (connection-state-tunnelled-routes s))))] (gestalt-filter g (connection-state-tunnelled-gestalt s))))]
[(? message? m) [(? message? m)
(sequence-transitions (sequence-transitions
(match m (match m
[(message (websocket-message from to data) 1 #f) [(message (websocket-message from to data) 1 #f)
(match (drop-json-action (string->jsexpr data)) (match (drop-json-action (string->jsexpr data))
[(routing-update rs-unfiltered) [(routing-update g-unfiltered)
(define rs (filter (lambda (r) (zero? (route-meta-level r))) rs-unfiltered)) (define g (gestalt-transform g-unfiltered
(transition (struct-copy connection-state s [tunnelled-routes rs]) (lambda (ml l p) (if (zero? ml) p '(#f . #f)))))
(routing-update (append rs relay-connections)))] (transition (struct-copy connection-state s [tunnelled-gestalt g])
(routing-update (gestalt-union g relay-connections)))]
[(? message? m) [(? message? m)
(transition s (if (zero? (message-meta-level m)) m '()))] (transition s (if (zero? (message-meta-level m)) m '()))]
['ping ['ping
@ -129,5 +109,5 @@
[#f #f])) [#f #f]))
(list (send (set-timer c ping-interval 'relative) #:meta-level 1) (list (send (set-timer c ping-interval 'relative) #:meta-level 1)
(spawn connection-handler (spawn connection-handler
(connection-state c '()) (connection-state c (gestalt-empty))
relay-connections))) relay-connections)))