mirror of
				https://github.com/go-gitea/gitea.git
				synced 2025-10-26 12:27:06 +00:00 
			
		
		
		
	Upgrade gopkg.in/testfixtures.v2 (#4999)
This commit is contained in:
		
							
								
								
									
										6
									
								
								Gopkg.lock
									
									
									
										generated
									
									
									
								
							
							
						
						
									
										6
									
								
								Gopkg.lock
									
									
									
										generated
									
									
									
								
							| @@ -885,12 +885,12 @@ | |||||||
|   version = "v2.3.2" |   version = "v2.3.2" | ||||||
|  |  | ||||||
| [[projects]] | [[projects]] | ||||||
|   digest = "1:e2144032dcf8e856fb733151391669dc2d32d79ebb69e3c4151efa605f5e8a01" |   digest = "1:9c541fc507676a69ea8aaed1af53278a5241d26ce0f192c993fec2ac5b78f795" | ||||||
|   name = "gopkg.in/testfixtures.v2" |   name = "gopkg.in/testfixtures.v2" | ||||||
|   packages = ["."] |   packages = ["."] | ||||||
|   pruneopts = "NUT" |   pruneopts = "NUT" | ||||||
|   revision = "b9ef14dc461bf934d8df2dfc6f1f456be5664cca" |   revision = "fa3fb89109b0b31957a5430cef3e93e535de362b" | ||||||
|   version = "v2.0.0" |   version = "v2.5.0" | ||||||
|  |  | ||||||
| [[projects]] | [[projects]] | ||||||
|   digest = "1:ad6f94355d292690137613735965bd3688844880fdab90eccf66321910344942" |   digest = "1:ad6f94355d292690137613735965bd3688844880fdab90eccf66321910344942" | ||||||
|   | |||||||
							
								
								
									
										28
									
								
								vendor/gopkg.in/testfixtures.v2/deprecated.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										28
									
								
								vendor/gopkg.in/testfixtures.v2/deprecated.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -5,22 +5,38 @@ import ( | |||||||
| ) | ) | ||||||
|  |  | ||||||
| type ( | type ( | ||||||
| 	DataBaseHelper Helper // Deprecated: Use Helper instead | 	// DataBaseHelper is the helper interface | ||||||
|  | 	// Deprecated: Use Helper instead | ||||||
|  | 	DataBaseHelper Helper | ||||||
|  |  | ||||||
| 	PostgreSQLHelper struct { // Deprecated: Use PostgreSQL{} instead | 	// PostgreSQLHelper is the PostgreSQL helper | ||||||
|  | 	// Deprecated: Use PostgreSQL{} instead | ||||||
|  | 	PostgreSQLHelper struct { | ||||||
| 		PostgreSQL | 		PostgreSQL | ||||||
| 		UseAlterConstraint bool | 		UseAlterConstraint bool | ||||||
| 	} | 	} | ||||||
| 	MySQLHelper struct { // Deprecated: Use MySQL{} instead |  | ||||||
|  | 	// MySQLHelper is the MySQL helper | ||||||
|  | 	// Deprecated: Use MySQL{} instead | ||||||
|  | 	MySQLHelper struct { | ||||||
| 		MySQL | 		MySQL | ||||||
| 	} | 	} | ||||||
| 	SQLiteHelper struct { // Deprecated: Use SQLite{} instead |  | ||||||
|  | 	// SQLiteHelper is the SQLite helper | ||||||
|  | 	// Deprecated: Use SQLite{} instead | ||||||
|  | 	SQLiteHelper struct { | ||||||
| 		SQLite | 		SQLite | ||||||
| 	} | 	} | ||||||
| 	SQLServerHelper struct { // Deprecated: Use SQLServer{} instead |  | ||||||
|  | 	// SQLServerHelper is the SQLServer helper | ||||||
|  | 	// Deprecated: Use SQLServer{} instead | ||||||
|  | 	SQLServerHelper struct { | ||||||
| 		SQLServer | 		SQLServer | ||||||
| 	} | 	} | ||||||
| 	OracleHelper struct { // Deprecated: Use Oracle{} instead |  | ||||||
|  | 	// OracleHelper is the Oracle helper | ||||||
|  | 	// Deprecated: Use Oracle{} instead | ||||||
|  | 	OracleHelper struct { | ||||||
| 		Oracle | 		Oracle | ||||||
| 	} | 	} | ||||||
| ) | ) | ||||||
|   | |||||||
							
								
								
									
										41
									
								
								vendor/gopkg.in/testfixtures.v2/errors.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										41
									
								
								vendor/gopkg.in/testfixtures.v2/errors.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,41 @@ | |||||||
