Duyệt tuần tự từng token - tok_traverse

ntkien36 · Jan 2, 2022 · 604e85d · 604e85d
1 parent e40cb42
commit 604e85d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/str.c b/src/str.c
@@ -10,10 +10,8 @@ const char* ascii_spaces = "\t\n\v\f\r ";
 
 gvec_t str_tokens(char *line, const char *delims) {
   gvec_t tokens = gvec_create(5, NULL);
-  char *tmp = strtok(line, delims);
-  while (tmp) {
-    gvec_append(tokens, gtype_s(tmp));
-    tmp = strtok(NULL, delims);
+  tok_traverse(tk, line, delims) {
+    gvec_append(tokens, gtype_s(tk));
   }
   return tokens;
 }
diff --git a/str.h b/str.h
@@ -12,5 +12,7 @@ extern const char* ascii_spaces;
 gvec_t str_tokens(char *line, const char *delims);
 
 #define str_split(line) str_tokens(line, ascii_spaces)
+#define tok_traverse(cur, line, delims) \
+  for (char *cur = strtok(line, delims); cur; cur = strtok(NULL, delims))
 
 #endif  // STR_H_