Merge pull request #1122 from get10101/feat/rollover-dlc

feat: Rollover DLC

CHANGELOG.md

@@ -12,6 +12,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Add support for push notifications.
 - Added new setting to coordinator to configure max channel size to traders.
 - Speed up DLC channel setup and settlement by checking for messages more often.
+- Add support for perpetual futures.
 
 ## [1.2.0] - 2023-08-04
 

coordinator/Cargo.toml

@@ -15,6 +15,7 @@ console-subscriber = "0.1.6"
 coordinator-commons = { path = "../crates/coordinator-commons" }
 diesel = { version = "2.0.0", features = ["r2d2", "postgres", "time", "uuid"] }
 diesel_migrations = "2.0.0"
+dlc = "0.4.0"
 dlc-manager = { version = "0.4.0", features = ["use-serde"] }
 dlc-messages = "0.4.0"
 dlc-trie = "0.4.0"

coordinator/migrations/2023-08-18-105400_rollover_dlc/down.sql

@@ -0,0 +1,2 @@
+-- This file should undo anything in `up.sql`
+-- Note: There is no down migration for removing the `Rollover` variant that was added to `PositionState_Type` because it is not feasible to remove enum variants in the db!

coordinator/migrations/2023-08-18-105400_rollover_dlc/up.sql

@@ -0,0 +1,4 @@
+-- Your SQL goes here
+ALTER TYPE "PositionState_Type"
+    ADD
+    VALUE IF NOT EXISTS 'Rollover';

coordinator/src/db/custom_types.rs

@@ -44,6 +44,7 @@ impl ToSql<PositionStateType, Pg> for PositionState {
             PositionState::Open => out.write_all(b"Open")?,
             PositionState::Closing => out.write_all(b"Closing")?,
             PositionState::Closed => out.write_all(b"Closed")?,
+            PositionState::Rollover => out.write_all(b"Rollover")?,
         }
         Ok(IsNull::No)
     }
@@ -55,6 +56,7 @@ impl FromSql<PositionStateType, Pg> for PositionState {
             b"Open" => Ok(PositionState::Open),
             b"Closing" => Ok(PositionState::Closing),
             b"Closed" => Ok(PositionState::Closed),
+            b"Rollover" => Ok(PositionState::Rollover),
             _ => Err("Unrecognized enum variant".into()),
         }
     }

coordinator/src/db/positions.rs

@@ -3,6 +3,7 @@ use crate::schema::positions;
 use crate::schema::sql_types::ContractSymbolType;
 use crate::schema::sql_types::PositionStateType;
 use anyhow::bail;
+use anyhow::ensure;
 use anyhow::Result;
 use autometrics::autometrics;
 use bitcoin::hashes::hex::ToHex;
@@ -129,6 +130,25 @@ impl Position {
         Ok(())
     }
 
+    pub fn set_position_to_open(
+        conn: &mut PgConnection,
+        trader_pubkey: String,
+        temporary_contract_id: ContractId,
+    ) -> Result<()> {
+        let affected_rows = diesel::update(positions::table)
+            .filter(positions::trader_pubkey.eq(trader_pubkey))
+            .set((
+                positions::position_state.eq(PositionState::Open),
+                positions::temporary_contract_id.eq(temporary_contract_id.to_hex()),
+                positions::update_timestamp.eq(OffsetDateTime::now_utc()),
+            ))
+            .execute(conn)?;
+
+        ensure!(affected_rows > 0, "Could not set position to open");
+
+        Ok(())
+    }
+
     pub fn update_unrealized_pnl(conn: &mut PgConnection, id: i32, pnl: i64) -> Result<()> {
         let affected_rows = diesel::update(positions::table)
             .filter(positions::id.eq(id))
@@ -145,6 +165,25 @@ impl Position {
         Ok(())
     }
 
+    pub fn rollover_position(
+        conn: &mut PgConnection,
+        trader_pubkey: String,
+        expiry_timestamp: &OffsetDateTime,
+    ) -> Result<()> {
+        let affected_rows = diesel::update(positions::table)
+            .filter(positions::trader_pubkey.eq(trader_pubkey))
+            .set((
+                positions::expiry_timestamp.eq(expiry_timestamp),
+                positions::position_state.eq(PositionState::Rollover),
+                positions::update_timestamp.eq(OffsetDateTime::now_utc()),
+            ))
+            .execute(conn)?;
+
+        ensure!(affected_rows > 0, "Could not set position to rollover");
+
+        Ok(())
+    }
+
     /// inserts the given position into the db. Returns the position if successful
     #[autometrics]
     pub fn insert(
@@ -226,6 +265,7 @@ impl From<crate::position::models::NewPosition> for NewPosition {
 pub enum PositionState {
     Open,
     Closing,
+    Rollover,
     Closed,
 }
 
@@ -254,6 +294,7 @@ impl From<(PositionState, Option<i64>, Option<f32>)> for crate::position::models
                 // `Closed` state
                 pnl: realized_pnl.unwrap_or(0),
             },
+            PositionState::Rollover => crate::position::models::PositionState::Rollover,
         }
     }
 }

coordinator/src/lib.rs

@@ -1,3 +1,15 @@
+use axum::http::StatusCode;
+use axum::response::IntoResponse;
+use axum::response::Response;
+use axum::Json;
+use diesel::PgConnection;
+use diesel_migrations::embed_migrations;
+use diesel_migrations::EmbeddedMigrations;
+use diesel_migrations::MigrationHarness;
+use serde_json::json;
+
+mod rollover;
+
 pub mod admin;
 pub mod cli;
 pub mod db;
@@ -12,16 +24,6 @@ pub mod schema;
 pub mod settings;
 pub mod trade;
 
-use axum::http::StatusCode;
-use axum::response::IntoResponse;
-use axum::response::Response;
-use axum::Json;
-use diesel::PgConnection;
-use diesel_migrations::embed_migrations;
-use diesel_migrations::EmbeddedMigrations;
-use diesel_migrations::MigrationHarness;
-use serde_json::json;
-
 pub const MIGRATIONS: EmbeddedMigrations = embed_migrations!();
 
 pub fn run_migration(conn: &mut PgConnection) {