src/jdk.jfr/share/classes/jdk/jfr/consumer/ConstantMap.java
author egahlin
Fri, 17 May 2019 16:02:27 +0200
branchJEP-349-branch
changeset 57360 5d043a159d5c
parent 50113 caf115bb98ad
child 57373 400db63e4937
permissions -rw-r--r--
Preview
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
     1
/*
57360
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
     2
 * Copyright (c) 2016, 2019, Oracle and/or its affiliates. All rights reserved.
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
     3
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
     4
 *
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
     5
 * This code is free software; you can redistribute it and/or modify it
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
     6
 * under the terms of the GNU General Public License version 2 only, as
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
     7
 * published by the Free Software Foundation.  Oracle designates this
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
     8
 * particular file as subject to the "Classpath" exception as provided
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
     9
 * by Oracle in the LICENSE file that accompanied this code.
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    10
 *
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    11
 * This code is distributed in the hope that it will be useful, but WITHOUT
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    12
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    13
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    14
 * version 2 for more details (a copy is included in the LICENSE file that
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    15
 * accompanied this code).
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    16
 *
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    17
 * You should have received a copy of the GNU General Public License version
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    18
 * 2 along with this work; if not, write to the Free Software Foundation,
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    19
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    20
 *
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    21
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    22
 * or visit www.oracle.com if you need additional information or have any
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    23
 * questions.
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    24
 */
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    25
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    26
package jdk.jfr.consumer;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    27
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    28
/**
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    29
 * Holds mapping between a set of keys and their corresponding object.
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    30
 *
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    31
 * If the type is a known type, i.e. {@link RecordedThread}, an
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    32
 * {@link ObjectFactory} can be supplied which will instantiate a typed object.
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    33
 */
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    34
final class ConstantMap {
57360
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    35
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    36
    private static final int RESOLUTION_FINISHED = 0;
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    37
    private static final int RESOLUTION_STARTED = 1;
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    38
    public static final ConstantMap EMPTY = new ConstantMap();
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    39
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    40
    // A temporary placeholder, so objects can
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    41
    // reference themselves (directly, or indirectly),
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    42
    // when making a transition from numeric id references
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    43
    // to normal Java references.
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    44
    private final static class Reference {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    45
        private final long key;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    46
        private final ConstantMap pool;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    47
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    48
        Reference(ConstantMap pool, long key) {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    49
            this.pool = pool;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    50
            this.key = key;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    51
        }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    52
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    53
        Object resolve() {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    54
            return pool.get(key);
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    55
        }
57360
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    56
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    57
        public String toString() {
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    58
            return "ref: " + pool.name + "[" + key + "]";
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    59
        }
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    60
    }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    61
57360
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    62
    final ObjectFactory<?> factory;
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    63
    final String name;
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    64
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    65
    private final LongMap<Object> objects;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    66
57360
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    67
    private boolean resolving;
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    68
    private boolean allResolved;
57360
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    69
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    70
    private ConstantMap() {
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    71
        this(null, "<empty>");
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    72
        allResolved = true;
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    73
    }
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    74
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    75
    ConstantMap(ObjectFactory<?> factory, String name) {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    76
        this.name = name;
57360
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    77
        this.objects = new LongMap<>(2);
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    78
        this.factory = factory;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    79
    }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    80
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    81
    Object get(long id) {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    82
        // fast path, all objects in pool resolved
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    83
        if (allResolved) {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    84
            return objects.get(id);
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    85
        }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    86
        // referenced from a pool, deal with this later
57360
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    87
        if (!resolving) {
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    88
            return new Reference(this, id);
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    89
        }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    90
57360
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    91
        // should always have a value
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    92
        Object value = objects.get(id);
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    93
        if (value == null) {
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    94
            // unless is 0 which is used to represent null
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    95
            if (id == 0) {
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    96
                return null;
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    97
            }
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    98
            throw new InternalError("Missing object id=" + id + " in pool " + name + ". All ids should reference object");
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
    99
        }
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   100
57360
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   101
        // id is resolved (but not the whole pool)
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   102
        if (objects.isSetId(id, RESOLUTION_FINISHED)) {
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   103
            return value;
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   104
        }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   105
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   106
        // resolving ourself, abort to avoid infinite recursion
57360
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   107
        if (objects.isSetId(id, RESOLUTION_STARTED)) {
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   108
            return null;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   109
        }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   110
57360
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   111
        // mark id as STARTED if we should
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   112
        // come back during object resolution
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   113
        objects.setId(id, RESOLUTION_STARTED);
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   114
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   115
        // resolve object!
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   116
        Object resolved = resolve(value);
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   117
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   118
        // mark id as FINISHED
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   119
        objects.setId(id, RESOLUTION_FINISHED);
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   120
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   121
        // if a factory exists, convert to RecordedThread.
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   122
        // RecordedClass etc. and store back results
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   123
        if (factory != null) {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   124
            Object factorized = factory.createObject(id, resolved);
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   125
            objects.put(id, factorized);
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   126
            return factorized;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   127
        } else {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   128
            objects.put(id, resolved);
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   129
            return resolved;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   130
        }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   131
    }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   132
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   133
    private static Object resolve(Object o) {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   134
        if (o instanceof Reference) {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   135
            return resolve(((Reference) o).resolve());
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   136
        }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   137
        if (o != null && o.getClass().isArray()) {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   138
            final Object[] array = (Object[]) o;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   139
            for (int i = 0; i < array.length; i++) {
57360
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   140
                Object element = array[i];
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   141
                array[i] = resolve(element);
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   142
            }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   143
            return array;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   144
        }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   145
        return o;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   146
    }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   147
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   148
    public void resolve() {
57360
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   149
        objects.forEachKey(k -> get(k));
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   150
    }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   151
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   152
    public void put(long key, Object value) {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   153
        objects.put(key, value);
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   154
    }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   155
57360
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   156
    public void setResolving() {
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   157
        resolving = true;
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   158
        allResolved = false;
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   159
    }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   160
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   161
    public void setResolved() {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   162
        allResolved = true;
57360
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   163
        resolving = false;
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   164
    }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   165
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   166
    public String getName() {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   167
        return name;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   168
    }
57360
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   169
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   170
    public Object getResolved(long id) {
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   171
        return objects.get(id);
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   172
    }
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   173
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   174
    public void putResolved(long id, Object object) {
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   175
        objects.setId(id, RESOLUTION_FINISHED);
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   176
        objects.put(id, object);
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   177
    }
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   178
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   179
    public void setAllResolved(boolean allResolved) {
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   180
        this.allResolved = allResolved;
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   181
    }
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   182
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   183
    public boolean isResolved(long id) {
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   184
        if (objects.hasKey(id)) {
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   185
            return objects.isSetId(id, RESOLUTION_FINISHED);
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   186
        }
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   187
        return false;
5d043a159d5c Preview
egahlin
parents: 50113
diff changeset
   188
    }
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
   189
}