summaryrefslogtreecommitdiff
path: root/ebus-racket/inserter-pq.rkt
diff options
context:
space:
mode:
Diffstat (limited to 'ebus-racket/inserter-pq.rkt')
-rwxr-xr-xebus-racket/inserter-pq.rkt100
1 files changed, 100 insertions, 0 deletions
diff --git a/ebus-racket/inserter-pq.rkt b/ebus-racket/inserter-pq.rkt
new file mode 100755
index 0000000..7ed87d9
--- /dev/null
+++ b/ebus-racket/inserter-pq.rkt
@@ -0,0 +1,100 @@
+#! /usr/bin/env racket
+#lang racket/base
+(require racket/cmdline
+ racket/tcp
+ racket/match
+ "db-pq.rkt"
+ "layer7.rkt"
+ "layer2.rkt"
+ "tcp-repl.rkt")
+
+(define logger (make-logger 'ebus-inserter (current-logger)))
+
+(define connect-host? (make-parameter null))
+(define connect-port? (make-parameter null))
+(define loglevel-layer2? (make-parameter 'info))
+(define loglevel-db? (make-parameter 'warning))
+(define insert? (make-parameter #f))
+
+(define (handle-fields fields)
+ (for ([field fields])
+ (log-message logger 'info (format "Field: ~a" field) #t)
+ (when (insert?)
+ (with-handlers ([exn:fail? (lambda (exn)
+ (log-message logger 'error (format "Failed to insert ~a: ~a" field exn) #t))])
+ (apply db-insert-field field)))))
+
+(define (make-ebus-loop7 input-port)
+ (lambda ()
+ (let loop ()
+ (with-handlers ([exn:fail? (lambda (exn)
+ (log-message logger 'error (format "Failed to parse paket: ~a" exn) #t))])
+ (let ([fields (layer7-read-ebus (current-input-port))])
+ (when (not (or (void? fields) (eof-object? fields)))
+ (handle-fields fields))
+ (when (not (eof-object? fields))
+ (loop)))))))
+
+;; Start Thread that observe all given log-receivers
+(define (start-logger-thread receiver1 . receiverN)
+ (define receivers (cons receiver1 receiverN))
+ (void
+ (thread
+ (lambda ()
+ (let loop ()
+ (match (apply sync receivers)
+ [(vector level msg data)
+ (printf "[~s] ~a~n" level msg)
+ (flush-output)])
+ (loop))))))
+
+(define-namespace-anchor repl-ns-anchor)
+
+(define (main)
+ ;; Parse commandline
+ (command-line
+ #:once-each
+ [("-c" "--connect") host port "Connect to server <host> <port>"
+ (connect-host? host)
+ (connect-port? (string->number port))]
+ ["--tcp-repl" port "Open REPL on TCP <port>"
+ (tcp-repl-run (namespace-anchor->namespace repl-ns-anchor) (string->number port))]
+ ["--debug-layer2" "Log level for Layer 2 Parser"
+ (loglevel-layer2? 'debug)]
+ ["--debug-db" "Log level for DB"
+ (loglevel-db? 'debug)]
+ ["--insert" "Do Insert into Database"
+ (insert? #t)]
+ ["--db-user" user "Datanase User"
+ (db-con-user? user)]
+ ["--db-password" password "Database password"
+ (db-con-password? password)]
+ ["--db-database" database "Database database-name"
+ (db-con-database? database)]
+ ["--db-server" server "Database Address/Server"
+ (db-con-server? server)])
+
+ ;; Init Logging
+ (start-logger-thread (make-log-receiver logger 'info)
+ (make-log-receiver db-logger (loglevel-db?))
+ (make-log-receiver layer2-logger (loglevel-layer2?))
+ (make-log-receiver layer7-logger 'info)
+ (make-log-receiver tcp-repl-logger 'info))
+
+ ;; Test Database Connection
+ (when (not (db-pgc-test))
+ (log-message logger 'fatal "Failed to connect to database" #t)
+ (exit 1))
+
+ ;; Connect, replacing input with tcp connection
+ (if (or (null? (connect-host?)) (null? (connect-port?)))
+ (log-message (current-logger) 'info "Using stdin" #t)
+ (let-values ([(cin cout) (tcp-connect (connect-host?) (connect-port?))])
+ (log-message logger 'info (format "Connected to ~s ~s ~n" (connect-host?) (connect-port?)) #t)
+ (current-input-port cin)))
+
+ ;; Process Ebus Packets
+ ((make-ebus-loop7 (current-input-port)) )
+ )
+
+(exit (main))