github.com/go-graphite/carbonapi@v0.17.0/expr/functions/toUpperCase/function.go (about)

     1  package toUpperCase
     2  
     3  import (
     4  	"context"
     5  	"strings"
     6  
     7  	"github.com/go-graphite/carbonapi/expr/helper"
     8  	"github.com/go-graphite/carbonapi/expr/interfaces"
     9  	"github.com/go-graphite/carbonapi/expr/types"
    10  	"github.com/go-graphite/carbonapi/pkg/parser"
    11  )
    12  
    13  type toUpperCase struct{}
    14  
    15  func GetOrder() interfaces.Order {
    16  	return interfaces.Any
    17  }
    18  
    19  func New(configFile string) []interfaces.FunctionMetadata {
    20  	res := make([]interfaces.FunctionMetadata, 0)
    21  	f := &toUpperCase{}
    22  	functions := []string{"upper", "toUpperCase"}
    23  	for _, n := range functions {
    24  		res = append(res, interfaces.FunctionMetadata{Name: n, F: f})
    25  	}
    26  	return res
    27  }
    28  
    29  // toUpperCase(seriesList, *pos)
    30  func (f *toUpperCase) Do(ctx context.Context, eval interfaces.Evaluator, e parser.Expr, from, until int64, values map[parser.MetricRequest][]*types.MetricData) ([]*types.MetricData, error) {
    31  	args, err := helper.GetSeriesArg(ctx, eval, e.Arg(0), from, until, values)
    32  	if err != nil {
    33  		return nil, err
    34  	}
    35  
    36  	pos, err := e.GetIntArgs(1)
    37  	if err != nil {
    38  		return nil, err
    39  	}
    40  
    41  	results := make([]*types.MetricData, 0, len(args)+1)
    42  
    43  	for _, a := range args {
    44  		r := a.CopyLink()
    45  
    46  		if len(pos) == 0 {
    47  			r.Name = strings.ToUpper(a.Name)
    48  		} else {
    49  			for _, i := range pos {
    50  				if i < 0 { // Handle negative indices by indexing backwards
    51  					i = len(r.Name) + i
    52  				}
    53  				uppered := strings.ToUpper(string(r.Name[i]))
    54  				r.Name = r.Name[:i] + uppered + r.Name[i+1:]
    55  			}
    56  		}
    57  		r.Tags["name"] = r.Name
    58  		results = append(results, r)
    59  	}
    60  
    61  	return results, nil
    62  }
    63  
    64  // Description is auto-generated description, based on output of https://github.com/graphite-project/graphite-web
    65  func (f *toUpperCase) Description() map[string]types.FunctionDescription {
    66  	return map[string]types.FunctionDescription{
    67  		"upper": {
    68  			Description: "Takes one metric or a wildcard seriesList and uppers the case of each letter. \n Optionally, a letter position to upper case can be specified, in which case only the letter at the specified position gets upper-cased.\n The position parameter may be given multiple times. The position parameter may be negative to define a position relative to the end of the metric name.",
    69  			Function:    "upper(seriesList, *pos)",
    70  			Group:       "Alias",
    71  			Module:      "graphite.render.functions",
    72  			Name:        "upper",
    73  			Params: []types.FunctionParam{
    74  				{
    75  					Name:     "seriesList",
    76  					Required: true,
    77  					Type:     types.SeriesList,
    78  				},
    79  				{
    80  					Multiple: true,
    81  					Name:     "pos",
    82  					Type:     types.Node,
    83  				},
    84  			},
    85  			NameChange:   true, // name changed
    86  			ValuesChange: true, // values changed
    87  		},
    88  		"toUpperCase": {
    89  			Description: "Takes one metric or a wildcard seriesList and uppers the case of each letter. \n Optionally, a letter position to upper case can be specified, in which case only the letter at the specified position gets upper-cased.\n The position parameter may be given multiple times. The position parameter may be negative to define a position relative to the end of the metric name.",
    90  			Function:    "toUpperCase(seriesList, *pos)",
    91  			Group:       "Alias",
    92  			Module:      "graphite.render.functions",
    93  			Name:        "toUpperCase",
    94  			Params: []types.FunctionParam{
    95  				{
    96  					Name:     "seriesList",
    97  					Required: true,
    98  					Type:     types.SeriesList,
    99  				},
   100  				{
   101  					Multiple: true,
   102  					Name:     "pos",
   103  					Type:     types.Node,
   104  				},
   105  			},
   106  			NameChange:   true, // name changed
   107  			ValuesChange: true, // values changed
   108  		},
   109  	}
   110  }