update code

Author: Yuukadesu

.github/workflows/ci.yml

@@ -38,6 +38,9 @@ jobs:
         with:
           shared-key: "linux-build"
 
+      - name: Generate protocol files
+        run: cargo check --package protocol
+
       - name: Check format
         run: cargo fmt --all -- --check
 

Dockerfile

@@ -45,7 +45,9 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 WORKDIR /app
 
 COPY --from=builder /build/target/release/function-stream bin/
-COPY --from=builder /build/python/functionstream-runtime/target/functionstream-python-runtime.wasm data/cache/python-runner/
+COPY --from=builder \
+    /build/python/functionstream-runtime/target/functionstream-python-runtime.wasm \
+    data/cache/python-runner/
 COPY config.yaml conf/
 RUN sed -i 's/127.0.0.1/0.0.0.0/' conf/config.yaml
 

python/functionstream-api/build_package.py

@@ -113,25 +113,30 @@ def clean(self) -> None:
     def build(self) -> None:
         """
         Executes the build process using the PyPA `build` module.
-        Runs from work_dir.parent with work_dir as srcdir so that a local build.py
-        in work_dir cannot shadow "python -m build", avoiding infinite recursion.
+        Runs from work_dir.parent with work_dir as srcdir so that a local
+        build.py in work_dir cannot shadow "python -m build", avoiding
+        infinite recursion.
         """
         logger.info("Building python package...")
 
         self.dist_dir.mkdir(parents=True, exist_ok=True)
 
-        # Run from parent directory and explicitly pass project path to avoid local build.py shadowing PyPA's build
+        # Run from parent directory and explicitly pass project path to avoid
+        # local build.py shadowing PyPA's build
         cmd = [
             self.python_exec, "-m", "build",
             "--outdir", str(self.dist_dir),
             str(self.work_dir),
         ]
-        run_cwd = self.work_dir.parent  # Cannot be work_dir, otherwise "python -m build" will load local build.py causing infinite recursion
+        # Cannot be work_dir, otherwise "python -m build" will load local
+        # build.py causing infinite recursion
+        run_cwd = self.work_dir.parent
 
         logger.info(f"Executing: {' '.join(cmd)}")
 
         try:
-            # Use inherit (stdout/stderr=None) to avoid PIPE buffer full causing subprocess blocking or timeout
+            # Use inherit (stdout/stderr=None) to avoid PIPE buffer full
+            # causing subprocess blocking or timeout
             subprocess.run(
                 cmd,
                 cwd=run_cwd,
@@ -141,7 +146,8 @@ def build(self) -> None:
             )
         except subprocess.CalledProcessError as e:
             logger.error("Build process failed.")
-            # When using inherit, no captured content, prompt user to check terminal output above
+            # When using inherit, no captured content, prompt user to check
+            # terminal output above
             if e.stdout or e.stderr:
                 logger.error(f"STDOUT:\n{e.stdout}")
                 logger.error(f"STDERR:\n{e.stderr}")

python/functionstream-api/src/fs_api/context.py

@@ -46,4 +46,3 @@ def getConfig(self) -> Dict[str, str]:
         pass
 
 __all__ = ['Context']
-

python/functionstream-api/src/fs_api/driver.py

@@ -13,6 +13,7 @@
 import abc
 from .context import Context
 
+
 class FSProcessorDriver(abc.ABC):
 
     @abc.abstractmethod

python/functionstream-api/src/fs_api/store/complexkey.py

@@ -10,7 +10,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-
 from dataclasses import dataclass
 
 

python/functionstream-api/src/fs_api/store/error.py

@@ -10,7 +10,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-
 class KvError(Exception):
     pass
 
@@ -30,5 +29,6 @@ def __init__(self, message: str):
         self.message = message
         super().__init__(message)
 
+
 __all__ = ['KvError', 'KvNotFoundError', 'KvIOError', 'KvOtherError']
 

python/functionstream-api/src/fs_api/store/iterator.py

@@ -26,4 +26,3 @@ def next(self) -> Optional[Tuple[bytes, bytes]]:
         pass
 
 __all__ = ['KvIterator']
-

python/functionstream-api/src/fs_api/store/store.py

@@ -65,7 +65,6 @@ def list_complex(self, key_group: bytes, key: bytes, namespace: bytes,
                      start_inclusive: bytes, end_exclusive: bytes) -> List[bytes]:
         pass
 
-
     @abc.abstractmethod
     def scan_complex(self, key_group: bytes, key: bytes, namespace: bytes) -> KvIterator:
         pass

python/functionstream-client/scripts/codegen.py

@@ -50,7 +50,9 @@ def __init__(self, proto_root: Path, output_dir: Path, proto_files: List[str]):
     def _check_paths(self):
         """Validates input paths."""
         if not self.proto_root.exists():
-            raise FileNotFoundError(f"Proto root directory not found: {self.proto_root}")
+            raise FileNotFoundError(
+                f"Proto root directory not found: {self.proto_root}"
+            )
 
         for p_file in self.proto_files:
             if not (self.proto_root / p_file).exists():
@@ -94,7 +96,9 @@ def _get_protoc_args(self) -> List[str]:
             )
             logger.info("Enabled mypy-protobuf type generation.")
         else:
-            logger.warning("mypy-protobuf not found. Skipping .pyi generation.")
+            logger.warning(
+                "mypy-protobuf not found. Skipping .pyi generation."
+            )
 
         for p_file in self.proto_files:
             args.append(str(self.proto_root / p_file))
@@ -106,15 +110,18 @@ def _resolve_protoc_gen_mypy() -> Optional[str]:
         """
         Resolves the protoc-gen-mypy executable path.
         Protoc looks for plugins via PATH; when run from make/CI, venv bin may
-        not be on PATH. Prefer the executable next to sys.executable, then PATH.
+        not be on PATH. Prefer the executable next to sys.executable,
+        then PATH.
         """
         import sys as _sys
 
         # 1. Same dir as current Python (venv bin when run via make)
         bin_dir = Path(_sys.executable).resolve().parent
         for name in ("protoc-gen-mypy", "protoc-gen-mypy.exe"):
             candidate = bin_dir / name
-            if candidate.exists() and (candidate.is_file() or candidate.is_symlink()):
+            if candidate.exists() and (
+                candidate.is_file() or candidate.is_symlink()
+            ):
                 return str(candidate)
 
         # 2. On PATH
@@ -166,7 +173,9 @@ def run(self):
             logger.info(f"Code generation complete. Output: {self.output_dir}")
 
         except Exception:
-            logger.exception("An unexpected error occurred during code generation.")
+            logger.exception(
+                "An unexpected error occurred during code generation."
+            )
             sys.exit(1)
 
 
@@ -195,7 +204,10 @@ def main():
         "files",
         nargs="*",
         default=["function_stream.proto"],
-        help="Specific .proto files to compile (default: function_stream.proto)",
+        help=(
+            "Specific .proto files to compile "
+            "(default: function_stream.proto)"
+        ),
     )
 
     args = parser.parse_args()