/* * CDDL HEADER START * * The contents of this file are subject to the terms of the * Common Development and Distribution License, Version 1.0 only * (the "License"). You may not use this file except in compliance * with the License. * * You can obtain a copy of the license at * trunk/opends/resource/legal-notices/OpenDS.LICENSE * or https://OpenDS.dev.java.net/OpenDS.LICENSE. * See the License for the specific language governing permissions * and limitations under the License. * * When distributing Covered Code, include this CDDL HEADER in each * file and include the License file at * trunk/opends/resource/legal-notices/OpenDS.LICENSE. If applicable, * add the following below this CDDL HEADER, with the fields enclosed * by brackets "[]" replaced with your own identifying information: * Portions Copyright [yyyy] [name of copyright owner] * * CDDL HEADER END * * * Copyright 2006-2008 Sun Microsystems, Inc. */ package org.opends.server.tools.makeldif; import org.opends.messages.Message; import java.util.List; import java.util.Random; import org.opends.server.types.InitializationException; import static org.opends.messages.ToolMessages.*; /** * This class defines a tag that is used to indicate that a value should only be * included in a percentage of the entries. */ public class PresenceTag extends Tag { // The percentage of the entries in which this attribute value should appear. private int percentage; // The random number generator for this tag. private Random random; /** * Creates a new instance of this presence tag. */ public PresenceTag() { percentage = 100; } /** * Retrieves the name for this tag. * * @return The name for this tag. */ public String getName() { return "Presence"; } /** * Indicates whether this tag is allowed for use in the extra lines for * branches. * * @return <CODE>true</CODE> if this tag may be used in branch definitions, * or <CODE>false</CODE> if not. */ public boolean allowedInBranch() { return true; } /** * Performs any initialization for this tag that may be needed while parsing * a branch definition. * * @param templateFile The template file in which this tag is used. * @param branch The branch in which this tag is used. * @param arguments The set of arguments provided for this tag. * @param lineNumber The line number on which this tag appears in the * template file. * @param warnings A list into which any appropriate warning messages * may be placed. * * @throws InitializationException If a problem occurs while initializing * this tag. */ public void initializeForBranch(TemplateFile templateFile, Branch branch, String[] arguments, int lineNumber, List<Message> warnings) throws InitializationException { initializeInternal(templateFile, arguments, lineNumber); } /** * Performs any initialization for this tag that may be needed while parsing * a template definition. * * @param templateFile The template file in which this tag is used. * @param template The template in which this tag is used. * @param arguments The set of arguments provided for this tag. * @param lineNumber The line number on which this tag appears in the * template file. * @param warnings A list into which any appropriate warning messages * may be placed. * * @throws InitializationException If a problem occurs while initializing * this tag. */ public void initializeForTemplate(TemplateFile templateFile, Template template, String[] arguments, int lineNumber, List<Message> warnings) throws InitializationException { initializeInternal(templateFile, arguments, lineNumber); } /** * Performs any initialization for this tag that may be needed for this tag. * * @param templateFile The template file in which this tag is used. * @param arguments The set of arguments provided for this tag. * @param lineNumber The line number on which this tag appears in the * template file. * * @throws InitializationException If a problem occurs while initializing * this tag. */ private void initializeInternal(TemplateFile templateFile, String[] arguments, int lineNumber) throws InitializationException { random = templateFile.getRandom(); if (arguments.length != 1) { Message message = ERR_MAKELDIF_TAG_INVALID_ARGUMENT_COUNT.get( getName(), lineNumber, 1, arguments.length); throw new InitializationException(message); } try { percentage = Integer.parseInt(arguments[0]); if (percentage < 0) { Message message = ERR_MAKELDIF_TAG_INTEGER_BELOW_LOWER_BOUND.get( percentage, 0, getName(), lineNumber); throw new InitializationException(message); } else if (percentage > 100) { Message message = ERR_MAKELDIF_TAG_INTEGER_ABOVE_UPPER_BOUND.get( percentage, 100, getName(), lineNumber); throw new InitializationException(message); } } catch (NumberFormatException nfe) { Message message = ERR_MAKELDIF_TAG_CANNOT_PARSE_AS_INTEGER.get( arguments[0], getName(), lineNumber); throw new InitializationException(message); } } /** * Generates the content for this tag by appending it to the provided tag. * * @param templateEntry The entry for which this tag is being generated. * @param templateValue The template value to which the generated content * should be appended. * * @return The result of generating content for this tag. */ public TagResult generateValue(TemplateEntry templateEntry, TemplateValue templateValue) { int intValue = random.nextInt(100); if (intValue < percentage) { return TagResult.SUCCESS_RESULT; } else { return TagResult.OMIT_FROM_ENTRY; } } }