Adjust Trino only setting

Author: ethanyzhang

clusters/params.json

@@ -11,5 +11,6 @@
   "native_buffer_mem_cap_gb": 32,
   "native_query_mem_percent_of_sys_mem": 0.95,
   "join_max_bcast_size_percent_of_container_mem": 0.01,
-  "memory_push_back_start_below_limit_gb": 5
+  "memory_push_back_start_below_limit_gb": 5,
+  "trino": "${PROVISION_TRINO}"
 }

clusters/templates/catalog/tpcds.properties

@@ -1,2 +1,4 @@
 connector.name=tpcds
+{{ if not (eq .GeneratorParameters.Trino "true") -}}
 tpcds.use-varchar-type=true
+{{ end -}}

cmd/genconfig/main.go

@@ -6,7 +6,7 @@ import (
 	"io/fs"
 	"os"
 	"path/filepath"
-	cluster_configs "pbench/clusters"
+	"pbench/clusters"
 	"pbench/log"
 
 	"github.com/spf13/cobra"
@@ -38,7 +38,7 @@ func Run(_ *cobra.Command, args []string) {
 		}
 	}
 	if TemplatePath == "" {
-		TemplateFS = cluster_configs.BuiltinTemplate
+		TemplateFS = clusters.BuiltinTemplate
 		TemplatePath = "templates"
 	} else {
 		TemplateFS = os.DirFS(TemplatePath)
@@ -77,5 +77,5 @@ func Run(_ *cobra.Command, args []string) {
 }
 
 func PrintDefaultParams(_ *cobra.Command, _ []string) {
-	fmt.Print(string(cluster_configs.BuiltinGeneratorParametersBytes))
+	fmt.Print(string(clusters.BuiltinGeneratorParametersBytes))
 }

cmd/genconfig/params.go

@@ -1,9 +1,8 @@
 package genconfig
 
 import (
-	_ "embed"
 	"encoding/json"
-	cluster_configs "pbench/clusters"
+	"pbench/clusters"
 )
 
 type GeneratorParameters struct {
@@ -21,12 +20,13 @@ type GeneratorParameters struct {
 	NativeQueryMemPercentOfSysMem         float64 `json:"native_query_mem_percent_of_sys_mem"`
 	JoinMaxBcastSizePercentOfContainerMem float64 `json:"join_max_bcast_size_percent_of_container_mem"`
 	MemoryPushBackStartBelowLimitGb       uint    `json:"memory_push_back_start_below_limit_gb"`
+	Trino                                 string  `json:"trino,omitempty"`
 }
 
 var DefaultGeneratorParameters = &GeneratorParameters{}
 
 func init() {
-	if err := json.Unmarshal(cluster_configs.BuiltinGeneratorParametersBytes, DefaultGeneratorParameters); err != nil {
+	if err := json.Unmarshal(clusters.BuiltinGeneratorParametersBytes, DefaultGeneratorParameters); err != nil {
 		panic(err)
 	}
 }