jnordwick / zig-string

Zig string library that includes small string optimization on the stack

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Introduction

This is a simple string library that implements an in situ buffer for the small string optimization. It can store up to 23 bytes without spill into an external allocation.

Alot of basic functionality, such as looping over the contents of the string, it is easier to just get a slice from the string and use that so things like basic iterators are not implemented, and I feel would just clutter the API, be slower, and be more places for bugs. Often the easiest way of interacting with the string is to use to_slice and to_const_slice. There are some operations that are pushed into the contains SmallString or LargeString where they can take advantage of specific properties of the string. And there are more I definitely want to do that to but haven’t had the time yet.

I tried to use the std.mem functions where possible but sometimes they were just too slow or didn’t fit the API well. In some cases I probably don’t know they even exist. Std is just a mess – I hope there are plans to clean it up.

Implementation

I’ve been kicking around changing the implementation. Since the length of small will always be less than 32 and a single byte, the 6th through 8th bits can be used to signal that the string is in large mode. This means there is no shifting, just see if any of those bits are set. For a large string, the length can’t be controlled so capacity is put first in the struct and it always need to generate a capaciity where of of those bits is set (it does this by zeroing the lower bits and adding 32, so any capacity may be grown up to 32 bit that isn’t too bad).

Allocations

This used to have optional allocator arguments, then I tried anytype, and now I’m trying comptime Allocator with a special null_alloc that just send everything to unreachable. I’m trying to find a way that is both easy (many times you know you don’t need to allocate but the function still needs one) and has top performance. Since these strings tend to be small, having a true vtable jump can begin to dominate is loop. I think this should work, but I’m still searching for that magical zero-cost for both mental load and function call cost.

Extend to small vec and Unicode

It should be possible to use this for u8 numbers already if you just ignore the word string all over the place. It should also be possible to wrap this in a type function and have it use u16 or other, but it becomes much more limited and I think there are better ways to do small vecs. By the time you are looking at sizes up to u32 for a unicode code point, you’re only storing 5 characters (I could add a dummy char to the LargeString that is there solely to get the SmallString up to 6 characters but that just seems trivially small to be useful).

Smallish vectors like three u64, if the padding byte was added to large string would be able to fit, but those cases are probably better handled by just a simple [3]f64 because you aren’t really growing those vectors and needing to worry about spill.

Having 11 bytes of UTF-16 might be useful though, but I despise UTF-16, and the only think I know of that uses it is Windows APIs, and 11 bytes seems too small for filenames or such.

Usage without the union String indirection

Often you will know if you have a small or large string. for example, if you use this as a hash key and know your key length. you can use SmallString and LargeString directly if you want or you can still use String and call into String.small or String.large. There is also some isSmall and isLargeStr predicates to help with that. String.lowbyte is also there to help determine the union type.

Stuff already in std.mem?

I’m tempted to reimplement some of stuff in std.mem or at least specialize it a little better. Some of the code is okay an some of it kind of sucks, but it can easily be used by grabbing the slice and calling it on that most of the time. There are some instances eg where you want to split the string on multiple indexes where that doesn’t work as well anymore though. I’ll add them to here as needed, and I think leave the calls to call the std.mem functions, i think – still not sure on it.

Everything is a slice?

The API might change somewhat substantially by making any function that takes in a string or null terminated string just take a slice instead. This should simplify the API a little.

More string functions

There are a few files with more string comparison functions. Right now there is a Boyer-Moore-Harspool string matcher and a Shift-XOR matcher. The former is great for midsize strings, and the latter is great for small strings. There is small string hash function that is extremely quick to run because all the hash functions in std are more appropriate for much much longer strings like webpages and were absolutely dominating runtimes. I’m not sure how to add those to the API yet, but they will be added, more are on the way (including an N-way Rabin-Karp), and the code is there to use now.

  • [ ] better documentation
  • [ ] refine shift xor search, add to API
  • [ ] N-way Rabin-Karp
  • [ ] Add asserts or other safe build protections

About

Zig string library that includes small string optimization on the stack

License:GNU Affero General Public License v3.0


Languages

Language:Zig 99.6%Language:Makefile 0.4%