Skip to content

Commit 4991847

Browse files
committed
server: Refactor code around service_fn
1 parent bacf0d2 commit 4991847

File tree

4 files changed

+27
-42
lines changed

4 files changed

+27
-42
lines changed

core/src/subgraph/registrar.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -256,7 +256,7 @@ where
256256
let sender = sender.clone();
257257
let provider = provider.clone();
258258
let logger = logger.clone();
259-
259+
260260
// Blocking due to store interactions. Won't be blocking after #905.
261261
graph::spawn_blocking(
262262
start_subgraph(id, &*provider, logger)

server/http/src/server.rs

Lines changed: 8 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -86,21 +86,14 @@ where
8686
let store = self.store.clone();
8787
let node_id = self.node_id.clone();
8888
let new_service = make_service_fn(move |_| {
89-
let logger_for_service = logger_for_service.clone();
90-
let metrics = metrics.clone();
91-
let graphql_runner = graphql_runner.clone();
92-
let store = store.clone();
93-
let node_id = node_id.clone();
94-
async move {
95-
Result::<_, Error>::Ok(GraphQLService::new(
96-
logger_for_service.clone(),
97-
metrics.clone(),
98-
graphql_runner.clone(),
99-
store.clone(),
100-
ws_port,
101-
node_id.clone(),
102-
))
103-
}
89+
futures03::future::ok::<_, Error>(GraphQLService::new(
90+
logger_for_service.clone(),
91+
metrics.clone(),
92+
graphql_runner.clone(),
93+
store.clone(),
94+
ws_port,
95+
node_id.clone(),
96+
))
10497
});
10598

10699
// Create a task to run the server and handle HTTP requests

server/index-node/src/server.rs

Lines changed: 6 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -81,18 +81,12 @@ where
8181
let store = self.store.clone();
8282
let node_id = self.node_id.clone();
8383
let new_service = make_service_fn(move |_| {
84-
let logger_for_service = logger_for_service.clone();
85-
let graphql_runner = graphql_runner.clone();
86-
let store = store.clone();
87-
let node_id = node_id.clone();
88-
async move {
89-
Result::<_, Error>::Ok(IndexNodeService::new(
90-
logger_for_service.clone(),
91-
graphql_runner.clone(),
92-
store.clone(),
93-
node_id.clone(),
94-
))
95-
}
84+
futures03::future::ok::<_, Error>(IndexNodeService::new(
85+
logger_for_service.clone(),
86+
graphql_runner.clone(),
87+
store.clone(),
88+
node_id.clone(),
89+
))
9690
});
9791

9892
// Create a task to run the server and handle HTTP requests

server/metrics/src/lib.rs

Lines changed: 12 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -63,22 +63,20 @@ impl MetricsServerTrait for PrometheusMetricsServer {
6363
let server = self.clone();
6464
let new_service = make_service_fn(move |_req| {
6565
let server = server.clone();
66+
let registry = server.registry.clone();
6667
async move {
6768
Ok::<_, Error>(service_fn(move |_| {
68-
let registry = server.registry.clone();
69-
async move {
70-
let metric_families = registry.gather();
71-
let mut buffer = vec![];
72-
let encoder = TextEncoder::new();
73-
encoder.encode(&metric_families, &mut buffer).unwrap();
74-
Ok::<_, Error>(
75-
Response::builder()
76-
.status(200)
77-
.header(hyper::header::CONTENT_TYPE, encoder.format_type())
78-
.body(Body::from(buffer))
79-
.unwrap(),
80-
)
81-
}
69+
let metric_families = registry.gather();
70+
let mut buffer = vec![];
71+
let encoder = TextEncoder::new();
72+
encoder.encode(&metric_families, &mut buffer).unwrap();
73+
futures03::future::ok::<_, Error>(
74+
Response::builder()
75+
.status(200)
76+
.header(hyper::header::CONTENT_TYPE, encoder.format_type())
77+
.body(Body::from(buffer))
78+
.unwrap(),
79+
)
8280
}))
8381
}
8482
});

0 commit comments

Comments
 (0)