From 2425c08bc372809c28bea021cd9ed0de3b8eebb2 Mon Sep 17 00:00:00 2001
From: Shulhan <m.shulhan@gmail.com>
Date: Thu, 13 Feb 2020 20:14:39 +0700
Subject: sql: a new package as an extension to "database/sql"

---
 lib/sql/table.go | 55 +++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 55 insertions(+)
 create mode 100644 lib/sql/table.go

(limited to 'lib/sql/table.go')

diff --git a/lib/sql/table.go b/lib/sql/table.go
new file mode 100644
index 00000000..3dbdb984
--- /dev/null
+++ b/lib/sql/table.go
@@ -0,0 +1,55 @@
+// Copyright 2020, Shulhan <m.shulhan@gmail.com>. All rights reserved.
+// Use of this source code is governed by a BSD-style
+// license that can be found in the LICENSE file.
+
+package sql
+
+import (
+	"database/sql"
+	"fmt"
+	"strings"
+)
+
+//
+// Table represent a tuple or table in database.
+//
+// A table has Name, PrimaryKey, and list of Row.
+//
+type Table struct {
+	Name       string // Table name, required.
+	PrimaryKey string // Primary key of table, optional.
+	Rows       []Row  // The row or data in the table, optional.
+}
+
+//
+// Insert all rows into table, one by one.
+//
+// On success, it will return list of ID, if table has primary key.
+//
+func (table *Table) Insert(tx *sql.Tx) (ids []int64, err error) {
+	for _, row := range table.Rows {
+		names, holders, values := row.ExtractSQLFields()
+		if len(names) == 0 {
+			continue
+		}
+
+		//nolint: gosec
+		q := fmt.Sprintf("INSERT INTO %s (%s) VALUES (%s)",
+			table.Name, strings.Join(names, ","),
+			strings.Join(holders, ","))
+
+		res, err := tx.Exec(q, values...)
+		if err != nil {
+			return nil, err
+		}
+
+		id, err := res.LastInsertId()
+		if err != nil {
+			continue
+		}
+
+		ids = append(ids, id)
+	}
+
+	return ids, nil
+}
-- 
cgit v1.3