feat(db) : newSQLinserter method

seipan · Oct 27, 2023 · 60bd67d · 60bd67d
1 parent 438c2bb
commit 60bd67d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/lib/db.go b/lib/db.go
@@ -43,12 +43,16 @@ func Exec(config Config) error {
 	return nil
 }
 
-func newSQLInserter(dbtype string) (query.Inserter, error) {
+func newSQLInserter(dbtype string, tablename string, db *sql.DB) (query.Inserter, error) {
 	switch dbtype {
 	case "mysql":
-		return &mysql.MySQLInserter{}, nil
+		return mysql.NewMySQLInserter(tablename, db), nil
 	case "sqlite":
-		return &sqlite.SQLiteInserter{}, nil
+		return sqlite.NewSQLiteInserter(tablename, db), nil
+	case "postgresql":
+		return postgresql.NewPostgresSQLInserter(tablename, db), nil
+	case "mariadb":
+		return mariadb.NewMariaDBInserter(tablename, db), nil
 	default:
 		return nil, fmt.Errorf("invalid dbtype: %s", dbtype)
 	}

diff --git a/mariadb/query.go b/mariadb/query.go
@@ -74,7 +74,7 @@ func (i *MariaDBSQLInserter) Insert() error {
 	return nil
 }
 
-func NewMariaDBSQLInserter(tableName string, db *sql.DB) query.Inserter {
+func NewMariaDBInserter(tableName string, db *sql.DB) query.Inserter {
 	return &MariaDBSQLInserter{
 		tableName: tableName,
 		db:        db,