select with timeout

Author: folospace

orm/query.go

@@ -36,6 +36,7 @@ type Query[T Table] struct {
     withCtes        []SubQuery
     windows         []SubQuery
     self            *Query[SubQuery]
+    selectTimeout   string
 }
 
 //query table[struct] generics

orm/query_select.go

@@ -1,67 +1,79 @@
 package orm
 
-import "strings"
+import (
+    "strconv"
+    "strings"
+    "time"
+)
 
 type SelectForUpdateType string
 
 const (
-	SelectForUpdateTypeDefault    SelectForUpdateType = "for update"
-	SelectForUpdateTypeNowait     SelectForUpdateType = "for update nowait"
-	SelectForUpdateTypeSkipLocked SelectForUpdateType = "for update skip locked"
+    SelectForUpdateTypeDefault    SelectForUpdateType = "for update"
+    SelectForUpdateTypeNowait     SelectForUpdateType = "for update nowait"
+    SelectForUpdateTypeSkipLocked SelectForUpdateType = "for update skip locked"
 )
 
 func (m Query[T]) SelectRank(column interface{}, alias string) Query[T] {
-	return m.SelectOver("rank()", func(query Query[T]) Query[T] {
-		return query.OrderBy(column)
-	}, alias)
+    return m.SelectOver("rank()", func(query Query[T]) Query[T] {
+        return query.OrderBy(column)
+    }, alias)
 }
 
 func (m Query[T]) SelectRankDesc(column interface{}, alias string) Query[T] {
-	return m.SelectOver("rank()", func(query Query[T]) Query[T] {
-		return query.OrderByDesc(column)
-	}, alias)
+    return m.SelectOver("rank()", func(query Query[T]) Query[T] {
+        return query.OrderByDesc(column)
+    }, alias)
 }
 
 func (m Query[T]) SelectOver(windowFunc string, f func(query Query[T]) Query[T], alias string) Query[T] {
-	partitionStart := len(m.partitionbys)
-	orderStart := len(m.orderbys)
-	nq := f(m)
-	partitions := nq.partitionbys[partitionStart:]
-	orders := nq.orderbys[orderStart:]
+    partitionStart := len(m.partitionbys)
+    orderStart := len(m.orderbys)
+    nq := f(m)
+    partitions := nq.partitionbys[partitionStart:]
+    orders := nq.orderbys[orderStart:]
 
-	m.setErr(nq.result.Err)
+    m.setErr(nq.result.Err)
 
-	newSelect := windowFunc + " over ("
-	if len(partitions) > 0 {
-		newSelect += "partition by " + strings.Join(partitions, ",") + " "
-	}
-	if len(orders) > 0 {
-		newSelect += "order by " + strings.Join(orders, ",")
-	}
-	newSelect += ")"
+    newSelect := windowFunc + " over ("
+    if len(partitions) > 0 {
+        newSelect += "partition by " + strings.Join(partitions, ",") + " "
+    }
+    if len(orders) > 0 {
+        newSelect += "order by " + strings.Join(orders, ",")
+    }
+    newSelect += ")"
 
-	newSelect += " as " + alias
+    newSelect += " as " + alias
 
-	m.columns = append(m.columns, newSelect)
-	return m
+    m.columns = append(m.columns, newSelect)
+    return m
 }
 
 func (m Query[T]) SelectOverRaw(windowFunc string, windowName string, alias string) Query[T] {
-	newSelect := windowFunc + " over " + windowName + " as " + alias
-	m.columns = append(m.columns, newSelect)
-	return m
+    newSelect := windowFunc + " over " + windowName + " as " + alias
+    m.columns = append(m.columns, newSelect)
+    return m
 }
 
 func (m Query[T]) Select(columns ...interface{}) Query[T] {
-	m.columns = append(m.columns, columns...)
-	return m
+    m.columns = append(m.columns, columns...)
+    return m
 }
 
 func (m Query[T]) ForUpdate(forUpdateType ...SelectForUpdateType) Query[T] {
-	if len(forUpdateType) == 0 {
-		m.forUpdate = SelectForUpdateTypeDefault
-	} else {
-		m.forUpdate = forUpdateType[0]
-	}
-	return m
+    if len(forUpdateType) == 0 {
+        m.forUpdate = SelectForUpdateTypeDefault
+    } else {
+        m.forUpdate = forUpdateType[0]
+    }
+    return m
+}
+
+func (m Query[T]) SelectWithTimeout(duration time.Duration) Query[T] {
+    ms := duration.Milliseconds()
+    if ms > 0 {
+        m.selectTimeout = "/*+ MAX_EXECUTION_TIME(+" + strconv.FormatInt(ms, 10) + "+) */"
+    }
+    return m
 }

orm/query_select_gen.go

@@ -18,7 +18,6 @@ func (m Query[T]) SubQuery() SubQuery {
         tempTable.err = mt.result.Err
 
         return tempTable
-
     } else {
         mt := m
         tempTable := mt.generateSelectQuery(mt.columns...)
@@ -84,7 +83,12 @@ func (m Query[T]) generateSelectQuery(columns ...interface{}) SubQuery {
 
         orderLimitOffsetStr := m.getOrderAndLimitSqlStr()
 
-        rawSql += "select " + selectStr
+        var selectKeyword = "select"
+        if m.selectTimeout != "" {
+            selectKeyword += " " + m.selectTimeout
+        }
+
+        rawSql += selectKeyword + " " + selectStr
 
         if tableStr != "" {
             rawSql += " from " + tableStr