| /* |
| * Copyright (c) 2017, 2018, Oracle and/or its affiliates. All rights reserved. |
| * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. |
| * |
| * This code is free software; you can redistribute it and/or modify it |
| * under the terms of the GNU General Public License version 2 only, as |
| * published by the Free Software Foundation. |
| * |
| * This code is distributed in the hope that it will be useful, but WITHOUT |
| * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
| * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
| * version 2 for more details (a copy is included in the LICENSE file that |
| * accompanied this code). |
| * |
| * You should have received a copy of the GNU General Public License version |
| * 2 along with this work; if not, write to the Free Software Foundation, |
| * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. |
| * |
| * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
| * or visit www.oracle.com if you need additional information or have any |
| * questions. |
| * |
| */ |
| |
| #ifndef SHARE_VM_JFR_CHECKPOINT_TYPES_JFRTYPESETWRITER_HPP |
| #define SHARE_VM_JFR_CHECKPOINT_TYPES_JFRTYPESETWRITER_HPP |
| |
| #include "jfr/recorder/checkpoint/jfrCheckpointWriter.hpp" |
| #include "jfr/utilities/jfrTypes.hpp" |
| #include "memory/allocation.hpp" |
| |
| template <typename WriterImpl, u4 ID> |
| class JfrArtifactWriterHost : public StackObj { |
| private: |
| WriterImpl _impl; |
| JfrCheckpointWriter* _writer; |
| JfrCheckpointContext _ctx; |
| jlong _count_offset; |
| int _count; |
| bool _skip_header; |
| public: |
| JfrArtifactWriterHost(JfrCheckpointWriter* writer, |
| JfrArtifactSet* artifacts, |
| bool class_unload, |
| bool skip_header = false) : _impl(writer, artifacts, class_unload), |
| _writer(writer), |
| _ctx(writer->context()), |
| _count(0), |
| _skip_header(skip_header) { |
| assert(_writer != NULL, "invariant"); |
| if (!_skip_header) { |
| _writer->write_type((JfrTypeId)ID); |
| _count_offset = _writer->reserve(sizeof(u4)); // Don't know how many yet |
| } |
| } |
| |
| ~JfrArtifactWriterHost() { |
| if (_count == 0) { |
| // nothing written, restore context for rewind |
| _writer->set_context(_ctx); |
| return; |
| } |
| assert(_count > 0, "invariant"); |
| if (!_skip_header) { |
| _writer->write_count(_count, _count_offset); |
| } |
| } |
| |
| bool operator()(typename WriterImpl::Type const & value) { |
| this->_count += _impl(value); |
| return true; |
| } |
| |
| int count() const { return _count; } |
| void add(int count) { _count += count; } |
| }; |
| |
| typedef int(*artifact_write_operation)(JfrCheckpointWriter*, JfrArtifactSet*, const void*); |
| |
| template <typename T, artifact_write_operation op> |
| class JfrArtifactWriterImplHost { |
| private: |
| JfrCheckpointWriter* _writer; |
| JfrArtifactSet* _artifacts; |
| bool _class_unload; |
| public: |
| typedef T Type; |
| JfrArtifactWriterImplHost(JfrCheckpointWriter* writer, JfrArtifactSet* artifacts, bool class_unload) : |
| _writer(writer), _artifacts(artifacts), _class_unload(class_unload) {} |
| int operator()(T const& value) { |
| return op(this->_writer, this->_artifacts, value); |
| } |
| }; |
| |
| template <typename T, typename Predicate, artifact_write_operation op> |
| class JfrPredicatedArtifactWriterImplHost : public JfrArtifactWriterImplHost<T, op> { |
| private: |
| Predicate _predicate; |
| typedef JfrArtifactWriterImplHost<T, op> Parent; |
| public: |
| JfrPredicatedArtifactWriterImplHost(JfrCheckpointWriter* writer, JfrArtifactSet* artifacts, bool class_unload) : |
| Parent(writer, artifacts, class_unload), _predicate(class_unload) {} |
| int operator()(T const& value) { |
| return _predicate(value) ? Parent::operator()(value) : 0; |
| } |
| }; |
| |
| #endif // SHARE_VM_JFR_CHECKPOINT_TYPES_JFRTYPESETWRITER_HPP |