Add standardised SQL connection setup for microservices.
This commit is contained in:
@@ -1,11 +1,13 @@
|
||||
from slingshot_microservice.typing import ReadFileFn, WriteFileFn
|
||||
from slingshot_microservice import Microservice
|
||||
from typing import Generator
|
||||
from sqlalchemy.engine.base import Connection
|
||||
|
||||
def process(
|
||||
request: int,
|
||||
read_file: ReadFileFn,
|
||||
write_file: WriteFileFn,
|
||||
_connection: Connection,
|
||||
) -> Generator[tuple[int, bool | int | str], None, None]:
|
||||
reader = read_file("in", request)
|
||||
input_data = reader.read().decode()
|
||||
|
||||
@@ -1,10 +1,12 @@
|
||||
use slingshot_microservice::{AnyError, Microservice, ReadFileFn, WriteFileFn};
|
||||
use diesel::pg::PgConnection;
|
||||
use std::io::{Read, Write};
|
||||
|
||||
fn process(
|
||||
request: u64,
|
||||
read_file: &ReadFileFn,
|
||||
write_file: &WriteFileFn,
|
||||
_connection: &mut PgConnection,
|
||||
) -> Result<Vec<(u64, String)>, AnyError> {
|
||||
let mut input = String::new();
|
||||
let mut reader = read_file("in", request)?;
|
||||
|
||||
Reference in New Issue
Block a user