feat: identify eof bug

speed2exe · Oct 26, 2023 · d78d0cf · d78d0cf
1 parent d7e0f1d
commit d78d0cf
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 5 deletions.
diff --git a/integration_tests/client.zig b/integration_tests/client.zig
@@ -9,10 +9,17 @@ test "ping" {
     try c.ping();
 }
 
-test "query" {
+test "query database create and drop" {
     var c = Client.init(test_config, std.testing.allocator);
     defer c.deinit();
 
     try c.query("CREATE DATABASE testdb");
     try c.query("DROP DATABASE testdb");
 }
+
+// test "query select 1" {
+//     var c = Client.init(test_config, std.testing.allocator);
+//     defer c.deinit();
+//
+//     try c.query("show databases");
+// }
diff --git a/src/config.zig b/src/config.zig
@@ -16,11 +16,8 @@ pub const Config = struct {
     pub fn capability_flags(config: *const Config) u32 {
         // zig fmt: off
         var flags: u32 = constants.CLIENT_PROTOCOL_41
-                       | constants.CLIENT_LONG_PASSWORD
-                       | constants.CLIENT_LOCAL_FILES
                        | constants.CLIENT_PLUGIN_AUTH
-                       | constants.CLIENT_MULTI_RESULTS
-                       | constants.CLIENT_CONNECT_ATTRS
+                       // TODO: Support more
                        ;
         // zig fmt: on
         if (config.client_found_rows) {

diff --git a/src/reade b/src/reade
diff --git a/src/stream_buffered.zig b/src/stream_buffered.zig
@@ -23,6 +23,9 @@ pub const Reader = struct {
 
     inline fn fill(s: *Reader) !void {
         s.len = try s.stream.read(&s.buf);
+        if (s.len == 0) {
+            return error.UnexpectedEndOfStream;
+        }
         s.pos = 0;
     }