Function GenerateOptimizeCode

go/codegen/optimize/codegen.go:80–158

GenerateOptimizeCode generates optimize codes for execution

(optimStmt *ir.OptimizeStmt, session *pb.Session, tableName string, useOptFlow bool)

// GenerateOptimizeCode generates optimize codes for execution
func GenerateOptimizeCode(optimStmt *ir.OptimizeStmt, session *pb.Session, tableName string, useOptFlow bool) (string, error) {
	const (
		optimizeTemplateName = "optimize"
	)

	dbName, err := database.GetDatabaseName(session.DbConnStr)
	if err != nil {
		return "", err
	}

	resultTable := optimStmt.ResultTable
	if !strings.Contains(resultTable, ".") {
		resultTable = fmt.Sprintf("%s.%s", dbName, resultTable)
	}

	attrJSON, err := generateOptimizeAttributeJSONString(optimStmt.Attributes)
	if err != nil {
		return "", err
	}

	if !useOptFlow {
		filler := pyomoNativeOptimizeFiller{
			DataSource:      session.DbConnStr,
			Select:          optimStmt.Select,
			Variables:       optimStmt.Variables,
			ResultValueName: optimStmt.ResultValueName,
			VariableType:    optimStmt.VariableType,
			Objective:       optimStmt.Objective,
			Direction:       optimStmt.Direction,
			Constraints:     optimStmt.Constraints,
			Solver:          optimStmt.Solver,
			AttributeJSON:   attrJSON,
			ResultTable:     resultTable,
		}
		tpl := template.Must(template.New(optimizeTemplateName).Parse(pyomoNativeOptimizeText))
		var program bytes.Buffer
		if err := tpl.Execute(&program, filler); err != nil {
			return "", err
		}
		return program.String(), nil
	}

	if !strings.Contains(tableName, ".") {
		tableName = fmt.Sprintf("%s.%s", dbName, tableName)
	}

	db, err := database.OpenAndConnectDB(session.DbConnStr)
	if err != nil {
		return "", err
	}
	defer db.Close()

	columns, _, err := verifier.GetSQLFieldType(fmt.Sprintf("SELECT * FROM %s", tableName), db)
	if err != nil {
		return "", err
	}

	filler := &optFlowOptimizeFiller{