Initial.

services/heartbeat.go

@@ -0,0 +1,40 @@
+package services
+
+import (
+	"database/sql"
+	"errors"
+	"fmt"
+
+	"github.com/n1try/wakapi/models"
+)
+
+const (
+	TableName = "heartbeat"
+)
+
+type HeartbeatService struct {
+	Db *sql.DB
+}
+
+func (srv *HeartbeatService) InsertMulti(heartbeats []models.Heartbeat) error {
+	qTpl := "INSERT INTO %+s (user, time, entity, type, category, is_write, project, branch, language) VALUES %+s;"
+	qFill := ""
+	vals := []interface{}{}
+
+	for _, h := range heartbeats {
+		qFill = "(?, ?, ?, ?, ?, ?, ?, ?, ?),"
+		vals = append(vals, h.User, h.Time.String(), h.Entity, h.Type, h.Category, h.IsWrite, h.Project, h.Branch, h.Language)
+	}
+
+	q := fmt.Sprintf(qTpl, TableName, qFill[0:len(qFill)-1])
+	stmt, _ := srv.Db.Prepare(q)
+	result, err := stmt.Exec(vals...)
+	if err != nil {
+		return err
+	}
+	n, err := result.RowsAffected()
+	if err != nil || n != int64(len(heartbeats)) {
+		return errors.New(fmt.Sprintf("Failed to insert %+v rows.", len(heartbeats)))
+	}
+	return nil
+}