Add Sqlserver database support #51
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,208 @@ | ||
| package database | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "strings" | ||
|
|
||
| "github.com/fraenky8/tables-to-go/pkg/settings" | ||
|
|
||
| // sqlserver database driver | ||
| _ "github.com/denisenkom/go-mssqldb" | ||
| ) | ||
|
|
||
| // Sqlserver implements the Database interface with help of GeneralDatabase. | ||
| type Sqlserver struct { | ||
|
Comment on lines
+13
to
+14
Owner
There was a problem hiding this comment. Lets use consistently |
||
| *GeneralDatabase | ||
|
|
||
| defaultUserName string | ||
| } | ||
|
|
||
| // NewSqlserver creates a new Sqlserver database. | ||
| func NewSqlserver(s *settings.Settings) *Sqlserver { | ||
| return &Sqlserver{ | ||
| GeneralDatabase: &GeneralDatabase{ | ||
| Settings: s, | ||
| driver: dbTypeToDriverMap[s.DbType], | ||
| }, | ||
| defaultUserName: "sa", | ||
| } | ||
| } | ||
|
|
||
| // Connect connects to the database by the given data source name (dsn) of the | ||
| // concrete database. | ||
| func (ss *Sqlserver) Connect() error { | ||
| return ss.GeneralDatabase.Connect(ss.DSN()) | ||
| } | ||
|
|
||
| // DSN creates the DSN String to connect to this database. | ||
| func (ss *Sqlserver) DSN() string { | ||
|
Owner
There was a problem hiding this comment. Lets add a unit test for that? I see the other database implementation have one, just for the sake of consistency? |
||
| user := ss.defaultUserName | ||
| if ss.Settings.User != "" { | ||
| user = ss.Settings.User | ||
| } | ||
| return fmt.Sprintf("server=%s;port=%s;user=%s;database=%s;password=%s", | ||
|
Owner
There was a problem hiding this comment. Is there a way to connect via a socket to SQLServer? I see MySQL and PG have this option. Can you check and add it here as well? |
||
| ss.Settings.Host, ss.Settings.Port, user, ss.Settings.DbName, ss.Settings.Pswd) | ||
| } | ||
|
|
||
| // GetTables gets all tables for a given schema by name. | ||
| func (ss *Sqlserver) GetTables() (tables []*Table, err error) { | ||
|
|
||
| err = ss.Select(&tables, ` | ||
| SELECT | ||
| table_name | ||
| FROM | ||
| INFORMATION_SCHEMA.TABLES | ||
| WHERE | ||
| TABLE_SCHEMA = @p1 | ||
| ORDER BY TABLE_NAME | ||
| `, ss.Schema) | ||
|
|
||
| if ss.Verbose { | ||
| if err != nil { | ||
| fmt.Println("> Error at GetTables()") | ||
| fmt.Printf("> schema: %q\r\n", ss.Schema) | ||
| } | ||
| } | ||
|
|
||
| return tables, err | ||
| } | ||
|
|
||
| // PrepareGetColumnsOfTableStmt prepares the statement for retrieving the | ||
| // columns of a specific table for a given database. | ||
| func (ss *Sqlserver) PrepareGetColumnsOfTableStmt() (err error) { | ||
|
|
||
| ss.GetColumnsOfTableStmt, err = ss.Preparex(` | ||
|
Owner
There was a problem hiding this comment. Just to be double sure, since I don't have any experience with SQLServer - this statement is the same as for PG except the placeholder format? |
||
| SELECT | ||
| ic.ordinal_position, | ||
| ic.column_name, | ||
| ic.data_type, | ||
| ic.column_default, | ||
| ic.is_nullable, | ||
| ic.character_maximum_length, | ||
| ic.numeric_precision, | ||
| itc.constraint_name, | ||
| itc.constraint_type | ||
| FROM information_schema.columns AS ic | ||
| LEFT JOIN information_schema.key_column_usage AS ikcu ON ic.table_name = ikcu.table_name | ||
| AND ic.table_schema = ikcu.table_schema | ||
| AND ic.column_name = ikcu.column_name | ||
| LEFT JOIN information_schema.table_constraints AS itc ON ic.table_name = itc.table_name | ||
| AND ic.table_schema = itc.table_schema | ||
| AND ikcu.constraint_name = itc.constraint_name | ||
| WHERE ic.table_name = @p1 | ||
| AND ic.table_schema = @p2 | ||
| ORDER BY ic.ordinal_position | ||
| `) | ||
|
|
||
| return err | ||
| } | ||
|
|
||
| // GetColumnsOfTable executes the statement for retrieving the columns of a | ||
| // specific table in a given schema. | ||
| func (ss *Sqlserver) GetColumnsOfTable(table *Table) (err error) { | ||
|
|
||
| err = ss.GetColumnsOfTableStmt.Select(&table.Columns, table.Name, ss.Schema) | ||
|
|
||
| if ss.Verbose { | ||
| if err != nil { | ||
| fmt.Printf("> Error at GetColumnsOfTable(%v)\r\n", table.Name) | ||
| fmt.Printf("> schema: %q\r\n", ss.Schema) | ||
| } | ||
| } | ||
|
|
||
| return err | ||
| } | ||
|
|
||
| // IsPrimaryKey checks if the column belongs to the primary key. | ||
| func (ss *Sqlserver) IsPrimaryKey(column Column) bool { | ||
| return strings.Contains(column.ConstraintType.String, "PRIMARY KEY") | ||
| } | ||
|
|
||
| // IsAutoIncrement checks if the column is an auto_increment column. | ||
| func (ss *Sqlserver) IsAutoIncrement(column Column) bool { | ||
| return strings.Contains(column.DefaultValue.String, "IDENTITY") | ||
| } | ||
|
|
||
| // GetStringDatatypes returns the string datatypes for the Sqlserver database. | ||
| // https://learn.microsoft.com/en-us/sql/t-sql/data-types/data-types-transact-sql?view=sql-server-ver16#character-strings | ||
| func (ss *Sqlserver) GetStringDatatypes() []string { | ||
| return []string{ | ||
| "char", | ||
| "varchar", | ||
| "nchar", | ||
| "nvarchar", | ||
| } | ||
| } | ||
|
|
||
| // IsString returns true if colum is of type string for the Sqlserver database. | ||
| func (ss *Sqlserver) IsString(column Column) bool { | ||
| return isStringInSlice(column.DataType, ss.GetStringDatatypes()) | ||
| } | ||
|
|
||
| // GetTextDatatypes returns the text datatypes for the Sqlserver database. | ||
| // https://learn.microsoft.com/en-us/sql/t-sql/data-types/data-types-transact-sql?view=sql-server-ver16#character-strings | ||
| func (ss *Sqlserver) GetTextDatatypes() []string { | ||
| return []string{ | ||
| "text", | ||
| "ntext", | ||
| "binary", | ||
| "varbinary", | ||
| "image", | ||
| } | ||
| } | ||
|
|
||
| // IsText returns true if colum is of type text for the Sqlserver database. | ||
| func (ss *Sqlserver) IsText(column Column) bool { | ||
| return isStringInSlice(column.DataType, ss.GetTextDatatypes()) | ||
| } | ||
|
|
||
| // GetIntegerDatatypes returns the integer datatypes for the Sqlserver database. | ||
| // https://learn.microsoft.com/en-us/sql/t-sql/data-types/data-types-transact-sql?view=sql-server-ver16#exact-numerics | ||
| func (ss *Sqlserver) GetIntegerDatatypes() []string { | ||
| return []string{ | ||
| "bigint", | ||
| "bit", | ||
| "smallint", | ||
| "int", | ||
| "tinyint", | ||
| } | ||
| } | ||
|
|
||
| // IsInteger returns true if colum is of type integer for the Sqlserver database. | ||
| func (ss *Sqlserver) IsInteger(column Column) bool { | ||
| return isStringInSlice(column.DataType, ss.GetIntegerDatatypes()) | ||
| } | ||
|
|
||
| // GetFloatDatatypes returns the float datatypes for the Sqlserver database. | ||
| // https://learn.microsoft.com/en-us/sql/t-sql/data-types/data-types-transact-sql?view=sql-server-ver16#exact-numerics | ||
| func (ss *Sqlserver) GetFloatDatatypes() []string { | ||
| return []string{ | ||
| "numeric", | ||
| "decimal", | ||
| "float", | ||
| "real", | ||
| } | ||
| } | ||
|
|
||
| // IsFloat returns true if colum is of type float for the Sqlserver database. | ||
| func (ss *Sqlserver) IsFloat(column Column) bool { | ||
| return isStringInSlice(column.DataType, ss.GetFloatDatatypes()) | ||
| } | ||
|
|
||
| // GetTemporalDatatypes returns the temporal datatypes for the Sqlserver database. | ||
| // https://learn.microsoft.com/en-us/sql/t-sql/data-types/data-types-transact-sql?view=sql-server-ver16#date-and-time | ||
| func (ss *Sqlserver) GetTemporalDatatypes() []string { | ||
| return []string{ | ||
| "date", | ||
| "datetimeoffset", | ||
| "datetime2", | ||
| "smalldatetime", | ||
| "datetime", | ||
| "time", | ||
| } | ||
| } | ||
|
|
||
| // IsTemporal returns true if colum is of type temporal for the Sqlserver database. | ||
| func (ss *Sqlserver) IsTemporal(column Column) bool { | ||
| return isStringInSlice(column.DataType, ss.GetTemporalDatatypes()) | ||
| } | ||
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Can you run
go mod tidyandgo mod vendorto include the vendor changes?