|  | package testfixtures | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"errors" | ||||||
|  | 	"fmt" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | var ( | ||||||
|  | 	// ErrWrongCastNotAMap is returned when a map is not a map[interface{}]interface{} | ||||||
|  | 	ErrWrongCastNotAMap = errors.New("Could not cast record: not a map[interface{}]interface{}") | ||||||
|  |  | ||||||
|  | 	// ErrFileIsNotSliceOrMap is returned the the fixture file is not a slice or map. | ||||||
|  | 	ErrFileIsNotSliceOrMap = errors.New("The fixture file is not a slice or map") | ||||||
|  |  | ||||||
|  | 	// ErrKeyIsNotString is returned when a record is not of type string | ||||||
|  | 	ErrKeyIsNotString = errors.New("Record map key is not string") | ||||||
|  |  | ||||||
|  | 	// ErrNotTestDatabase is returned when the database name doesn't contains "test" | ||||||
|  | 	ErrNotTestDatabase = errors.New(`Loading aborted because the database name does not contains "test"`) | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // InsertError will be returned if any error happens on database while | ||||||
|  | // inserting the record | ||||||
|  | type InsertError struct { | ||||||
|  | 	Err    error | ||||||
|  | 	File   string | ||||||
|  | 	Index  int | ||||||
|  | 	SQL    string | ||||||
|  | 	Params []interface{} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (e *InsertError) Error() string { | ||||||
|  | 	return fmt.Sprintf( | ||||||
|  | 		"testfixtures: error inserting record: %v, on file: %s, index: %d, sql: %s, params: %v", | ||||||
|  | 		e.Err, | ||||||
|  | 		e.File, | ||||||
|  | 		e.Index, | ||||||
|  | 		e.SQL, | ||||||
|  | 		e.Params, | ||||||
|  | 	) | ||||||
|  | } | ||||||
							
								
								
									
										110
									
								
								vendor/gopkg.in/testfixtures.v2/generate.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										110
									
								
								vendor/gopkg.in/testfixtures.v2/generate.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,110 @@ | |||||||
|  | package testfixtures | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"database/sql" | ||||||
|  | 	"fmt" | ||||||
|  | 	"os" | ||||||
|  | 	"path" | ||||||
|  | 	"unicode/utf8" | ||||||
|  |  | ||||||
|  | 	"gopkg.in/yaml.v2" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | // TableInfo is settings for generating a fixture for table. | ||||||
|  | type TableInfo struct { | ||||||
|  | 	Name  string // Table name | ||||||
|  | 	Where string // A condition for extracting records. If this value is empty, extracts all records. | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (ti *TableInfo) whereClause() string { | ||||||
|  | 	if ti.Where == "" { | ||||||
|  | 		return "" | ||||||
|  | 	} | ||||||
|  | 	return fmt.Sprintf(" WHERE %s", ti.Where) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // GenerateFixtures generates fixtures for the current contents of a database, and saves | ||||||
|  | // them to the specified directory | ||||||
|  | func GenerateFixtures(db *sql.DB, helper Helper, dir string) error { | ||||||
|  | 	tables, err := helper.tableNames(db) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	for _, table := range tables { | ||||||
|  | 		filename := path.Join(dir, table+".yml") | ||||||
|  | 		if err := generateFixturesForTable(db, helper, &TableInfo{Name: table}, filename); err != nil { | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // GenerateFixturesForTables generates fixtures for the current contents of specified tables in a database, and saves | ||||||
|  | // them to the specified directory | ||||||
|  | func GenerateFixturesForTables(db *sql.DB, tables []*TableInfo, helper Helper, dir string) error { | ||||||
|  | 	for _, table := range tables { | ||||||
|  | 		filename := path.Join(dir, table.Name+".yml") | ||||||
|  | 		if err := generateFixturesForTable(db, helper, table, filename); err != nil { | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func generateFixturesForTable(db *sql.DB, h Helper, table *TableInfo, filename string) error { | ||||||
|  | 	query := fmt.Sprintf("SELECT * FROM %s%s", h.quoteKeyword(table.Name), table.whereClause()) | ||||||
|  | 	rows, err := db.Query(query) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	defer rows.Close() | ||||||
|  |  | ||||||
|  | 	columns, err := rows.Columns() | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	fixtures := make([]interface{}, 0, 10) | ||||||
|  | 	for rows.Next() { | ||||||
|  | 		entries := make([]interface{}, len(columns)) | ||||||
|  | 		entryPtrs := make([]interface{}, len(entries)) | ||||||
|  | 		for i := range entries { | ||||||
|  | 			entryPtrs[i] = &entries[i] | ||||||
|  | 		} | ||||||
|  | 		if err := rows.Scan(entryPtrs...); err != nil { | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		entryMap := make(map[string]interface{}, len(entries)) | ||||||
|  | 		for i, column := range columns { | ||||||
|  | 			entryMap[column] = convertValue(entries[i]) | ||||||
|  | 		} | ||||||
|  | 		fixtures = append(fixtures, entryMap) | ||||||
|  | 	} | ||||||
|  | 	if err = rows.Err(); err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	f, err := os.Create(filename) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	defer f.Close() | ||||||
|  |  | ||||||
|  | 	marshaled, err := yaml.Marshal(fixtures) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	_, err = f.Write(marshaled) | ||||||
|  | 	return err | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func convertValue(value interface{}) interface{} { | ||||||
|  | 	switch v := value.(type) { | ||||||
|  | 	case []byte: | ||||||
|  | 		if utf8.Valid(v) { | ||||||
|  | 			return string(v) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 	return value | ||||||
|  | } | ||||||
							
								
								
									
										43
									
								
								vendor/gopkg.in/testfixtures.v2/helper.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										43
									
								
								vendor/gopkg.in/testfixtures.v2/helper.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -18,21 +18,56 @@ type Helper interface { | |||||||
| 	init(*sql.DB) error | 	init(*sql.DB) error | ||||||
| 	disableReferentialIntegrity(*sql.DB, loadFunction) error | 	disableReferentialIntegrity(*sql.DB, loadFunction) error | ||||||
| 	paramType() int | 	paramType() int | ||||||
| 	databaseName(*sql.DB) string | 	databaseName(queryable) (string, error) | ||||||
|  | 	tableNames(queryable) ([]string, error) | ||||||
|  | 	isTableModified(queryable, string) (bool, error) | ||||||
|  | 	afterLoad(queryable) error | ||||||
| 	quoteKeyword(string) string | 	quoteKeyword(string) string | ||||||
| 	whileInsertOnTable(*sql.Tx, string, func() error) error | 	whileInsertOnTable(*sql.Tx, string, func() error) error | ||||||
| } | } | ||||||
|  |  | ||||||
|  | type queryable interface { | ||||||
|  | 	Exec(string, ...interface{}) (sql.Result, error) | ||||||
|  | 	Query(string, ...interface{}) (*sql.Rows, error) | ||||||
|  | 	QueryRow(string, ...interface{}) *sql.Row | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // batchSplitter is an interface with method which returns byte slice for | ||||||
|  | // splitting SQL batches. This need to split sql statements and run its | ||||||
|  | // separately. | ||||||
|  | // | ||||||
|  | // For Microsoft SQL Server batch splitter is "GO". For details see | ||||||
|  | // https://docs.microsoft.com/en-us/sql/t-sql/language-elements/sql-server-utilities-statements-go | ||||||
|  | type batchSplitter interface { | ||||||
|  | 	splitter() []byte | ||||||
|  | } | ||||||
|  |  | ||||||
|  | var ( | ||||||
|  | 	_ Helper = &MySQL{} | ||||||
|  | 	_ Helper = &PostgreSQL{} | ||||||
|  | 	_ Helper = &SQLite{} | ||||||
|  | 	_ Helper = &Oracle{} | ||||||
|  | 	_ Helper = &SQLServer{} | ||||||
|  | ) | ||||||
|  |  | ||||||
| type baseHelper struct{} | type baseHelper struct{} | ||||||
|  |  | ||||||
| func (*baseHelper) init(_ *sql.DB) error { | func (baseHelper) init(_ *sql.DB) error { | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (*baseHelper) quoteKeyword(str string) string { | func (baseHelper) quoteKeyword(str string) string { | ||||||
| 	return fmt.Sprintf(`"%s"`, str) | 	return fmt.Sprintf(`"%s"`, str) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (*baseHelper) whileInsertOnTable(_ *sql.Tx, _ string, fn func() error) error { | func (baseHelper) whileInsertOnTable(_ *sql.Tx, _ string, fn func() error) error { | ||||||
| 	return fn() | 	return fn() | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (baseHelper) isTableModified(_ queryable, _ string) (bool, error) { | ||||||
|  | 	return true, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (baseHelper) afterLoad(_ queryable) error { | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|   | |||||||
							
								
								
									
										44
									
								
								vendor/gopkg.in/testfixtures.v2/json.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										44
									
								
								vendor/gopkg.in/testfixtures.v2/json.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,44 @@ | |||||||
|  | package testfixtures | ||||||
|  |  | ||||||
|  | import ( | ||||||
|  | 	"database/sql/driver" | ||||||
|  | 	"encoding/json" | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | var ( | ||||||
|  | 	_ driver.Valuer = jsonArray{} | ||||||
|  | 	_ driver.Valuer = jsonMap{} | ||||||
|  | ) | ||||||
|  |  | ||||||
|  | type jsonArray []interface{} | ||||||
|  |  | ||||||
|  | func (a jsonArray) Value() (driver.Value, error) { | ||||||
|  | 	return json.Marshal(a) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | type jsonMap map[string]interface{} | ||||||
|  |  | ||||||
|  | func (m jsonMap) Value() (driver.Value, error) { | ||||||
|  | 	return json.Marshal(m) | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // Go refuses to convert map[interface{}]interface{} to JSON because JSON only support string keys | ||||||
|  | // So it's necessary to recursively convert all map[interface]interface{} to map[string]interface{} | ||||||
|  | func recursiveToJSON(v interface{}) (r interface{}) { | ||||||
|  | 	switch v := v.(type) { | ||||||
|  | 	case []interface{}: | ||||||
|  | 		for i, e := range v { | ||||||
|  | 			v[i] = recursiveToJSON(e) | ||||||
|  | 		} | ||||||
|  | 		r = jsonArray(v) | ||||||
|  | 	case map[interface{}]interface{}: | ||||||
|  | 		newMap := make(map[string]interface{}, len(v)) | ||||||
|  | 		for k, e := range v { | ||||||
|  | 			newMap[k.(string)] = recursiveToJSON(e) | ||||||
|  | 		} | ||||||
|  | 		r = jsonMap(newMap) | ||||||
|  | 	default: | ||||||
|  | 		r = v | ||||||
|  | 	} | ||||||
|  | 	return | ||||||
|  | } | ||||||
							
								
								
									
										101
									
								
								vendor/gopkg.in/testfixtures.v2/mysql.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										101
									
								
								vendor/gopkg.in/testfixtures.v2/mysql.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -8,6 +8,18 @@ import ( | |||||||
| // MySQL is the MySQL helper for this package | // MySQL is the MySQL helper for this package | ||||||
| type MySQL struct { | type MySQL struct { | ||||||
| 	baseHelper | 	baseHelper | ||||||
|  | 	tables         []string | ||||||
|  | 	tablesChecksum map[string]int64 | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (h *MySQL) init(db *sql.DB) error { | ||||||
|  | 	var err error | ||||||
|  | 	h.tables, err = h.tableNames(db) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (*MySQL) paramType() int { | func (*MySQL) paramType() int { | ||||||
| @@ -18,28 +30,105 @@ func (*MySQL) quoteKeyword(str string) string { | |||||||
| 	return fmt.Sprintf("`%s`", str) | 	return fmt.Sprintf("`%s`", str) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (*MySQL) databaseName(db *sql.DB) (dbName string) { | func (*MySQL) databaseName(q queryable) (string, error) { | ||||||
| 	db.QueryRow("SELECT DATABASE()").Scan(&dbName) | 	var dbName string | ||||||
| 	return | 	err := q.QueryRow("SELECT DATABASE()").Scan(&dbName) | ||||||
|  | 	return dbName, err | ||||||
| } | } | ||||||
|  |  | ||||||
| func (h *MySQL) disableReferentialIntegrity(db *sql.DB, loadFn loadFunction) error { | func (h *MySQL) tableNames(q queryable) ([]string, error) { | ||||||
|  | 	query := ` | ||||||
|  | 		SELECT table_name | ||||||
|  | 		FROM information_schema.tables | ||||||
|  | 		WHERE table_schema = ? | ||||||
|  | 		  AND table_type = 'BASE TABLE'; | ||||||
|  | 	` | ||||||
|  | 	dbName, err := h.databaseName(q) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	rows, err := q.Query(query, dbName) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	defer rows.Close() | ||||||
|  |  | ||||||
|  | 	var tables []string | ||||||
|  | 	for rows.Next() { | ||||||
|  | 		var table string | ||||||
|  | 		if err = rows.Scan(&table); err != nil { | ||||||
|  | 			return nil, err | ||||||
|  | 		} | ||||||
|  | 		tables = append(tables, table) | ||||||
|  | 	} | ||||||
|  | 	if err = rows.Err(); err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	return tables, nil | ||||||
|  |  | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (h *MySQL) disableReferentialIntegrity(db *sql.DB, loadFn loadFunction) (err error) { | ||||||
| 	// re-enable after load | 	// re-enable after load | ||||||
| 	defer db.Exec("SET FOREIGN_KEY_CHECKS = 1") | 	defer func() { | ||||||
|  | 		if _, err2 := db.Exec("SET FOREIGN_KEY_CHECKS = 1"); err2 != nil && err == nil { | ||||||
|  | 			err = err2 | ||||||
|  | 		} | ||||||
|  | 	}() | ||||||
|  |  | ||||||
| 	tx, err := db.Begin() | 	tx, err := db.Begin() | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
|  | 	defer tx.Rollback() | ||||||
|  |  | ||||||
| 	if _, err = tx.Exec("SET FOREIGN_KEY_CHECKS = 0"); err != nil { | 	if _, err = tx.Exec("SET FOREIGN_KEY_CHECKS = 0"); err != nil { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if err = loadFn(tx); err != nil { | 	if err = loadFn(tx); err != nil { | ||||||
| 		tx.Rollback() |  | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	return tx.Commit() | 	return tx.Commit() | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (h *MySQL) isTableModified(q queryable, tableName string) (bool, error) { | ||||||
|  | 	checksum, err := h.getChecksum(q, tableName) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return true, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	oldChecksum := h.tablesChecksum[tableName] | ||||||
|  |  | ||||||
|  | 	return oldChecksum == 0 || checksum != oldChecksum, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (h *MySQL) afterLoad(q queryable) error { | ||||||
|  | 	if h.tablesChecksum != nil { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	h.tablesChecksum = make(map[string]int64, len(h.tables)) | ||||||
|  | 	for _, t := range h.tables { | ||||||
|  | 		checksum, err := h.getChecksum(q, t) | ||||||
|  | 		if err != nil { | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  | 		h.tablesChecksum[t] = checksum | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (h *MySQL) getChecksum(q queryable, tableName string) (int64, error) { | ||||||
|  | 	sql := fmt.Sprintf("CHECKSUM TABLE %s", h.quoteKeyword(tableName)) | ||||||
|  | 	var ( | ||||||
|  | 		table    string | ||||||
|  | 		checksum int64 | ||||||
|  | 	) | ||||||
|  | 	if err := q.QueryRow(sql).Scan(&table, &checksum); err != nil { | ||||||
|  | 		return 0, err | ||||||
|  | 	} | ||||||
|  | 	return checksum, nil | ||||||
|  | } | ||||||
|   | |||||||
							
								
								
									
										73
									
								
								vendor/gopkg.in/testfixtures.v2/oracle.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										73
									
								
								vendor/gopkg.in/testfixtures.v2/oracle.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -43,14 +43,41 @@ func (*Oracle) quoteKeyword(str string) string { | |||||||
| 	return fmt.Sprintf("\"%s\"", strings.ToUpper(str)) | 	return fmt.Sprintf("\"%s\"", strings.ToUpper(str)) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (*Oracle) databaseName(db *sql.DB) (dbName string) { | func (*Oracle) databaseName(q queryable) (string, error) { | ||||||
| 	db.QueryRow("SELECT user FROM DUAL").Scan(&dbName) | 	var dbName string | ||||||
| 	return | 	err := q.QueryRow("SELECT user FROM DUAL").Scan(&dbName) | ||||||
|  | 	return dbName, err | ||||||
| } | } | ||||||
|  |  | ||||||
| func (*Oracle) getEnabledConstraints(db *sql.DB) ([]oracleConstraint, error) { | func (*Oracle) tableNames(q queryable) ([]string, error) { | ||||||
| 	constraints := make([]oracleConstraint, 0) | 	query := ` | ||||||
| 	rows, err := db.Query(` | 		SELECT TABLE_NAME | ||||||
|  | 		FROM USER_TABLES | ||||||
|  | 	` | ||||||
|  | 	rows, err := q.Query(query) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	defer rows.Close() | ||||||
|  |  | ||||||
|  | 	var tables []string | ||||||
|  | 	for rows.Next() { | ||||||
|  | 		var table string | ||||||
|  | 		if err = rows.Scan(&table); err != nil { | ||||||
|  | 			return nil, err | ||||||
|  | 		} | ||||||
|  | 		tables = append(tables, table) | ||||||
|  | 	} | ||||||
|  | 	if err = rows.Err(); err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	return tables, nil | ||||||
|  |  | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (*Oracle) getEnabledConstraints(q queryable) ([]oracleConstraint, error) { | ||||||
|  | 	var constraints []oracleConstraint | ||||||
|  | 	rows, err := q.Query(` | ||||||
| 		SELECT table_name, constraint_name | 		SELECT table_name, constraint_name | ||||||
| 		FROM user_constraints | 		FROM user_constraints | ||||||
| 		WHERE constraint_type = 'R' | 		WHERE constraint_type = 'R' | ||||||
| @@ -60,37 +87,46 @@ func (*Oracle) getEnabledConstraints(db *sql.DB) ([]oracleConstraint, error) { | |||||||
| 		return nil, err | 		return nil, err | ||||||
| 	} | 	} | ||||||
| 	defer rows.Close() | 	defer rows.Close() | ||||||
|  |  | ||||||
| 	for rows.Next() { | 	for rows.Next() { | ||||||
| 		var constraint oracleConstraint | 		var constraint oracleConstraint | ||||||
| 		rows.Scan(&constraint.tableName, &constraint.constraintName) | 		rows.Scan(&constraint.tableName, &constraint.constraintName) | ||||||
| 		constraints = append(constraints, constraint) | 		constraints = append(constraints, constraint) | ||||||
| 	} | 	} | ||||||
|  | 	if err = rows.Err(); err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
| 	return constraints, nil | 	return constraints, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (*Oracle) getSequences(db *sql.DB) ([]string, error) { | func (*Oracle) getSequences(q queryable) ([]string, error) { | ||||||
| 	sequences := make([]string, 0) | 	var sequences []string | ||||||
| 	rows, err := db.Query("SELECT sequence_name FROM user_sequences") | 	rows, err := q.Query("SELECT sequence_name FROM user_sequences") | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return nil, err | 		return nil, err | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	defer rows.Close() | 	defer rows.Close() | ||||||
|  |  | ||||||
| 	for rows.Next() { | 	for rows.Next() { | ||||||
| 		var sequence string | 		var sequence string | ||||||
| 		rows.Scan(&sequence) | 		if err = rows.Scan(&sequence); err != nil { | ||||||
|  | 			return nil, err | ||||||
|  | 		} | ||||||
| 		sequences = append(sequences, sequence) | 		sequences = append(sequences, sequence) | ||||||
| 	} | 	} | ||||||
|  | 	if err = rows.Err(); err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
| 	return sequences, nil | 	return sequences, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (h *Oracle) resetSequences(db *sql.DB) error { | func (h *Oracle) resetSequences(q queryable) error { | ||||||
| 	for _, sequence := range h.sequences { | 	for _, sequence := range h.sequences { | ||||||
| 		_, err := db.Exec(fmt.Sprintf("DROP SEQUENCE %s", h.quoteKeyword(sequence))) | 		_, err := q.Exec(fmt.Sprintf("DROP SEQUENCE %s", h.quoteKeyword(sequence))) | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			return err | 			return err | ||||||
| 		} | 		} | ||||||
| 		_, err = db.Exec(fmt.Sprintf("CREATE SEQUENCE %s START WITH %d", h.quoteKeyword(sequence), resetSequencesTo)) | 		_, err = q.Exec(fmt.Sprintf("CREATE SEQUENCE %s START WITH %d", h.quoteKeyword(sequence), resetSequencesTo)) | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			return err | 			return err | ||||||
| 		} | 		} | ||||||
| @@ -98,11 +134,14 @@ func (h *Oracle) resetSequences(db *sql.DB) error { | |||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (h *Oracle) disableReferentialIntegrity(db *sql.DB, loadFn loadFunction) error { | func (h *Oracle) disableReferentialIntegrity(db *sql.DB, loadFn loadFunction) (err error) { | ||||||
| 	// re-enable after load | 	// re-enable after load | ||||||
| 	defer func() { | 	defer func() { | ||||||
| 		for _, c := range h.enabledConstraints { | 		for _, c := range h.enabledConstraints { | ||||||
| 			db.Exec(fmt.Sprintf("ALTER TABLE %s ENABLE CONSTRAINT %s", h.quoteKeyword(c.tableName), h.quoteKeyword(c.constraintName))) | 			_, err2 := db.Exec(fmt.Sprintf("ALTER TABLE %s ENABLE CONSTRAINT %s", h.quoteKeyword(c.tableName), h.quoteKeyword(c.constraintName))) | ||||||
|  | 			if err2 != nil && err == nil { | ||||||
|  | 				err = err2 | ||||||
|  | 			} | ||||||
| 		} | 		} | ||||||
| 	}() | 	}() | ||||||
|  |  | ||||||
| @@ -118,9 +157,9 @@ func (h *Oracle) disableReferentialIntegrity(db *sql.DB, loadFn loadFunction) er | |||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
|  | 	defer tx.Rollback() | ||||||
|  |  | ||||||
| 	if err = loadFn(tx); err != nil { | 	if err = loadFn(tx); err != nil { | ||||||
| 		tx.Rollback() |  | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|   | |||||||
							
								
								
									
										150
									
								
								vendor/gopkg.in/testfixtures.v2/postgresql.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										150
									
								
								vendor/gopkg.in/testfixtures.v2/postgresql.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -3,6 +3,7 @@ package testfixtures | |||||||
| import ( | import ( | ||||||
| 	"database/sql" | 	"database/sql" | ||||||
| 	"fmt" | 	"fmt" | ||||||
|  | 	"strings" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // PostgreSQL is the PG helper for this package | // PostgreSQL is the PG helper for this package | ||||||
| @@ -18,6 +19,7 @@ type PostgreSQL struct { | |||||||
| 	tables                   []string | 	tables                   []string | ||||||
| 	sequences                []string | 	sequences                []string | ||||||
| 	nonDeferrableConstraints []pgConstraint | 	nonDeferrableConstraints []pgConstraint | ||||||
|  | 	tablesChecksum           map[string]string | ||||||
| } | } | ||||||
|  |  | ||||||
| type pgConstraint struct { | type pgConstraint struct { | ||||||
| @@ -28,7 +30,7 @@ type pgConstraint struct { | |||||||
| func (h *PostgreSQL) init(db *sql.DB) error { | func (h *PostgreSQL) init(db *sql.DB) error { | ||||||
| 	var err error | 	var err error | ||||||
|  |  | ||||||
| 	h.tables, err = h.getTables(db) | 	h.tables, err = h.tableNames(db) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| @@ -50,44 +52,57 @@ func (*PostgreSQL) paramType() int { | |||||||
| 	return paramTypeDollar | 	return paramTypeDollar | ||||||
| } | } | ||||||
|  |  | ||||||
| func (*PostgreSQL) databaseName(db *sql.DB) (dbName string) { | func (*PostgreSQL) databaseName(q queryable) (string, error) { | ||||||
| 	db.QueryRow("SELECT current_database()").Scan(&dbName) | 	var dbName string | ||||||
| 	return | 	err := q.QueryRow("SELECT current_database()").Scan(&dbName) | ||||||
|  | 	return dbName, err | ||||||
| } | } | ||||||
|  |  | ||||||
| func (h *PostgreSQL) getTables(db *sql.DB) ([]string, error) { | func (h *PostgreSQL) tableNames(q queryable) ([]string, error) { | ||||||
| 	var tables []string | 	var tables []string | ||||||
|  |  | ||||||
| 	sql := ` | 	sql := ` | ||||||
| SELECT table_name | 	        SELECT pg_namespace.nspname || '.' || pg_class.relname | ||||||
| FROM information_schema.tables | 		FROM pg_class | ||||||
| WHERE table_schema = 'public' | 		INNER JOIN pg_namespace ON pg_namespace.oid = pg_class.relnamespace | ||||||
|   AND table_type = 'BASE TABLE'; | 		WHERE pg_class.relkind = 'r' | ||||||
| ` | 		  AND pg_namespace.nspname NOT IN ('pg_catalog', 'information_schema') | ||||||
| 	rows, err := db.Query(sql) | 		  AND pg_namespace.nspname NOT LIKE 'pg_toast%'; | ||||||
|  | 	` | ||||||
|  | 	rows, err := q.Query(sql) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return nil, err | 		return nil, err | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	defer rows.Close() | 	defer rows.Close() | ||||||
|  |  | ||||||
| 	for rows.Next() { | 	for rows.Next() { | ||||||
| 		var table string | 		var table string | ||||||
| 		rows.Scan(&table) | 		if err = rows.Scan(&table); err != nil { | ||||||
|  | 			return nil, err | ||||||
|  | 		} | ||||||
| 		tables = append(tables, table) | 		tables = append(tables, table) | ||||||
| 	} | 	} | ||||||
|  | 	if err = rows.Err(); err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
| 	return tables, nil | 	return tables, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (h *PostgreSQL) getSequences(db *sql.DB) ([]string, error) { | func (h *PostgreSQL) getSequences(q queryable) ([]string, error) { | ||||||
| 	var sequences []string | 	const sql = ` | ||||||
|  | 		SELECT pg_namespace.nspname || '.' || pg_class.relname AS sequence_name | ||||||
|  | 		FROM pg_class | ||||||
|  | 		INNER JOIN pg_namespace ON pg_namespace.oid = pg_class.relnamespace | ||||||
|  | 		WHERE pg_class.relkind = 'S' | ||||||
|  | 	` | ||||||
|  |  | ||||||
| 	sql := "SELECT relname FROM pg_class WHERE relkind = 'S'" | 	rows, err := q.Query(sql) | ||||||
| 	rows, err := db.Query(sql) |  | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return nil, err | 		return nil, err | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	defer rows.Close() | 	defer rows.Close() | ||||||
|  |  | ||||||
|  | 	var sequences []string | ||||||
| 	for rows.Next() { | 	for rows.Next() { | ||||||
| 		var sequence string | 		var sequence string | ||||||
| 		if err = rows.Scan(&sequence); err != nil { | 		if err = rows.Scan(&sequence); err != nil { | ||||||
| @@ -95,18 +110,22 @@ func (h *PostgreSQL) getSequences(db *sql.DB) ([]string, error) { | |||||||
| 		} | 		} | ||||||
| 		sequences = append(sequences, sequence) | 		sequences = append(sequences, sequence) | ||||||
| 	} | 	} | ||||||
|  | 	if err = rows.Err(); err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
| 	return sequences, nil | 	return sequences, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (*PostgreSQL) getNonDeferrableConstraints(db *sql.DB) ([]pgConstraint, error) { | func (*PostgreSQL) getNonDeferrableConstraints(q queryable) ([]pgConstraint, error) { | ||||||
| 	var constraints []pgConstraint | 	var constraints []pgConstraint | ||||||
|  |  | ||||||
| 	sql := ` | 	sql := ` | ||||||
| SELECT table_name, constraint_name | 		SELECT table_schema || '.' || table_name, constraint_name | ||||||
| FROM information_schema.table_constraints | 		FROM information_schema.table_constraints | ||||||
| WHERE constraint_type = 'FOREIGN KEY' | 		WHERE constraint_type = 'FOREIGN KEY' | ||||||
|   AND is_deferrable = 'NO'` | 		  AND is_deferrable = 'NO' | ||||||
| 	rows, err := db.Query(sql) |   	` | ||||||
|  | 	rows, err := q.Query(sql) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return nil, err | 		return nil, err | ||||||
| 	} | 	} | ||||||
| @@ -114,23 +133,27 @@ WHERE constraint_type = 'FOREIGN KEY' | |||||||
| 	defer rows.Close() | 	defer rows.Close() | ||||||
| 	for rows.Next() { | 	for rows.Next() { | ||||||
| 		var constraint pgConstraint | 		var constraint pgConstraint | ||||||
| 		err = rows.Scan(&constraint.tableName, &constraint.constraintName) | 		if err = rows.Scan(&constraint.tableName, &constraint.constraintName); err != nil { | ||||||
| 		if err != nil { |  | ||||||
| 			return nil, err | 			return nil, err | ||||||
| 		} | 		} | ||||||
| 		constraints = append(constraints, constraint) | 		constraints = append(constraints, constraint) | ||||||
| 	} | 	} | ||||||
|  | 	if err = rows.Err(); err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
| 	return constraints, nil | 	return constraints, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (h *PostgreSQL) disableTriggers(db *sql.DB, loadFn loadFunction) error { | func (h *PostgreSQL) disableTriggers(db *sql.DB, loadFn loadFunction) (err error) { | ||||||
| 	defer func() { | 	defer func() { | ||||||
| 		// re-enable triggers after load | 		// re-enable triggers after load | ||||||
| 		var sql string | 		var sql string | ||||||
| 		for _, table := range h.tables { | 		for _, table := range h.tables { | ||||||
| 			sql += fmt.Sprintf("ALTER TABLE %s ENABLE TRIGGER ALL;", h.quoteKeyword(table)) | 			sql += fmt.Sprintf("ALTER TABLE %s ENABLE TRIGGER ALL;", h.quoteKeyword(table)) | ||||||
| 		} | 		} | ||||||
| 		db.Exec(sql) | 		if _, err2 := db.Exec(sql); err2 != nil && err == nil { | ||||||
|  | 			err = err2 | ||||||
|  | 		} | ||||||
| 	}() | 	}() | ||||||
|  |  | ||||||
| 	tx, err := db.Begin() | 	tx, err := db.Begin() | ||||||
| @@ -154,14 +177,16 @@ func (h *PostgreSQL) disableTriggers(db *sql.DB, loadFn loadFunction) error { | |||||||
| 	return tx.Commit() | 	return tx.Commit() | ||||||
| } | } | ||||||
|  |  | ||||||
| func (h *PostgreSQL) makeConstraintsDeferrable(db *sql.DB, loadFn loadFunction) error { | func (h *PostgreSQL) makeConstraintsDeferrable(db *sql.DB, loadFn loadFunction) (err error) { | ||||||
| 	defer func() { | 	defer func() { | ||||||
| 		// ensure constraint being not deferrable again after load | 		// ensure constraint being not deferrable again after load | ||||||
| 		var sql string | 		var sql string | ||||||
| 		for _, constraint := range h.nonDeferrableConstraints { | 		for _, constraint := range h.nonDeferrableConstraints { | ||||||
| 			sql += fmt.Sprintf("ALTER TABLE %s ALTER CONSTRAINT %s NOT DEFERRABLE;", h.quoteKeyword(constraint.tableName), h.quoteKeyword(constraint.constraintName)) | 			sql += fmt.Sprintf("ALTER TABLE %s ALTER CONSTRAINT %s NOT DEFERRABLE;", h.quoteKeyword(constraint.tableName), h.quoteKeyword(constraint.constraintName)) | ||||||
| 		} | 		} | ||||||
| 		db.Exec(sql) | 		if _, err2 := db.Exec(sql); err2 != nil && err == nil { | ||||||
|  | 			err = err2 | ||||||
|  | 		} | ||||||
| 	}() | 	}() | ||||||
|  |  | ||||||
| 	var sql string | 	var sql string | ||||||
| @@ -176,28 +201,31 @@ func (h *PostgreSQL) makeConstraintsDeferrable(db *sql.DB, loadFn loadFunction) | |||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
|  | 	defer tx.Rollback() | ||||||
|  |  | ||||||
| 	if _, err = tx.Exec("SET CONSTRAINTS ALL DEFERRED"); err != nil { | 	if _, err = tx.Exec("SET CONSTRAINTS ALL DEFERRED"); err != nil { | ||||||
| 		return nil | 		return err | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	if err = loadFn(tx); err != nil { | 	if err = loadFn(tx); err != nil { | ||||||
| 		tx.Rollback() |  | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	return tx.Commit() | 	return tx.Commit() | ||||||
| } | } | ||||||
|  |  | ||||||
| func (h *PostgreSQL) disableReferentialIntegrity(db *sql.DB, loadFn loadFunction) error { | func (h *PostgreSQL) disableReferentialIntegrity(db *sql.DB, loadFn loadFunction) (err error) { | ||||||
| 	// ensure sequences being reset after load | 	// ensure sequences being reset after load | ||||||
| 	defer h.resetSequences(db) | 	defer func() { | ||||||
|  | 		if err2 := h.resetSequences(db); err2 != nil && err == nil { | ||||||
|  | 			err = err2 | ||||||
|  | 		} | ||||||
|  | 	}() | ||||||
|  |  | ||||||
| 	if h.UseAlterConstraint { | 	if h.UseAlterConstraint { | ||||||
| 		return h.makeConstraintsDeferrable(db, loadFn) | 		return h.makeConstraintsDeferrable(db, loadFn) | ||||||
| 	} else { |  | ||||||
| 		return h.disableTriggers(db, loadFn) |  | ||||||
| 	} | 	} | ||||||
|  | 	return h.disableTriggers(db, loadFn) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (h *PostgreSQL) resetSequences(db *sql.DB) error { | func (h *PostgreSQL) resetSequences(db *sql.DB) error { | ||||||
| @@ -209,3 +237,53 @@ func (h *PostgreSQL) resetSequences(db *sql.DB) error { | |||||||
| 	} | 	} | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (h *PostgreSQL) isTableModified(q queryable, tableName string) (bool, error) { | ||||||
|  | 	checksum, err := h.getChecksum(q, tableName) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return false, err | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	oldChecksum := h.tablesChecksum[tableName] | ||||||
|  |  | ||||||
|  | 	return oldChecksum == "" || checksum != oldChecksum, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (h *PostgreSQL) afterLoad(q queryable) error { | ||||||
|  | 	if h.tablesChecksum != nil { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	h.tablesChecksum = make(map[string]string, len(h.tables)) | ||||||
|  | 	for _, t := range h.tables { | ||||||
|  | 		checksum, err := h.getChecksum(q, t) | ||||||
|  | 		if err != nil { | ||||||
|  | 			return err | ||||||
|  | 		} | ||||||
|  | 		h.tablesChecksum[t] = checksum | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (h *PostgreSQL) getChecksum(q queryable, tableName string) (string, error) { | ||||||
|  | 	sqlStr := fmt.Sprintf(` | ||||||
|  | 			SELECT md5(CAST((array_agg(t.*)) AS TEXT)) | ||||||
|  | 			FROM %s AS t | ||||||
|  | 		`, | ||||||
|  | 		h.quoteKeyword(tableName), | ||||||
|  | 	) | ||||||
|  |  | ||||||
|  | 	var checksum sql.NullString | ||||||
|  | 	if err := q.QueryRow(sqlStr).Scan(&checksum); err != nil { | ||||||
|  | 		return "", err | ||||||
|  | 	} | ||||||
|  | 	return checksum.String, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (*PostgreSQL) quoteKeyword(s string) string { | ||||||
|  | 	parts := strings.Split(s, ".") | ||||||
|  | 	for i, p := range parts { | ||||||
|  | 		parts[i] = fmt.Sprintf(`"%s"`, p) | ||||||
|  | 	} | ||||||
|  | 	return strings.Join(parts, ".") | ||||||
|  | } | ||||||
|   | |||||||
							
								
								
									
										54
									
								
								vendor/gopkg.in/testfixtures.v2/sqlite.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										54
									
								
								vendor/gopkg.in/testfixtures.v2/sqlite.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -14,23 +14,59 @@ func (*SQLite) paramType() int { | |||||||
| 	return paramTypeQuestion | 	return paramTypeQuestion | ||||||
| } | } | ||||||
|  |  | ||||||
| func (*SQLite) databaseName(db *sql.DB) (dbName string) { | func (*SQLite) databaseName(q queryable) (string, error) { | ||||||
| 	var seq int | 	var seq int | ||||||
| 	var main string | 	var main, dbName string | ||||||
| 	db.QueryRow("PRAGMA database_list").Scan(&seq, &main, &dbName) | 	err := q.QueryRow("PRAGMA database_list").Scan(&seq, &main, &dbName) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return "", err | ||||||
|  | 	} | ||||||
| 	dbName = filepath.Base(dbName) | 	dbName = filepath.Base(dbName) | ||||||
| 	return | 	return dbName, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (*SQLite) disableReferentialIntegrity(db *sql.DB, loadFn loadFunction) error { | func (*SQLite) tableNames(q queryable) ([]string, error) { | ||||||
|  | 	query := ` | ||||||
|  | 		SELECT name | ||||||
|  | 		FROM sqlite_master | ||||||
|  | 		WHERE type = 'table'; | ||||||
|  | 	` | ||||||
|  | 	rows, err := q.Query(query) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	defer rows.Close() | ||||||
|  |  | ||||||
|  | 	var tables []string | ||||||
|  | 	for rows.Next() { | ||||||
|  | 		var table string | ||||||
|  | 		if err = rows.Scan(&table); err != nil { | ||||||
|  | 			return nil, err | ||||||
|  | 		} | ||||||
|  | 		tables = append(tables, table) | ||||||
|  | 	} | ||||||
|  | 	if err = rows.Err(); err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
|  | 	return tables, nil | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (*SQLite) disableReferentialIntegrity(db *sql.DB, loadFn loadFunction) (err error) { | ||||||
|  | 	defer func() { | ||||||
|  | 		if _, err2 := db.Exec("PRAGMA defer_foreign_keys = OFF"); err2 != nil && err == nil { | ||||||
|  | 			err = err2 | ||||||
|  | 		} | ||||||
|  | 	}() | ||||||
|  |  | ||||||
|  | 	if _, err = db.Exec("PRAGMA defer_foreign_keys = ON"); err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	tx, err := db.Begin() | 	tx, err := db.Begin() | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
|  | 	defer tx.Rollback() | ||||||
| 	if _, err = tx.Exec("PRAGMA defer_foreign_keys = ON"); err != nil { |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	if err = loadFn(tx); err != nil { | 	if err = loadFn(tx); err != nil { | ||||||
| 		return err | 		return err | ||||||
|   | |||||||
							
								
								
									
										75
									
								
								vendor/gopkg.in/testfixtures.v2/sqlserver.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										75
									
								
								vendor/gopkg.in/testfixtures.v2/sqlserver.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -3,6 +3,7 @@ package testfixtures | |||||||
| import ( | import ( | ||||||
| 	"database/sql" | 	"database/sql" | ||||||
| 	"fmt" | 	"fmt" | ||||||
|  | 	"strings" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // SQLServer is the helper for SQL Server for this package. | // SQLServer is the helper for SQL Server for this package. | ||||||
| @@ -16,7 +17,7 @@ type SQLServer struct { | |||||||
| func (h *SQLServer) init(db *sql.DB) error { | func (h *SQLServer) init(db *sql.DB) error { | ||||||
| 	var err error | 	var err error | ||||||
|  |  | ||||||
| 	h.tables, err = h.getTables(db) | 	h.tables, err = h.tableNames(db) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
| @@ -28,46 +29,62 @@ func (*SQLServer) paramType() int { | |||||||
| 	return paramTypeQuestion | 	return paramTypeQuestion | ||||||
| } | } | ||||||
|  |  | ||||||
| func (*SQLServer) quoteKeyword(str string) string { | func (*SQLServer) quoteKeyword(s string) string { | ||||||
| 	return fmt.Sprintf("[%s]", str) | 	parts := strings.Split(s, ".") | ||||||
|  | 	for i, p := range parts { | ||||||
|  | 		parts[i] = fmt.Sprintf(`[%s]`, p) | ||||||
|  | 	} | ||||||
|  | 	return strings.Join(parts, ".") | ||||||
| } | } | ||||||
|  |  | ||||||
| func (*SQLServer) databaseName(db *sql.DB) (dbname string) { | func (*SQLServer) databaseName(q queryable) (string, error) { | ||||||
| 	db.QueryRow("SELECT DB_NAME()").Scan(&dbname) | 	var dbName string | ||||||
| 	return | 	err := q.QueryRow("SELECT DB_NAME()").Scan(&dbName) | ||||||
|  | 	return dbName, err | ||||||
| } | } | ||||||
|  |  | ||||||
| func (*SQLServer) getTables(db *sql.DB) ([]string, error) { | func (*SQLServer) tableNames(q queryable) ([]string, error) { | ||||||
| 	rows, err := db.Query("SELECT table_name FROM information_schema.tables") | 	rows, err := q.Query("SELECT table_schema + '.' + table_name FROM information_schema.tables") | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return nil, err | 		return nil, err | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	tables := make([]string, 0) |  | ||||||
| 	defer rows.Close() | 	defer rows.Close() | ||||||
|  |  | ||||||
|  | 	var tables []string | ||||||
| 	for rows.Next() { | 	for rows.Next() { | ||||||
| 		var table string | 		var table string | ||||||
| 		rows.Scan(&table) | 		if err = rows.Scan(&table); err != nil { | ||||||
|  | 			return nil, err | ||||||
|  | 		} | ||||||
| 		tables = append(tables, table) | 		tables = append(tables, table) | ||||||
| 	} | 	} | ||||||
|  | 	if err = rows.Err(); err != nil { | ||||||
|  | 		return nil, err | ||||||
|  | 	} | ||||||
| 	return tables, nil | 	return tables, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| func (*SQLServer) tableHasIdentityColumn(tx *sql.Tx, tableName string) bool { | func (h *SQLServer) tableHasIdentityColumn(q queryable, tableName string) bool { | ||||||
| 	sql := ` | 	sql := ` | ||||||
| SELECT COUNT(*) | 		SELECT COUNT(*) | ||||||
| FROM SYS.IDENTITY_COLUMNS | 		FROM SYS.IDENTITY_COLUMNS | ||||||
| WHERE OBJECT_NAME(OBJECT_ID) = ? | 		WHERE OBJECT_ID = OBJECT_ID(?) | ||||||
| ` | 	` | ||||||
| 	var count int | 	var count int | ||||||
| 	tx.QueryRow(sql, tableName).Scan(&count) | 	q.QueryRow(sql, h.quoteKeyword(tableName)).Scan(&count) | ||||||
| 	return count > 0 | 	return count > 0 | ||||||
|  |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func (h *SQLServer) whileInsertOnTable(tx *sql.Tx, tableName string, fn func() error) error { | func (h *SQLServer) whileInsertOnTable(tx *sql.Tx, tableName string, fn func() error) (err error) { | ||||||
| 	if h.tableHasIdentityColumn(tx, tableName) { | 	if h.tableHasIdentityColumn(tx, tableName) { | ||||||
| 		defer tx.Exec(fmt.Sprintf("SET IDENTITY_INSERT %s OFF", h.quoteKeyword(tableName))) | 		defer func() { | ||||||
|  | 			_, err2 := tx.Exec(fmt.Sprintf("SET IDENTITY_INSERT %s OFF", h.quoteKeyword(tableName))) | ||||||
|  | 			if err2 != nil && err == nil { | ||||||
|  | 				err = err2 | ||||||
|  | 			} | ||||||
|  | 		}() | ||||||
|  |  | ||||||
| 		_, err := tx.Exec(fmt.Sprintf("SET IDENTITY_INSERT %s ON", h.quoteKeyword(tableName))) | 		_, err := tx.Exec(fmt.Sprintf("SET IDENTITY_INSERT %s ON", h.quoteKeyword(tableName))) | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			return err | 			return err | ||||||
| @@ -76,19 +93,19 @@ func (h *SQLServer) whileInsertOnTable(tx *sql.Tx, tableName string, fn func() e | |||||||
| 	return fn() | 	return fn() | ||||||
| } | } | ||||||
|  |  | ||||||
| func (h *SQLServer) disableReferentialIntegrity(db *sql.DB, loadFn loadFunction) error { | func (h *SQLServer) disableReferentialIntegrity(db *sql.DB, loadFn loadFunction) (err error) { | ||||||
| 	// ensure the triggers are re-enable after all | 	// ensure the triggers are re-enable after all | ||||||
| 	defer func() { | 	defer func() { | ||||||
| 		sql := "" | 		var sql string | ||||||
| 		for _, table := range h.tables { | 		for _, table := range h.tables { | ||||||
| 			sql += fmt.Sprintf("ALTER TABLE %s WITH CHECK CHECK CONSTRAINT ALL;", h.quoteKeyword(table)) | 			sql += fmt.Sprintf("ALTER TABLE %s WITH CHECK CHECK CONSTRAINT ALL;", h.quoteKeyword(table)) | ||||||
| 		} | 		} | ||||||
| 		if _, err := db.Exec(sql); err != nil { | 		if _, err2 := db.Exec(sql); err2 != nil && err == nil { | ||||||
| 			fmt.Printf("Error on re-enabling constraints: %v\n", err) | 			err = err2 | ||||||
| 		} | 		} | ||||||
| 	}() | 	}() | ||||||
|  |  | ||||||
| 	sql := "" | 	var sql string | ||||||
| 	for _, table := range h.tables { | 	for _, table := range h.tables { | ||||||
| 		sql += fmt.Sprintf("ALTER TABLE %s NOCHECK CONSTRAINT ALL;", h.quoteKeyword(table)) | 		sql += fmt.Sprintf("ALTER TABLE %s NOCHECK CONSTRAINT ALL;", h.quoteKeyword(table)) | ||||||
| 	} | 	} | ||||||
| @@ -100,11 +117,19 @@ func (h *SQLServer) disableReferentialIntegrity(db *sql.DB, loadFn loadFunction) | |||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
|  | 	defer tx.Rollback() | ||||||
|  |  | ||||||
| 	if err = loadFn(tx); err != nil { | 	if err = loadFn(tx); err != nil { | ||||||
| 		tx.Rollback() |  | ||||||
| 		return err | 		return err | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	return tx.Commit() | 	return tx.Commit() | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // splitter is a batchSplitter interface implementation. We need it for | ||||||
|  | // SQL Server because commands like a `CREATE SCHEMA...` and a `CREATE TABLE...` | ||||||
|  | // could not be executed in the same batch. | ||||||
|  | // See https://docs.microsoft.com/en-us/previous-versions/sql/sql-server-2008-r2/ms175502(v=sql.105)#rules-for-using-batches | ||||||
|  | func (*SQLServer) splitter() []byte { | ||||||
|  | 	return []byte("GO\n") | ||||||
|  | } | ||||||
|   | |||||||
							
								
								
									
										86
									
								
								vendor/gopkg.in/testfixtures.v2/testfixtures.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										86
									
								
								vendor/gopkg.in/testfixtures.v2/testfixtures.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -2,7 +2,6 @@ package testfixtures | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"database/sql" | 	"database/sql" | ||||||
| 	"errors" |  | ||||||
| 	"fmt" | 	"fmt" | ||||||
| 	"io/ioutil" | 	"io/ioutil" | ||||||
| 	"path" | 	"path" | ||||||
| @@ -33,22 +32,10 @@ type insertSQL struct { | |||||||
| } | } | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| 	// ErrWrongCastNotAMap is returned when a map is not a map[interface{}]interface{} |  | ||||||
| 	ErrWrongCastNotAMap = errors.New("Could not cast record: not a map[interface{}]interface{}") |  | ||||||
|  |  | ||||||
| 	// ErrFileIsNotSliceOrMap is returned the the fixture file is not a slice or map. |  | ||||||
| 	ErrFileIsNotSliceOrMap = errors.New("The fixture file is not a slice or map") |  | ||||||
|  |  | ||||||
| 	// ErrKeyIsNotString is returned when a record is not of type string |  | ||||||
| 	ErrKeyIsNotString = errors.New("Record map key is not string") |  | ||||||
|  |  | ||||||
| 	// ErrNotTestDatabase is returned when the database name doesn't contains "test" |  | ||||||
| 	ErrNotTestDatabase = errors.New(`Loading aborted because the database name does not contains "test"`) |  | ||||||
|  |  | ||||||
| 	dbnameRegexp = regexp.MustCompile("(?i)test") | 	dbnameRegexp = regexp.MustCompile("(?i)test") | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // NewFolder craetes a context for all fixtures in a given folder into the database: | // NewFolder creates a context for all fixtures in a given folder into the database: | ||||||
| //     NewFolder(db, &PostgreSQL{}, "my/fixtures/folder") | //     NewFolder(db, &PostgreSQL{}, "my/fixtures/folder") | ||||||
| func NewFolder(db *sql.DB, helper Helper, folderName string) (*Context, error) { | func NewFolder(db *sql.DB, helper Helper, folderName string) (*Context, error) { | ||||||
| 	fixtures, err := fixturesFromFolder(folderName) | 	fixtures, err := fixturesFromFolder(folderName) | ||||||
| @@ -64,7 +51,7 @@ func NewFolder(db *sql.DB, helper Helper, folderName string) (*Context, error) { | |||||||
| 	return c, nil | 	return c, nil | ||||||
| } | } | ||||||
|  |  | ||||||
| // NewFiles craetes a context for all specified fixtures files into database: | // NewFiles creates a context for all specified fixtures files into database: | ||||||
| //     NewFiles(db, &PostgreSQL{}, | //     NewFiles(db, &PostgreSQL{}, | ||||||
| //         "fixtures/customers.yml", | //         "fixtures/customers.yml", | ||||||
| //         "fixtures/orders.yml" | //         "fixtures/orders.yml" | ||||||
| @@ -102,27 +89,55 @@ func newContext(db *sql.DB, helper Helper, fixtures []*fixtureFile) (*Context, e | |||||||
| 	return c, nil | 	return c, nil | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // DetectTestDatabase returns nil if databaseName matches regexp | ||||||
|  | //     if err := fixtures.DetectTestDatabase(); err != nil { | ||||||
|  | //         log.Fatal(err) | ||||||
|  | //     } | ||||||
|  | func (c *Context) DetectTestDatabase() error { | ||||||
|  | 	dbName, err := c.helper.databaseName(c.db) | ||||||
|  | 	if err != nil { | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	if !dbnameRegexp.MatchString(dbName) { | ||||||
|  | 		return ErrNotTestDatabase | ||||||
|  | 	} | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  |  | ||||||
| // Load wipes and after load all fixtures in the database. | // Load wipes and after load all fixtures in the database. | ||||||
| //     if err := fixtures.Load(); err != nil { | //     if err := fixtures.Load(); err != nil { | ||||||
| //         log.Fatal(err) | //         log.Fatal(err) | ||||||
| //     } | //     } | ||||||
| func (c *Context) Load() error { | func (c *Context) Load() error { | ||||||
| 	if !skipDatabaseNameCheck { | 	if !skipDatabaseNameCheck { | ||||||
| 		if !dbnameRegexp.MatchString(c.helper.databaseName(c.db)) { | 		if err := c.DetectTestDatabase(); err != nil { | ||||||
| 			return ErrNotTestDatabase | 			return err | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	err := c.helper.disableReferentialIntegrity(c.db, func(tx *sql.Tx) error { | 	err := c.helper.disableReferentialIntegrity(c.db, func(tx *sql.Tx) error { | ||||||
| 		for _, file := range c.fixturesFiles { | 		for _, file := range c.fixturesFiles { | ||||||
|  | 			modified, err := c.helper.isTableModified(tx, file.fileNameWithoutExtension()) | ||||||
|  | 			if err != nil { | ||||||
|  | 				return err | ||||||
|  | 			} | ||||||
|  | 			if !modified { | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
| 			if err := file.delete(tx, c.helper); err != nil { | 			if err := file.delete(tx, c.helper); err != nil { | ||||||
| 				return err | 				return err | ||||||
| 			} | 			} | ||||||
|  |  | ||||||
| 			err := c.helper.whileInsertOnTable(tx, file.fileNameWithoutExtension(), func() error { | 			err = c.helper.whileInsertOnTable(tx, file.fileNameWithoutExtension(), func() error { | ||||||
| 				for _, i := range file.insertSQLs { | 				for j, i := range file.insertSQLs { | ||||||
| 					if _, err := tx.Exec(i.sql, i.params...); err != nil { | 					if _, err := tx.Exec(i.sql, i.params...); err != nil { | ||||||
| 						return err | 						return &InsertError{ | ||||||
|  | 							Err:    err, | ||||||
|  | 							File:   file.fileName, | ||||||
|  | 							Index:  j, | ||||||
|  | 							SQL:    i.sql, | ||||||
|  | 							Params: i.params, | ||||||
|  | 						} | ||||||
| 					} | 					} | ||||||
| 				} | 				} | ||||||
| 				return nil | 				return nil | ||||||
| @@ -133,7 +148,10 @@ func (c *Context) Load() error { | |||||||
| 		} | 		} | ||||||
| 		return nil | 		return nil | ||||||
| 	}) | 	}) | ||||||
|  | 	if err != nil { | ||||||
| 		return err | 		return err | ||||||
|  | 	} | ||||||
|  | 	return c.helper.afterLoad(c.db) | ||||||
| } | } | ||||||
|  |  | ||||||
| func (c *Context) buildInsertSQLs() error { | func (c *Context) buildInsertSQLs() error { | ||||||
| @@ -204,25 +222,33 @@ func (f *fixtureFile) buildInsertSQL(h Helper, record map[interface{}]interface{ | |||||||
|  |  | ||||||
| 		sqlColumns = append(sqlColumns, h.quoteKeyword(keyStr)) | 		sqlColumns = append(sqlColumns, h.quoteKeyword(keyStr)) | ||||||
|  |  | ||||||
|  | 		// if string, try convert to SQL or time | ||||||
|  | 		// if map or array, convert to json | ||||||
|  | 		switch v := value.(type) { | ||||||
|  | 		case string: | ||||||
|  | 			if strings.HasPrefix(v, "RAW=") { | ||||||
|  | 				sqlValues = append(sqlValues, strings.TrimPrefix(v, "RAW=")) | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			if t, err := tryStrToDate(v); err == nil { | ||||||
|  | 				value = t | ||||||
|  | 			} | ||||||
|  | 		case []interface{}, map[interface{}]interface{}: | ||||||
|  | 			value = recursiveToJSON(v) | ||||||
|  | 		} | ||||||
|  |  | ||||||
| 		switch h.paramType() { | 		switch h.paramType() { | ||||||
| 		case paramTypeDollar: | 		case paramTypeDollar: | ||||||
| 			sqlValues = append(sqlValues, fmt.Sprintf("$%d", i)) | 			sqlValues = append(sqlValues, fmt.Sprintf("$%d", i)) | ||||||
| 		case paramTypeQuestion: | 		case paramTypeQuestion: | ||||||
| 			sqlValues = append(sqlValues, "?") | 			sqlValues = append(sqlValues, "?") | ||||||
| 		case paramTypeColon: | 		case paramTypeColon: | ||||||
| 			switch { |  | ||||||
| 			case isDateTime(value): |  | ||||||
| 				sqlValues = append(sqlValues, fmt.Sprintf("to_date(:%d, 'YYYY-MM-DD HH24:MI:SS')", i)) |  | ||||||
| 			case isDate(value): |  | ||||||
| 				sqlValues = append(sqlValues, fmt.Sprintf("to_date(:%d, 'YYYY-MM-DD')", i)) |  | ||||||
| 			case isTime(value): |  | ||||||
| 				sqlValues = append(sqlValues, fmt.Sprintf("to_date(:%d, 'HH24:MI:SS')", i)) |  | ||||||
| 			default: |  | ||||||
| 			sqlValues = append(sqlValues, fmt.Sprintf(":%d", i)) | 			sqlValues = append(sqlValues, fmt.Sprintf(":%d", i)) | ||||||
| 		} | 		} | ||||||
| 		} |  | ||||||
| 		i++ |  | ||||||
| 		values = append(values, value) | 		values = append(values, value) | ||||||
|  | 		i++ | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	sqlStr = fmt.Sprintf( | 	sqlStr = fmt.Sprintf( | ||||||
|   | |||||||
							
								
								
									
										52
									
								
								vendor/gopkg.in/testfixtures.v2/time.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										52
									
								
								vendor/gopkg.in/testfixtures.v2/time.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,36 +1,34 @@ | |||||||
| package testfixtures | package testfixtures | ||||||
|  |  | ||||||
| import "regexp" | import ( | ||||||
|  | 	"errors" | ||||||
| var ( | 	"time" | ||||||
| 	regexpDate     = regexp.MustCompile("\\d\\d\\d\\d-\\d\\d-\\d\\d") |  | ||||||
| 	regexpDateTime = regexp.MustCompile("\\d\\d\\d\\d-\\d\\d-\\d\\d \\d\\d:\\d\\d:\\d\\d") |  | ||||||
| 	regexpTime     = regexp.MustCompile("\\d\\d:\\d\\d:\\d\\d") |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| func isDate(value interface{}) bool { | var timeFormats = []string{ | ||||||
| 	str, isStr := value.(string) | 	"2006-01-02", | ||||||
| 	if !isStr { | 	"2006-01-02 15:04", | ||||||
| 		return false | 	"2006-01-02 15:04:05", | ||||||
| 	} | 	"20060102", | ||||||
|  | 	"20060102 15:04", | ||||||
| 	return regexpDate.MatchString(str) | 	"20060102 15:04:05", | ||||||
|  | 	"02/01/2006", | ||||||
|  | 	"02/01/2006 15:04", | ||||||
|  | 	"02/01/2006 15:04:05", | ||||||
|  | 	"2006-01-02T15:04-07:00", | ||||||
|  | 	"2006-01-02T15:04:05-07:00", | ||||||
| } | } | ||||||
|  |  | ||||||
| func isDateTime(value interface{}) bool { | // ErrCouldNotConvertToTime is returns when a string is not a reconizable time format | ||||||
| 	str, isStr := value.(string) | var ErrCouldNotConvertToTime = errors.New("Could not convert string to time") | ||||||
| 	if !isStr { |  | ||||||
| 		return false | func tryStrToDate(s string) (time.Time, error) { | ||||||
|  | 	for _, f := range timeFormats { | ||||||
|  | 		t, err := time.ParseInLocation(f, s, time.Local) | ||||||
|  | 		if err != nil { | ||||||
|  | 			continue | ||||||
| 		} | 		} | ||||||
|  | 		return t, nil | ||||||
| 	return regexpDateTime.MatchString(str) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func isTime(value interface{}) bool { |  | ||||||
| 	str, isStr := value.(string) |  | ||||||
| 	if !isStr { |  | ||||||
| 		return false |  | ||||||
| 	} | 	} | ||||||
|  | 	return time.Time{}, ErrCouldNotConvertToTime | ||||||
| 	return regexpTime.MatchString(str) |  | ||||||
| } | } | ||||||
|   | |||||||
		Reference in New Issue
	
	Block a user
	 Mura Li
					Mura Li