factor/basis/logging/logging.factor

150 lines
3.8 KiB
Factor

! Copyright (C) 2003, 2008 Slava Pestov.
! See http://factorcode.org/license.txt for BSD license.
USING: logging.server sequences namespaces concurrency.messaging
words kernel arrays shuffle tools.annotations
prettyprint.config prettyprint debugger io.streams.string
splitting continuations effects generalizations parser strings
quotations fry accessors math assocs math.order
sequences.generalizations ;
IN: logging
SYMBOLS: DEBUG NOTICE WARNING ERROR CRITICAL ;
SYMBOL: log-level
log-level [ DEBUG ] initialize
: log-levels ( -- assoc )
H{
{ DEBUG 0 }
{ NOTICE 10 }
{ WARNING 20 }
{ ERROR 30 }
{ CRITICAL 40 }
} ; inline
ERROR: undefined-log-level ;
: log-level<=> ( log-level log-level -- <=> )
[ log-levels at* [ undefined-log-level ] unless ] compare ;
: log? ( log-level -- ? )
log-level get log-level<=> +lt+ = not ;
: send-to-log-server ( array string -- )
prefix "log-server" get send ;
SYMBOL: log-service
ERROR: bad-log-message-parameters msg word level ;
: check-log-message ( msg word level -- msg word level )
3dup [ string? ] [ word? ] [ word? ] tri* and and
[ bad-log-message-parameters ] unless ; inline
: log-message ( msg word level -- )
check-log-message
log-service get
2dup [ log? ] [ ] bi* and [
[ [ string-lines ] [ name>> ] [ name>> ] tri* ] dip
4array "log-message" send-to-log-server
] [
4drop
] if ;
: rotate-logs ( -- )
{ } "rotate-logs" send-to-log-server ;
: close-logs ( -- )
{ } "close-logs" send-to-log-server ;
: with-logging ( service quot -- )
[ log-service ] dip with-variable ; inline
! Aspect-oriented programming idioms
<PRIVATE
: stack>message ( obj -- inputs>message )
dup array? [ dup length 1 = [ first ] when ] when
dup string? [
[
boa-tuples? on
string-limit? off
1 line-limit set
3 nesting-limit set
0 margin set
unparse
] with-scope
] unless ;
PRIVATE>
: (define-logging) ( word level quot -- )
[ dup ] 2dip 2curry annotate ; inline
: call-logging-quot ( quot word level -- quot' )
[ "called" ] 2dip [ log-message ] 3curry prepose ;
: add-logging ( word level -- )
[ call-logging-quot ] (define-logging) ;
: log-stack ( n word level -- )
log-service get [
[ [ ndup ] keep narray stack>message ] 2dip log-message
] [
3drop
] if ; inline
: input# ( word -- n ) stack-effect in>> length ;
: input-logging-quot ( quot word level -- quot' )
rot [ [ input# ] keep ] 2dip '[ _ _ _ log-stack @ ] ;
: add-input-logging ( word level -- )
[ input-logging-quot ] (define-logging) ;
: output# ( word -- n ) stack-effect out>> length ;
: output-logging-quot ( quot word level -- quot' )
[ [ output# ] keep ] dip '[ @ _ _ _ log-stack ] ;
: add-output-logging ( word level -- )
[ output-logging-quot ] (define-logging) ;
: error>string ( error -- string )
[ print-error :c ] with-string-writer ;
: (log-error) ( object word level -- )
log-service get [
[ error>string ] 2dip log-message
] [ 2drop rethrow ] if ;
: log-error ( error word -- ) ERROR (log-error) ;
: log-critical ( error word -- ) CRITICAL (log-error) ;
: stack-balancer ( effect -- quot )
[ in>> length [ ndrop ] curry ]
[ out>> length f <repetition> >quotation ]
bi append ;
: error-logging-quot ( quot word -- quot' )
dup stack-effect stack-balancer
'[ _ [ _ log-error @ ] recover ] ;
: add-error-logging ( word level -- )
[ [ input-logging-quot ] keepd error-logging-quot ]
(define-logging) ;
SYNTAX: LOG:
! Syntax: name level
scan-new-word dup scan-word
'[ 1array stack>message _ _ log-message ]
( message -- ) define-declared ;
USE: vocabs
"logging.parser" require
"logging.analysis" require