Allow sql_driver: "github.com/go-sql-driver/mysql"

In preparation of sqlc-dev#2179 for LOAD DATA LOCAL INFILE
Jille · Apr 21, 2023 · 378ecab · 378ecab
1 parent e99f1db
commit 378ecab
Show file tree

Hide file tree

Showing 7 changed files with 228 additions and 164 deletions.
diff --git a/internal/cmd/shim.go b/internal/cmd/shim.go
@@ -96,6 +96,7 @@ func pluginGoCode(s config.SQLGo) *plugin.GoCode {
 		EmitAllEnumValues:           s.EmitAllEnumValues,
 		JsonTagsCaseStyle:           s.JSONTagsCaseStyle,
 		Package:                     s.Package,
+		Driver:                      s.Driver,
 		Out:                         s.Out,
 		SqlPackage:                  s.SQLPackage,
 		OutputDbFileName:            s.OutputDBFileName,

diff --git a/internal/codegen/golang/driver.go b/internal/codegen/golang/driver.go
@@ -14,6 +14,8 @@ const (
 	SQLDriverLibPQ
 )
 
+const SQLDriverGoSQLDriverMySQL = "github.com/go-sql-driver/mysql"
+
 func parseDriver(sqlPackage string) SQLDriver {
 	switch sqlPackage {
 	case SQLPackagePGXV4:

diff --git a/internal/config/config.go b/internal/config/config.go
@@ -135,6 +135,7 @@ type SQLGo struct {
 	Overrides                   []Override        `json:"overrides,omitempty" yaml:"overrides"`
 	Rename                      map[string]string `json:"rename,omitempty" yaml:"rename"`
 	SQLPackage                  string            `json:"sql_package" yaml:"sql_package"`
+	SQLDriver                   string            `json:"sql_driver" yaml:"sql_driver"`
 	OutputBatchFileName         string            `json:"output_batch_file_name,omitempty" yaml:"output_batch_file_name"`
 	OutputDBFileName            string            `json:"output_db_file_name,omitempty" yaml:"output_db_file_name"`
 	OutputModelsFileName        string            `json:"output_models_file_name,omitempty" yaml:"output_models_file_name"`

diff --git a/internal/config/v_one.go b/internal/config/v_one.go
@@ -38,6 +38,7 @@ type v1PackageSettings struct {
 	EmitAllEnumValues         bool       `json:"emit_all_enum_values,omitempty" yaml:"emit_all_enum_values"`
 	JSONTagsCaseStyle         string     `json:"json_tags_case_style,omitempty" yaml:"json_tags_case_style"`
 	SQLPackage                string     `json:"sql_package" yaml:"sql_package"`
+	SQLDriver                 string     `json:"sql_driver" yaml:"sql_driver"`
 	Overrides                 []Override `json:"overrides" yaml:"overrides"`
 	OutputBatchFileName       string     `json:"output_batch_file_name,omitempty" yaml:"output_batch_file_name"`
 	OutputDBFileName          string     `json:"output_db_file_name,omitempty" yaml:"output_db_file_name"`
@@ -151,6 +152,7 @@ func (c *V1GenerateSettings) Translate() Config {
 					Package:                   pkg.Name,
 					Out:                       pkg.Path,
 					SQLPackage:                pkg.SQLPackage,
+					SQLDriver:                 pkg.SQLDriver,
 					Overrides:                 pkg.Overrides,
 					JSONTagsCaseStyle:         pkg.JSONTagsCaseStyle,
 					OutputBatchFileName:       pkg.OutputBatchFileName,

diff --git a/internal/plugin/codegen.pb.go b/internal/plugin/codegen.pb.go
diff --git a/internal/plugin/codegen_vtproto.pb.go b/internal/plugin/codegen_vtproto.pb.go
diff --git a/protos/plugin/codegen.proto b/protos/plugin/codegen.proto
@@ -82,6 +82,7 @@ message GoCode
   bool emit_methods_with_db_argument = 10;
   string json_tags_case_style = 11;
   string package = 12;
+  string driver = 25;
   string out = 13;
   string sql_package = 14;
   string output_db_file_name = 15;