117 lines
3.6 KiB
Go
117 lines
3.6 KiB
Go
package formula
|
||
|
||
import (
|
||
"errors"
|
||
"strconv"
|
||
"strings"
|
||
|
||
polardbxv1 "github.com/alibaba/polardbx-operator/api/v1"
|
||
"github.com/alibaba/polardbx-operator/pkg/util/math"
|
||
)
|
||
|
||
/*
|
||
* 支持变量
|
||
|
||
AllocatedStorage：实例购买的存储空间大小，整数型
|
||
DBInstanceClassMemory：实例规格的内存大小，整数型
|
||
DBInstanceClassCPU：实例规格的CPU核数，整数型
|
||
|
||
* 支持公式
|
||
|
||
数据库参数公式支持两个运算符：除法和乘法。
|
||
除法运算符：/
|
||
用除数除以被除数，返回整数型商。商中的小数不四舍五入，直接截断。
|
||
语法
|
||
dividend / divisor
|
||
被除数和除数参数必须是整数型表达式。
|
||
乘法运算符：*
|
||
用除数除以被除数，返回整数型商。商中的小数不四舍五入，直接截断。
|
||
语法
|
||
expression * expression
|
||
两个表达式必须都是整数型。
|
||
|
||
* 支持函数
|
||
|
||
GREATEST()
|
||
返回整数型或者参数公式列表中最大的值。
|
||
语法
|
||
GREATEST(argument1, argument2,...argumentn)
|
||
返回整数。
|
||
LEAST()
|
||
返回整数型或者参数公式列表中最小的值。
|
||
语法
|
||
LEAST(argument1, argument2,...argumentn)
|
||
返回整数。
|
||
SUM()
|
||
添加指定整数型或者参数公式的值。
|
||
语法
|
||
SUM(argument1, argument2,...argumentn)
|
||
返回整数。
|
||
|
||
例如
|
||
innodb_buffer_pool_size = {DBInstanceClassMemory*3/4}
|
||
read_buffer_size = {LEAST(DBInstanceClassMemory/1048576*128, 262144)}
|
||
*/
|
||
func formulaComputing(valueStr string, memory, cpu, storage int) (int, error) {
|
||
if valueStr[len(valueStr)-1] != '}' {
|
||
return 0, errors.New("invalid format")
|
||
}
|
||
|
||
r := strings.NewReplacer("{", "", "}", "", "(", "", ")", "", " ", "")
|
||
valueStr = r.Replace(valueStr)
|
||
|
||
if strings.Contains(valueStr, "DBInstanceClassMemory") {
|
||
valueStr = strings.Replace(valueStr, "DBInstanceClassMemory", strconv.Itoa(memory), -1)
|
||
} else if strings.Contains(valueStr, "DBInstanceClassCPU") {
|
||
valueStr = strings.Replace(valueStr, "DBInstanceClassCPU", strconv.Itoa(cpu), -1)
|
||
} else if strings.Contains(valueStr, "AllocatedStorage") {
|
||
valueStr = strings.Replace(valueStr, "AllocatedStorage", strconv.Itoa(storage), -1)
|
||
} else {
|
||
return 0, errors.New("error format, Invalid VarLabel")
|
||
}
|
||
|
||
numsCalculate := strings.Split(valueStr, ",")
|
||
|
||
var result int
|
||
var err error
|
||
if len(numsCalculate) > 1 {
|
||
exists := false
|
||
for k, v := range math.Funcs {
|
||
if strings.Contains(valueStr, k) {
|
||
nums := make([]int, 0)
|
||
for _, numStr := range numsCalculate {
|
||
nums = append(nums, math.Calculate(strings.ReplaceAll(numStr, k, "")))
|
||
}
|
||
result, err = v.(func([]int) (int, error))(nums)
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
exists = true
|
||
}
|
||
}
|
||
if !exists {
|
||
return 0, errors.New("error format, Invalid Formula")
|
||
}
|
||
} else {
|
||
result = math.Calculate(numsCalculate[0])
|
||
}
|
||
|
||
return result, nil
|
||
}
|
||
|
||
func FormulaComputingPolarDBXCluster(valueStr string, polardbxcluster *polardbxv1.PolarDBXCluster) (int, error) {
|
||
memory := int(polardbxcluster.Spec.Topology.Nodes.DN.Template.Resources.Limits.Memory().Value())
|
||
cpu := int(polardbxcluster.Spec.Topology.Nodes.DN.Template.Resources.Limits.Cpu().Value())
|
||
storage := int(polardbxcluster.Spec.Topology.Nodes.DN.Template.Resources.Limits.Storage().Value())
|
||
|
||
return formulaComputing(valueStr, memory, cpu, storage)
|
||
}
|
||
|
||
func FormulaComputingXStore(valueStr string, xstore *polardbxv1.XStore) (int, error) {
|
||
memory := int(xstore.Spec.Topology.NodeSets[0].Template.Spec.Resources.Limits.Memory().Value())
|
||
cpu := int(xstore.Spec.Topology.NodeSets[0].Template.Spec.Resources.Limits.Cpu().Value())
|
||
storage := int(xstore.Spec.Topology.NodeSets[0].Template.Spec.Resources.Limits.Storage().Value())
|
||
|
||
return formulaComputing(valueStr, memory, cpu, storage)
|
||
}
|