diff options
| -rw-r--r-- | pkg/datamaps/db.go | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/pkg/datamaps/db.go b/pkg/datamaps/db.go index 2501da8..ee934c4 100644 --- a/pkg/datamaps/db.go +++ b/pkg/datamaps/db.go @@ -80,7 +80,14 @@ func DatamapToDB(opts *Options) error { if err != nil { return err } - _, err = stmtDm.Exec(opts.DMName, time.Now()) + res, err := stmtDm.Exec(opts.DMName, time.Now()) + if err != nil { + return err + } + lastId, err := res.LastInsertId() + if err != nil { + return err + } stmtDml, err := tx.Prepare("INSERT INTO datamap_line (dm_id, key, sheet, cellref) VALUES(?,?,?,?);") if err != nil { @@ -89,7 +96,7 @@ func DatamapToDB(opts *Options) error { defer stmtDm.Close() defer stmtDml.Close() for _, dml := range data { - _, err = stmtDml.Exec(1, dml.Key, dml.Sheet, dml.Cellref) + _, err = stmtDml.Exec(lastId, dml.Key, dml.Sheet, dml.Cellref) if err != nil { return err } |