cdleo /
go-sqldb
| Total Lines | 47 |
| Duplicated Lines | 0 % |
| Changes | 0 | ||
| 1 | package adapter |
||
| 2 | |||
| 3 | import ( |
||
| 4 | "fmt" |
||
| 5 | |||
| 6 | "github.com/cdleo/go-commons/sqlcommons" |
||
| 7 | "github.com/mattn/go-sqlite3" |
||
| 8 | ) |
||
| 9 | |||
| 10 | type sqlite3Adapter struct{} |
||
| 11 | |||
| 12 | func NewSQLite3Adapter() sqlcommons.SQLAdapter { |
||
| 13 | return &sqlite3Adapter{} |
||
| 14 | } |
||
| 15 | |||
| 16 | func (t *sqlite3Adapter) Translate(query string) string { |
||
| 17 | return query |
||
| 18 | } |
||
| 19 | |||
| 20 | func (s *sqlite3Adapter) ErrorHandler(err error) error { |
||
| 21 | |||
| 22 | if sqliteError, ok := err.(sqlite3.Error); ok { |
||
| 23 | |||
| 24 | if sqliteError.Code == 18 { //SQLITE_TOOBIG |
||
| 25 | return sqlcommons.ValueTooLargeForColumn |
||
| 26 | |||
| 27 | } else if sqliteError.Code == 19 { //SQLITE_CONSTRAINT |
||
| 28 | if sqliteError.ExtendedCode == 787 || /*SQLITE_CONSTRAINT_FOREIGNKEY*/ |
||
| 29 | sqliteError.ExtendedCode == 1555 || /*SQLITE_CONSTRAINT_PRIMARYKEY*/ |
||
| 30 | sqliteError.ExtendedCode == 1811 { /*SQLITE_CONSTRAINT_TRIGGER*/ |
||
| 31 | return sqlcommons.IntegrityConstraintViolation |
||
| 32 | |||
| 33 | } else if sqliteError.ExtendedCode == 1299 { //SQLITE_CONSTRAINT_NOTNULL |
||
| 34 | return sqlcommons.CannotSetNullColumn |
||
| 35 | |||
| 36 | } else if sqliteError.ExtendedCode == 2067 { //SQLITE_CONSTRAINT_UNIQUE |
||
| 37 | return sqlcommons.UniqueConstraintViolation |
||
| 38 | |||
| 39 | } |
||
| 40 | } else if sqliteError.Code == 25 { //SQLITE_RANGE |
||
| 41 | return sqlcommons.InvalidNumericValue |
||
| 42 | } |
||
| 43 | |||
| 44 | return fmt.Errorf("Unhandled SQLite3 error. Code:[%s] Extended:[%s] Desc:[%s]", sqliteError.Code, sqliteError.ExtendedCode, sqliteError.Error()) |
||
| 45 | |||
| 46 | } else { |
||
| 47 | return err |
||
| 48 | } |
||
| 50 |
