com.redhat.smonkey.RndString.java Source code

Introduction

Here is the source code for com.redhat.smonkey.RndString.java

Source

/*
 Copyright 2013 Red Hat, Inc. and/or its affiliates.
    
 This file is part of lightblue.
    
 This program is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation, either version 3 of the License, or
 (at your option) any later version.
    
 This program 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 for more details.
    
 You should have received a copy of the GNU General Public License
 along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.redhat.smonkey;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.node.JsonNodeFactory;

public class RndString implements Generator {

    public static final String DEFAULT_CHARSET = " abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789-.,";

    @Override
    public String describe() {
        return " { \"$str\": { \"minlength\": 1, \"maxlength\":32,\n"
                + "               \"length\": len, \"charset\":\"String of allowed chars\" } }\n"
                + "   Generates a random string using the given charset. The default charset includes all\n"
                + "   numeric and alphanumerics, space, and '-', '.', '.'.";
    }

    @Override
    public String getName() {
        return "$str";
    }

    @Override
    public JsonNode generate(JsonNodeFactory nodeFactory, JsonNode data, Monkey mon) {
        int minlength = 1;
        int maxlength = 32;
        String charset = DEFAULT_CHARSET;

        minlength = Utils.asInt(data.get("minlength"), minlength);
        maxlength = Utils.asInt(data.get("maxlength"), maxlength);
        int len = Utils.asInt(data.get("length"), -1);
        charset = Utils.asString(data.get("charset"), charset);

        if (len <= 0) {
            len = Utils.rndi(minlength, maxlength);
        }
        StringBuilder bld = new StringBuilder();
        for (int i = 0; i < len; i++)
            bld.append(charset.charAt(Utils.rnd.nextInt(charset.length())));
        return nodeFactory.textNode(bld.toString());
    }

}