150 lines
		
	
	
		
			3.8 KiB
		
	
	
	
		
			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
 |