From 1e83a5cc4e074db78277cf77aacbeae6771a0696 Mon Sep 17 00:00:00 2001
From: An Tao <antao2002@gmail.com>
Date: Tue, 10 Sep 2019 15:58:42 +0800
Subject: [PATCH] Fix a bug of creating models for mysql (#237)

---
 drogon_ctl/templates/model_cc.csp | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/drogon_ctl/templates/model_cc.csp b/drogon_ctl/templates/model_cc.csp
index a3902aa2..7e0172b2 100644
--- a/drogon_ctl/templates/model_cc.csp
+++ b/drogon_ctl/templates/model_cc.csp
@@ -209,7 +209,7 @@ const std::string &[[className]]::getColumnName(size_t index) noexcept(false)
         }
         $$<<"\n";    
     }
-    if(@@.get<std::string>("rdbms")=="postgresql")
+    if(@@.get<std::string>("rdbms")=="postgresql"||@@.get<int>("hasPrimaryKey")!=1)
     {
         $$<<"void "<<className<<"::updateId(const uint64_t id)\n";
         $$<<"{\n";
@@ -217,13 +217,21 @@ const std::string &[[className]]::getColumnName(size_t index) noexcept(false)
     }
     else if(@@.get<std::string>("rdbms")=="mysql"||@@.get<std::string>("rdbms")=="sqlite3")
     {
+        auto primaryKeyTypeString=@@.get<std::string>("primaryKeyType");
         $$<<"void "<<className<<"::updateId(const uint64_t id)\n";
         $$<<"{\n";
         for(auto col:cols)
         {
             if(col._isAutoVal)
             {
-                $$<<"    _"<<col._colValName<<" = std::make_shared<uint64_t>(id);\n";
+                if(primaryKeyTypeString!="uint64_t")
+                {
+                    $$<<"    _"<<col._colValName<<" = std::make_shared<"<<primaryKeyTypeString<<">(static_cast<"<<primaryKeyTypeString<<">(id));\n";
+                }
+                else
+                {
+                    $$<<"    _"<<col._colValName<<" = std::make_shared<uint64_t>(id);\n";
+                }
                 break;
             }
         }