| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #include "StreamHandler.h" |
| |
| #include <decaf/util/logging/Level.h> |
| #include <decaf/util/logging/Formatter.h> |
| #include <decaf/util/logging/ErrorManager.h> |
| #include <decaf/util/logging/Filter.h> |
| #include <decaf/io/OutputStream.h> |
| #include <decaf/io/OutputStreamWriter.h> |
| #include <decaf/io/Writer.h> |
| |
| using namespace decaf; |
| using namespace decaf::lang; |
| using namespace decaf::lang::exceptions; |
| using namespace decaf::io; |
| using namespace decaf::util; |
| using namespace decaf::util::logging; |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| StreamHandler::StreamHandler() : Handler(), stream(NULL), writer(NULL), writerNotInitialized(true) { |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| StreamHandler::StreamHandler( OutputStream* stream, Formatter* formatter ) : |
| Handler(), stream(stream), writer(NULL), writerNotInitialized(true) { |
| |
| if( stream == NULL ) { |
| throw NullPointerException( |
| __FILE__, __LINE__, "OutputStream cannot be NULL." ); |
| } |
| |
| if( formatter == NULL ) { |
| throw NullPointerException( |
| __FILE__, __LINE__, "Formatter cannot be NULL." ); |
| } |
| |
| setFormatter( formatter ); |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| StreamHandler::~StreamHandler() { |
| |
| try { |
| this->close(true); |
| } |
| DECAF_CATCH_NOTHROW( lang::Exception) |
| DECAF_CATCHALL_NOTHROW() |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| void StreamHandler::close() { |
| this->close( true ); |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| void StreamHandler::flush() { |
| |
| if( this->stream != NULL ) { |
| try { |
| if( this->writer != NULL ) { |
| this->writer->flush(); |
| } else { |
| this->stream->flush(); |
| } |
| } catch( Exception& e ) { |
| this->getErrorManager()->error( |
| "Failed to flush the output stream", &e, ErrorManager::FLUSH_FAILURE ); |
| } |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| void StreamHandler::publish( const LogRecord& record ) { |
| |
| try { |
| |
| if( this->isLoggable( record ) ) { |
| |
| if( this->writerNotInitialized ) { |
| initializeWritter(); |
| } |
| |
| std::string msg = ""; |
| try { |
| msg = getFormatter()->format( record ); |
| } catch( Exception& e ) { |
| this->getErrorManager()->error( |
| "Failed to format the LogRecord", &e, ErrorManager::FORMAT_FAILURE ); |
| } |
| |
| this->write( msg ); |
| } |
| } catch( Exception& e ) { |
| this->getErrorManager()->error( |
| "Failed to publish the LogRecord", &e, ErrorManager::GENERIC_FAILURE ); |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| bool StreamHandler::isLoggable( const LogRecord& record ) const { |
| |
| if( this->stream != NULL && Handler::isLoggable( record ) ) { |
| return true; |
| } |
| |
| return false; |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| void StreamHandler::setOuputStream( decaf::io::OutputStream* stream ) { |
| |
| if( stream == NULL ) { |
| throw NullPointerException( |
| __FILE__, __LINE__, "OutputStream cannot be set to NULL." ); |
| } |
| |
| this->close( true ); |
| this->stream = stream; |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| void StreamHandler::close( bool closeStream ) { |
| |
| if( this->stream != NULL ) { |
| |
| if( this->writerNotInitialized ) { |
| initializeWritter(); |
| } |
| |
| this->write( getFormatter()->getTail( this ) ); |
| |
| try { |
| |
| this->writer->flush(); |
| |
| if( closeStream ) { |
| this->writer->close(); |
| this->writer = NULL; |
| this->stream = NULL; |
| } |
| |
| } catch( Exception& e ) { |
| this->getErrorManager()->error( |
| "Failed to close the OutputStream", &e, ErrorManager::CLOSE_FAILURE ); |
| } |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| void StreamHandler::write( const std::string& value ) { |
| |
| try{ |
| this->stream->write( (const unsigned char*)value.c_str(), (int)value.length(), 0, (int)value.length() ); |
| } catch( Exception& e ) { |
| this->getErrorManager()->error( |
| "Failed to write to the OutputStream", &e, ErrorManager::WRITE_FAILURE ); |
| } |
| } |
| |
| //////////////////////////////////////////////////////////////////////////////// |
| void StreamHandler::initializeWritter() { |
| |
| this->writerNotInitialized = false; |
| this->writer = new OutputStreamWriter( this->stream ); |
| |
| // Start the Document by writing out the Head element. |
| this->write( getFormatter()->getHead( this ) ); |
| } |