added comparison operator support

Author: pupizoid

ormlite.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"database/sql"
 	"fmt"
+	"github.com/davecgh/go-spew/spew"
 	"reflect"
 	"strings"
 	"time"
@@ -49,6 +50,16 @@ type OrderBy struct {
 // Where is a map containing fields and their values to meet in the result
 type Where map[string]interface{}
 
+type Greater float64
+
+type Less float64
+
+type GreaterOrEqual float64
+
+type LessOrEqual float64
+
+type NotEqual float64
+
 const (
 	// AND is a glue between multiple statements after `where`
 	AND = " and "
@@ -257,7 +268,21 @@ func queryWithOptions(ctx context.Context, db *sql.DB, table string, columns []s
 						keys = append(keys, fmt.Sprintf("%s like ?", k))
 						values = append(values, fmt.Sprintf("%%%s%%", v))
 					default:
-						keys = append(keys, fmt.Sprintf("%s = ?", k))
+						spew.Dump(v)
+						switch v.(type) {
+						case Greater:
+							keys = append(keys, fmt.Sprintf("%s > ?", k))
+						case GreaterOrEqual:
+							keys = append(keys, fmt.Sprintf("%s >= ?", k))
+						case Less:
+							keys = append(keys, fmt.Sprintf("%s < ?", k))
+						case LessOrEqual:
+							keys = append(keys, fmt.Sprintf("%s <= ?", k))
+						case NotEqual:
+							keys = append(keys, fmt.Sprintf("%s != ?", k))
+						default:
+							keys = append(keys, fmt.Sprintf("%s = ?", k))
+						}
 						values = append(values, v)
 					}
 				} else {

ormlite_test.go

@@ -1080,3 +1080,48 @@ func (s *testCustomFieldInMTMModel) TestParenthesis() {
 func TestCustomFieldInMTM(t *testing.T) {
 	suite.Run(t, new(testCustomFieldInMTMModel))
 }
+
+type testOperatorsModel struct {
+	ID     int64 `ormlite:"primary"`
+	Number int
+}
+
+func (m *testOperatorsModel) Table() string { return "test" }
+
+func TestGreaterOrLessOperator(t *testing.T) {
+	db, err := sql.Open("sqlite3", ":memory:?_fk=1")
+	require.NoError(t, err)
+
+	_, err = db.Exec(`
+		create table test(id integer primary key, number integer);
+		insert into test(number) values (1), (2), (3), (4), (5);
+	`)
+	require.NoError(t, err)
+
+	var mm []*testOperatorsModel
+	if assert.NoError(t, QuerySlice(db, &Options{Where: Where{"number": Greater(4)}}, &mm)) {
+		if assert.Len(t, mm, 1) {
+			assert.EqualValues(t, 5, mm[0].ID)
+		}
+	}
+
+	mm = nil
+	if assert.NoError(t, QuerySlice(db, &Options{Where: Where{"number": Less(3)}}, &mm)) {
+		assert.Len(t, mm, 2)
+	}
+
+	mm = nil
+	if assert.NoError(t, QuerySlice(db, &Options{Where: Where{"number": GreaterOrEqual(3)}}, &mm)) {
+		assert.Len(t, mm, 3)
+	}
+
+	mm = nil
+	if assert.NoError(t, QuerySlice(db, &Options{Where: Where{"number": LessOrEqual(2)}}, &mm)) {
+		assert.Len(t, mm, 2)
+	}
+
+	mm = nil
+	if assert.NoError(t, QuerySlice(db, &Options{Where: Where{"number": NotEqual(3)}}, &mm)) {
+		assert.Len(t, mm, 4)
+	}
+}