Feature/leader schedule (#232)

bloXroute-Labs · Jan 2, 2025 · d826b2c · d826b2c
1 parent 6136429
commit d826b2c
Show file tree

Hide file tree

Showing 9 changed files with 1,852 additions and 1,326 deletions.
diff --git a/.gitignore b/.gitignore
@@ -5,11 +5,12 @@ python/src/bxsolana_trader_proto.egg-info/
 /python/.pypirc
 python/.venv
 target/
+.venv
+.DS_Store
 
 # Remove Cargo.lock from gitignore if creating an executable, leave it for libraries
 # More information here https://doc.rust-lang.org/cargo/guide/cargo-toml-vs-cargo-lock.html
 Cargo.lock
 
 # These are backup files generated by rustfmt
 **/*.rs.bk
-.venv
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "solana-trader-proto"
-version = "0.1.1"
+version = "0.1.2"
 edition = "2021"
 authors = ["[email protected]"]
 license = "MIT"

diff --git a/Makefile b/Makefile
@@ -10,7 +10,7 @@ all: clean proto
 clean:
 	rm -rf js api common
 
-proto: proto-build-api-go proto-build-common-go proto-build-swagger proto-build-gw proto-build-api-python
+proto: proto-build-api-go proto-build-common-go proto-build-swagger proto-build-gw proto-build-api-python proto-build-api-rust
 
 proto-build-gw:
 	docker run -v $(CURDIR)/api:/go/protobuf/out \
@@ -43,6 +43,9 @@ proto-build-api-python:
 		--python_betterproto_out=$(CURDIR)/python/src/bxsolana_trader_proto/ \
 		$(CURDIR)/proto/api.proto
 
+proto-build-api-rust:
+	cargo build
+
 proto-build-common-go:
 	docker run -v $(CURDIR)/common:/go/protobuf/out \
 			   -v $(CURDIR)/proto:/go/protobuf/in $(PB_GO_IMAGE_NAME) \

diff --git a/api/api.pb.go b/api/api.pb.go
diff --git a/api/api.pb.gw.go b/api/api.pb.gw.go
diff --git a/api/api_grpc.pb.go b/api/api_grpc.pb.go
diff --git a/proto/api.proto b/proto/api.proto
@@ -977,6 +977,17 @@ service Api {
       tags: "AMMs";
     };
   }
+
+  rpc GetLeaderSchedule(GetLeaderScheduleRequest) returns (GetLeaderScheduleResponse) {
+    option (google.api.http) = {
+      get: "/api/v2/system/leader-schedule"
+    };
+    option (grpc.gateway.protoc_gen_openapiv2.options.openapiv2_operation) = {
+      description: "Returns leaders schedule for up to 100000 next slots, starting from the current slot. Result can be limited by number of slots.";
+      summary: "Leaders schedule";
+      tags: "System";
+    };
+  }
 }
 
 message GetMarketsRequest {
@@ -2512,3 +2523,20 @@ message PostPumpFunSwapRequestSol {
 message PostPumpFunSwapResponse{
   TransactionMessageV2 transaction = 1;
 }
+
+
+message GetLeaderScheduleRequest {
+  uint64 maxSlots = 1;
+}
+
+message GetLeaderScheduleResponse {
+  uint64 currentSlot = 1;  
+	repeated LeaderSchedule leaderSchedule = 2;
+}
+
+message LeaderSchedule {
+  uint64 slot = 1;
+	string leader = 2;    
+	bool isJito = 3;    
+	bool isLowRisk = 4; 
+}
diff --git a/python/src/bxsolana_trader_proto/api/__init__.py b/python/src/bxsolana_trader_proto/api/__init__.py