first commit

pgm.go

@@ -0,0 +1,134 @@
+// Patial Tech.
+// Author, Ankit Patial
+
+package pgm
+
+import (
+	"errors"
+	"strings"
+	"time"
+
+	"github.com/jackc/pgx/v5"
+	"github.com/jackc/pgx/v5/pgtype"
+)
+
+// Table in database
+type Table struct {
+	Name       string
+	PK         []string
+	FieldCount uint16
+	debug      bool
+}
+
+// Debug when set true will print generated query string in stdout
+func (t Table) Debug() Clause {
+	t.debug = true
+	return t
+}
+
+//
+//  Field ==>
+//
+
+// Field related to a table
+type Field string
+
+func (f Field) Name() string {
+	return strings.Split(string(f), ".")[1]
+}
+
+func (f Field) String() string {
+	return string(f)
+}
+
+// Count fn wrapping of field
+func (f Field) Count() Field {
+	return Field("COUNT(" + f.String() + ")")
+}
+
+// Avg fn wrapping of field
+func (f Field) Avg() Field {
+	return Field("AVG(" + f.String() + ")")
+}
+
+func (f Field) Eq(val any) Conditioner {
+	col := f.String()
+	return &Cond{Field: col, Val: val, op: " = $", len: len(col) + 5}
+}
+
+// EqualFold will user LOWER() for comparision
+func (f Field) EqFold(val any) Conditioner {
+	col := f.String()
+	return &Cond{Field: "LOWER(" + col + ")", Val: val, op: " = LOWER($", action: CondActionNeedToClose, len: len(col) + 5}
+}
+
+func (f Field) NEq(val any) Conditioner {
+	col := f.String()
+	return &Cond{Field: col, Val: val, op: " != $", len: len(col) + 5}
+}
+
+func (f Field) Gt(val any) Conditioner {
+	col := f.String()
+	return &Cond{Field: col, Val: val, op: " > $", len: len(col) + 5}
+}
+
+func (f Field) Gte(val any) Conditioner {
+	col := f.String()
+	return &Cond{Field: col, Val: val, op: " >= $", len: len(col) + 5}
+}
+
+func (f Field) Like(val string) Conditioner {
+	col := f.String()
+	return &Cond{Field: col, Val: val, op: " LIKE $", len: len(f.String()) + 5}
+}
+
+func (f Field) LikeFold(val string) Conditioner {
+	col := f.String()
+	return &Cond{Field: "LOWER(" + col + ")", Val: val, op: " LIKE LOWER($", action: CondActionNeedToClose, len: len(col) + 5}
+}
+
+// ILIKE is case-insensitive
+func (f Field) ILike(val string) Conditioner {
+	col := f.String()
+	return &Cond{Field: col, Val: val, op: " ILIKE $", len: len(col) + 5}
+}
+
+func (f Field) NotIn(val ...any) Conditioner {
+	col := f.String()
+	return &Cond{Field: col, Val: val, op: " NOT IN($", action: CondActionNeedToClose, len: len(col) + 5}
+}
+
+func (f Field) NotInSubQuery(qry WhereClause) Conditioner {
+	col := f.String()
+	return &Cond{Field: col, Val: qry, op: " NOT IN($)", action: CondActionSubQuery}
+}
+
+//
+//  Helper func ==>
+//
+
+// PgTime as in UTC
+func PgTime(t time.Time) pgtype.Timestamptz {
+	return pgtype.Timestamptz{Time: t, Valid: true}
+}
+
+func PgTimeNow() pgtype.Timestamptz {
+	return pgtype.Timestamptz{Time: time.Now(), Valid: true}
+}
+
+// IsNotFound error check
+func IsNotFound(err error) bool {
+	return errors.Is(err, pgx.ErrNoRows)
+}
+
+func ConcatWs(sep string, fields ...Field) string {
+	return "concat_ws('" + sep + "'," + joinFileds(fields) + ")"
+}
+
+func StringAgg(exp, sep string) string {
+	return "string_agg(" + exp + ",'" + sep + "')"
+}
+
+func StringAggCast(exp, sep string) string {
+	return "string_agg(cast(" + exp + " as varchar),'" + sep + "')"
+}