diff --git a/cmd/bigquery-emulator/main.go b/cmd/bigquery-emulator/main.go index 0effb880f..c52433852 100644 --- a/cmd/bigquery-emulator/main.go +++ b/cmd/bigquery-emulator/main.go @@ -7,6 +7,7 @@ import ( "net/http" "os" "os/signal" + "strings" "syscall" "github.com/goccy/bigquery-emulator/server" @@ -16,7 +17,8 @@ import ( type option struct { Project string `description:"specify the project name" long:"project"` - Dataset string `description:"specify the dataset name" long:"dataset"` + Dataset string `description:"specify the dataset name [DEPRECATED]" long:"dataset"` + Datasets string `description:"specify dataset names, separated by a comma" long:"datasets"` HTTPPort uint16 `description:"specify the http port number. this port used by bigquery api" long:"port" default:"9050"` GRPCPort uint16 `description:"specify the grpc port number. this port used by bigquery storage api" long:"grpc-port" default:"9060"` LogLevel server.LogLevel `description:"specify the log level (debug/info/warn/error)" long:"log-level" default:"error"` @@ -85,8 +87,13 @@ func runServer(args []string, opt option) error { } project := types.NewProject(opt.Project) if opt.Dataset != "" { + fmt.Println("DeprecationWarning: the flag --dataset is deprecated. use the flag --datasets instead") project.Datasets = append(project.Datasets, types.NewDataset(opt.Dataset)) } + + for _, dataset := range strings.Split(opt.Datasets, ",") { + project.Datasets = append(project.Datasets, types.NewDataset(dataset)) + } bqServer, err := server.New(db) if err != nil { return err