From f3608d99ad9046eaae0d47f66cc67f740669e93b Mon Sep 17 00:00:00 2001
From: andy <andynewjack@protonmail.com>
Date: Mon, 26 Oct 2020 23:39:03 -0300
Subject: [PATCH] add armor_slot to postgres model for unit

---
 src/entity/gateway/postgres/models.rs | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/entity/gateway/postgres/models.rs b/src/entity/gateway/postgres/models.rs
index 7d508d6..ea8fc21 100644
--- a/src/entity/gateway/postgres/models.rs
+++ b/src/entity/gateway/postgres/models.rs
@@ -384,6 +384,7 @@ impl Into<shield::Shield> for PgShield {
 pub struct PgUnit {
     unit: unit::UnitType,
     modifier: Option<unit::UnitModifier>,
+    armor_slot: u8,
 }
 
 impl From<unit::Unit> for PgUnit {
@@ -391,6 +392,7 @@ impl From<unit::Unit> for PgUnit {
         PgUnit {
             unit: other.unit,
             modifier: other.modifier,
+            armor_slot: other.armor_slot,
         }
     }
 }
@@ -400,6 +402,7 @@ impl Into<unit::Unit> for PgUnit {
         unit::Unit {
             unit: self.unit,
             modifier: self.modifier,
+            armor_slot: self.armor_slot,
         }
     }
 }