mirror of https://github.com/OpenIdentityPlatform/OpenDJ.git

Jean-Noël Rouvignac
21.30.2016 66f58163fc4828fd9c776e89fbb60ffb60be6633
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
/*
 * The contents of this file are subject to the terms of the Common Development and
 * Distribution License (the License). You may not use this file except in compliance with the
 * License.
 *
 * You can obtain a copy of the License at legal/CDDLv1.0.txt. See the License for the
 * specific language governing permission and limitations under the License.
 *
 * When distributing Covered Software, include this CDDL Header Notice in each file and include
 * the License file at legal/CDDLv1.0.txt. If applicable, add the following below the CDDL
 * Header, with the fields enclosed by brackets [] replaced by your own identifying
 * information: "Portions Copyright [year] [name of copyright owner]".
 *
 * Copyright 2016 ForgeRock AS.
 */
package org.opends.server.api;
 
import static org.opends.server.core.DirectoryServer.*;
 
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.Collection;
import java.util.Date;
import java.util.Iterator;
import java.util.List;
import java.util.UUID;
 
import org.forgerock.opendj.ldap.ByteString;
import org.forgerock.opendj.ldap.DN;
import org.forgerock.opendj.ldap.GeneralizedTime;
import org.forgerock.opendj.ldap.schema.AttributeType;
import org.forgerock.opendj.ldap.schema.CoreSchema;
import org.forgerock.opendj.ldap.schema.Syntax;
import org.opends.server.types.Attribute;
import org.opends.server.types.Attribute.RemoveOnceSwitchingAttributes;
import org.opends.server.types.AttributeBuilder;
import org.opends.server.types.Attributes;
import org.opends.server.types.PublicAPI;
import org.opends.server.types.StabilityLevel;
 
/**
 * This class is used to hold monitoring data, i.e. a list of attributes. It provides convenient
 * methods to easily build such data.
 * <p>
 * <strong>Note:</strong> <br>
 * Creating monitor entries may become a lot easier once we've migrated to the SDK Entry class:
 *
 * <pre>
 * Entry entry = ...;
 * entry.addAttribute("stringStat", "aString")
 *       .addAttribute("integerStat", 12345)
 *       .addAttribute("dnStat", DN.valueOf("dc=aDN");
 * </pre>
 *
 * We could also envisage an annotation based approach where we determine the monitor content from
 * annotated fields/methods in an object.
 */
@PublicAPI(stability = StabilityLevel.PRIVATE)
public final class MonitorData implements Iterable<Attribute>
{
  private final List<Attribute> attrs;
 
  /** Constructor to use when the number of attributes to create is unknown. */
  public MonitorData()
  {
    attrs = new ArrayList<>();
  }
 
  /**
   * Constructor that accepts the number of attributes to create.
   *
   * @param expectedAttributesCount
   *          number of attributes that will be added
   */
  public MonitorData(int expectedAttributesCount)
  {
    attrs = new ArrayList<>(expectedAttributesCount);
  }
 
  /**
   * Adds an attribute with the provided name and value.
   *
   * @param attrName
   *          the attribute name
   * @param attrValue
   *          the attribute value
   */
  public void add(String attrName, Object attrValue)
  {
    Syntax syntax;
    if (attrValue instanceof String
        || attrValue instanceof ByteString
        || attrValue instanceof Float
        || attrValue instanceof Double)
    {
      // coming first because they are the most common types
      syntax = CoreSchema.getDirectoryStringSyntax();
    }
    else if (attrValue instanceof Number)
    {
      syntax = CoreSchema.getIntegerSyntax();
    }
    else if (attrValue instanceof Boolean)
    {
      syntax = CoreSchema.getBooleanSyntax();
    }
    else if (attrValue instanceof DN)
    {
      syntax = CoreSchema.getDNSyntax();
    }
    else if (attrValue instanceof Date)
    {
      syntax = CoreSchema.getGeneralizedTimeSyntax();
      attrValue = GeneralizedTime.valueOf((Date) attrValue);
    }
    else if (attrValue instanceof Calendar)
    {
      syntax = CoreSchema.getGeneralizedTimeSyntax();
      attrValue = GeneralizedTime.valueOf((Calendar) attrValue);
    }
    else if (attrValue instanceof UUID)
    {
      syntax = CoreSchema.getUUIDSyntax();
    }
    else
    {
      syntax = CoreSchema.getDirectoryStringSyntax();
    }
    add(attrName, syntax, attrValue);
  }
 
  private void add(String attrName, Syntax syntax, Object attrValue)
  {
    AttributeType attrType = getSchema().getAttributeType(attrName, syntax);
    attrs.add(Attributes.create(attrType, String.valueOf(attrValue)));
  }
 
  /**
   * Adds an attribute with the provided name and value if the value is not null.
   *
   * @param attrName
   *          the attribute name
   * @param attrValue
   *          the attribute value
   */
  public void addIfNotNull(String attrName, Object attrValue)
  {
    if (attrValue != null)
    {
      add(attrName, attrValue);
    }
  }
 
  /**
   * Adds an attribute with the provided name and values.
   *
   * @param attrName
   *          the attribute name
   * @param attrValues
   *          the attribute values
   */
  @RemoveOnceSwitchingAttributes(comment = "once using the non immutable SDK's Attribute class, "
      + "we can incrementally build an attribute by using the add(String attrName, Object attrValue) method")
  public void add(String attrName, Collection<?> attrValues)
  {
    AttributeBuilder builder = new AttributeBuilder(attrName);
    builder.addAllStrings(attrValues);
    attrs.add(builder.toAttribute());
  }
 
  /**
   * Adds all the properties from the provided bean as attributes, prepending the provided prefix.
   *
   * @param bean
   *          the bean from which to read the properties
   * @param attributesPrefix
   *          the prefix to prepend to the attributes read from the bean
   * @throws ReflectiveOperationException
   *           if a problem occurs while reading the properties of the bean
   */
  public void addBean(Object bean, String attributesPrefix) throws ReflectiveOperationException
  {
    for (Method method : bean.getClass().getMethods())
    {
      if (method.getName().startsWith("get"))
      {
        Class<?> returnType = method.getReturnType();
        if (returnType.equals(int.class) || returnType.equals(long.class) || returnType.equals(String.class))
        {
          addStatAttribute(attributesPrefix, bean, method, 3);
        }
      }
      else if (method.getName().startsWith("is") && method.getReturnType().equals(boolean.class))
      {
        addStatAttribute(attributesPrefix, bean, method, 2);
      }
    }
  }
 
  private void addStatAttribute(String attrPrefix, Object stats, Method method, int skipNameLen)
      throws ReflectiveOperationException
  {
    String attrName = attrPrefix + method.getName().substring(skipNameLen);
    add(attrName, method.invoke(stats));
  }
 
  @Override
  public Iterator<Attribute> iterator()
  {
    return attrs.iterator();
  }
 
  /**
   * Returns the number of attributes.
   *
   * @return the number of attributes
   */
  public int size()
  {
    return attrs.size();
  }
 
  @Override
  public String toString()
  {
    return getClass().getSimpleName() + attrs;
  }
}