performance/util.go

// Copyright 2022 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

// Utils.go contains common helper functions used across multiple other
// packages under performance folder.
package performance

import (
	"fmt"
	"math/rand"
	"strings"
	"time"
)

// Generate random integer in the range of (min, max).
func RandomInt(min, max int64) int64 {
	return min + rand.Int63n(max-min+1)
}

// Generate random float in the range of (min, max).
func RandomFloat(min, max int64) float64 {
	return float64(min) + rand.Float64()*float64(max-min)
}

const alphabet = "abcdefghijklmnopqrstuvwxyz"

// Generate random string of specific length.
func RandomString(n int) string {
	var sb strings.Builder
	k := len(alphabet)

	for i := 0; i < n; i++ {
		c := alphabet[rand.Intn(k)]
		sb.WriteByte(c)
	}

	return sb.String()
}

// Generate random date in the format: 2006-01-02
func RandomDate() string {
	min := time.Date(1970, 1, 0, 0, 0, 0, 0, time.UTC).Unix()
	max := time.Date(2010, 1, 0, 0, 0, 0, 0, time.UTC).Unix()
	delta := max - min

	sec := rand.Int63n(delta) + min
	return time.Unix(sec, 0).Format("2006-01-02")
}

func RandomBool() bool {
	i := RandomInt(0, 10)
	return i%2 == 0
}

func CurrentTimestamp() string {
	return time.Now().Format("2006-01-02 15:04:05")
}

// Generate MYSQL connection string with server, port, user, password, database specified
func GetMYSQLConnectionStr(server, port, user, password, dbName string) string {
	return fmt.Sprintf("%s:%s@tcp(%s:%s)/%s", user, password, server, port, dbName)
}