Пример #1
0
// Bag will fetch a random line from file and return it.
// args[0] - filename.
func Bag(scope common.Scope, args ...interface{}) interface{} {
	var err error

	filename := args[0].(string)
	if !filepath.IsAbs(filename) {
		if bagdir, _, ok := scope.GetString("_bagdir"); ok {
			filename = filepath.Join(bagdir, filename)
		} else if prodfile, _, ok := scope.GetString("_prodfile"); ok {
			dirpath := filepath.Dir(prodfile)
			filename = filepath.Join(dirpath, filename)
		}
	}
	if filename, err = filepath.Abs(filename); err != nil {
		panic(fmt.Errorf("bad filepath: %v\n", filename))
	}

	bagrw.RLock()
	records, ok := cacheBagRecords[filename]
	bagrw.RUnlock()
	if !ok {
		records = readBag(filename)
		bagrw.Lock()
		cacheBagRecords[filename] = records
		bagrw.Unlock()
	}
	if len(records) > 0 {
		rnd := scope.GetRandom()
		record := records[rnd.Intn(len(records))]
		if len(record) > 0 {
			return record[0]
		}
	}
	return ""
}
Пример #2
0
// Rangef will randomly pick a value from args[0] to args[1]
// and return the same.
// args... are expected to be in float64
func Rangef(scope common.Scope, args ...interface{}) interface{} {
	rnd := scope.GetRandom()
	if len(args) == 2 {
		min, max := args[0].(float64), args[1].(float64)
		f := (rnd.Float64() * (max - min)) + min
		return f

	} else if len(args) == 1 {
		max := args[0].(float64)
		return rnd.Float64() * max
	}
	panic(fmt.Errorf("atleast one argument expected for range-form\n"))
}
Пример #3
0
// Ranget will randomly pick a value from args[0] to args[1]
// and return the same.
// args... are expected to be in time.RFC3339 format
func Ranget(scope common.Scope, args ...interface{}) interface{} {
	rnd := scope.GetRandom()
	start, err := time.Parse(time.RFC3339, args[0].(string))
	if err != nil {
		panic(fmt.Errorf("parsing first argument %v: %v\n", args[0], err))
	}
	end, err := time.Parse(time.RFC3339, args[1].(string))
	if err != nil {
		panic(fmt.Errorf("parsing second argument %v: %v\n", args[0], err))
	}
	t := start.Add(time.Duration(rnd.Int63n(int64(end.Sub(start)))))
	return t.Format(time.RFC3339)
}
Пример #4
0
// Range will randomly pick a value from args[0] to args[1]
// and return the same.
// args... are expected to be in int64
func Range(scope common.Scope, args ...interface{}) interface{} {
	var min, max int64
	var err error

	rnd := scope.GetRandom()
	if len(args) == 2 {
		min, max = args[0].(int64), args[1].(int64)
		if err != nil {
			panic(fmt.Errorf("parsing argument %v\n", args[1]))
		}

	} else if len(args) == 1 {
		max = args[0].(int64)

	} else {
		panic(fmt.Errorf("atleast one argument expected for range-form\n"))
	}
	return rnd.Int63n(max-min) + min
}
Пример #5
0
// Choice will randomly pick one of the passed argument
// and return back.
func Choice(scope common.Scope, args ...interface{}) interface{} {
	rnd := scope.GetRandom()
	return args[rnd.Intn(len(args))]
}