feat(db): update reader

agerr/log.go

@@ -0,0 +1,23 @@
+package agerr
+
+import "github.com/agflow/tools/log"
+
+// CallAndLog calls function which may return error and logs it.
+// The intention of this function is to be used with `go` and `defer` clauses.
+func CallAndLog(f func() error) {
+	Log(f())
+}
+
+// Log logs error unless nil
+func Log(err error) {
+	if err != nil {
+		log.Errorf("unhandled error %+v", err)
+	}
+}
+
+// Assert panics and breaks the program unless err is nil
+func Assert(err error) {
+	if err != nil {
+		log.Panic(err)
+	}
+}

log/log.go

@@ -92,3 +92,8 @@ func Fatalf(format string, v ...interface{}) {
 func Fatal(v ...interface{}) {
 	log.Fatal(v...)
 }
+
+// Panic logs in error level with a posterior Panic()
+func Panic(v ...interface{}) {
+	log.Panic(v...)
+}

sql/db/service.go

@@ -15,13 +15,20 @@ type Client struct {
 type Service interface {
 	Select(interface{}, string, ...interface{}) error
 	Close() error
+	Exec(string, ...interface{}) error
 }
 
-// Select selects from `client` using the `query` and `args`
+// Select selects from `client` using the `query` and `args` and set the result on `dest`
 func (c *Client) Select(dest interface{}, query string, args ...interface{}) error {
 	return Select(c.DB, dest, query, args...)
 }
 
+// Exec executes from `client` using the `query` and `args`
+func (c *Client) Exec(query string, args ...interface{}) error {
+	_, err := c.DB.Exec(query, args...)
+	return err
+}
+
 // New return a new db.Client
 func New(url string) (*Client, error) {
 	db, err := sql.Open("postgres", url)

sql/query/reader.go

@@ -9,6 +9,7 @@ import (
 
 	"github.com/pkg/errors"
 
+	"github.com/agflow/tools/agerr"
 	"github.com/agflow/tools/log"
 )
 
@@ -76,3 +77,23 @@ func MustLoad(queriesFS embed.FS, query interface{}) {
 		mustReadFiles(r.Name(), queriesFS, query)
 	}
 }
+
+// MustLoadSQLFiles queries loads queries that are located on `dir`
+func MustLoadSQLFiles(dir string, queriesFS embed.FS, dest interface{}) {
+	v := reflect.Indirect(reflect.ValueOf(dest))
+	for i := 0; i < v.NumField(); i++ {
+		sField := v.Type().Field(i)
+		vField := v.FieldByName(sField.Name)
+		setFile(dir+sField.Tag.Get("sql"), queriesFS, vField)
+	}
+}
+
+func setFile(dir string, queriesFS embed.FS, v reflect.Value) {
+	if v.Kind() == reflect.Struct {
+		MustLoadSQLFiles(dir+"/", queriesFS, v.Addr().Interface())
+		return
+	}
+	file, err := queriesFS.ReadFile(dir + ".sql")
+	agerr.Assert(err)
+	v.SetString(string(file))
+}