From d5ced6bc3fdbb58d19d6633b16b82ede4ee950eb Mon Sep 17 00:00:00 2001
From: Vincent Rischmann
Date: Sun, 30 May 2021 17:00:45 +0200
Subject: set the eval branch quota in the return type block

`prepare` and `prepareWithDiags` parse the query at comptime in the
return type definition.

This exceeds the branch quota since https://github.com/ziglang/zig/commit/b11ac9c5bfb4048eb4aa561f8b3b058a76787f7e.
Use a block and `@setEvalBranchQuota` to work around this.

Fixes #30.
---
 sqlite.zig | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/sqlite.zig b/sqlite.zig
index 79286e3..143c279 100644
--- a/sqlite.zig
+++ b/sqlite.zig
@@ -443,8 +443,11 @@ pub const Db = struct {
     }
 
     /// prepareWithDiags is like `prepare` but takes an additional options argument.
-    pub fn prepareWithDiags(self: *Self, comptime query: []const u8, options: QueryOptions) !Statement(.{}, ParsedQuery.from(query)) {
-        @setEvalBranchQuota(10000);
+    pub fn prepareWithDiags(self: *Self, comptime query: []const u8, options: QueryOptions) !comptime blk: {
+        @setEvalBranchQuota(100000);
+        break :blk Statement(.{}, ParsedQuery.from(query));
+    } {
+        @setEvalBranchQuota(100000);
         const parsed_query = ParsedQuery.from(query);
         return Statement(.{}, comptime parsed_query).prepare(self, options, 0);
     }
@@ -463,8 +466,11 @@ pub const Db = struct {
     /// This is done because we type check the bind parameters when executing the statement later.
     ///
     /// If you want additional error information in case of failures, use `prepareWithDiags`.
-    pub fn prepare(self: *Self, comptime query: []const u8) !Statement(.{}, ParsedQuery.from(query)) {
-        @setEvalBranchQuota(10000);
+    pub fn prepare(self: *Self, comptime query: []const u8) !comptime blk: {
+        @setEvalBranchQuota(100000);
+        break :blk Statement(.{}, ParsedQuery.from(query));
+    } {
+        @setEvalBranchQuota(100000);
         const parsed_query = ParsedQuery.from(query);
         return Statement(.{}, comptime parsed_query).prepare(self, .{}, 0);
     }
@@ -1531,7 +1537,6 @@ test "sqlite: read a single integer value" {
     inline for (types) |typ| {
         const query = "SELECT age FROM user WHERE id = ?{usize}";
 
-        @setEvalBranchQuota(5000);
         var stmt: Statement(.{}, ParsedQuery.from(query)) = try db.prepare(query);
         defer stmt.deinit();
 
-- 
cgit v1.2.3