添加 '3.6.cpp'

3.6.cpp

@@ -0,0 +1,193 @@
+#include "MySQLWrapper.h"
+#include <iostream>
+#include <sstream>
+#include <vector>
+#include <map>
+#include <set>
+#include <cstdlib>
+#include <cstdio>
+#include <cstring>
+#include <algorithm>
+#include <fstream>
+using namespace std;
+
+#include "local_db_account.h"
+#include "util.hpp"
+
+int main()
+{
+    /// 3.6 计算机专业不同编程语言平均薪资柱状图. Based on 3.5 code.
+
+    MySQLConn conn;
+    conn.connect("127.0.0.1",db_user,db_passwd,"bigdata3",3306);
+    cout<<"Connected to DB"<<endl;
+    conn.exec("set names gbk",nullptr);
+
+    /// Based on TIOBE 2017
+    vector<vector<string>> langs
+    {
+        { "Java" },
+        { "C++" },
+        { "C#" },
+        { "Python", "Py" },
+        { "PHP" } ,
+        { "Javascript", "JS" },
+        { "Swift" },
+        { "Ruby" },
+        { "Perl" },
+        { "Visual Basic","VisualBasic","VB" },
+        { "Go" },
+        { "MATLAB" }
+    };
+
+    map<string,int> mp;
+
+    int szLang=langs.size();
+    for(int i=0;i<szLang;i++)
+    {
+        int szAlias=langs[i].size();
+        for(int j=0;j<szAlias;j++)
+        {
+            ostringstream ostr;
+            ostr<<"select count(*) from lagou_job where ((title like '%"
+            <<langs[i][j]<<"%') or (tags like '%"
+            <<langs[i][j]<<"%'))";
+
+            string sqlcmd=ostr.str();
+
+            auto fn1=[&](const string& SQLCMD) {
+                conn.exec(SQLCMD,[&](MySQLResult& res)
+                      {
+                          res.stepRow([&](char** val,unsigned long* len)
+                                      {
+                                          int cc=ParseInt(val[0]);
+
+                                          cout<<SQLCMD<<" "<<langs[i][0]<<"-"<<cc<<endl;
+
+                                          auto iter=mp.find(langs[i][0]);
+                                          if(iter!=mp.end())
+                                          {
+                                              iter->second += cc;
+                                          }
+                                          else
+                                          {
+                                              mp.insert(make_pair(langs[i][0],cc));
+                                          }
+                                      });
+                      });
+            };
+
+            conn.exec("use bigdata3",nullptr);
+            fn1(sqlcmd);
+
+            ostr.str("");
+            ostr<<"select count(*) from pages where ((title like '%"
+            <<langs[i][j]<<"%') or (tags like '%"
+            <<langs[i][j]<<"%'))";
+
+            sqlcmd=ostr.str();
+
+            conn.exec("use bigdata5",nullptr);
+            fn1(sqlcmd);
+        }
+    }
+
+    vector<pair<string,int>> vec;
+    vec.insert(vec.end(),mp.begin(),mp.end()); /// ??? Not Naive Code?
+    sort(vec.begin(),vec.end(),[](const pair<string,int>& a,const pair<string,int>& b)
+         {
+             return a.second > b.second;
+         });
+
+    /// ----------- Above : 3.5 --------------
+    /// ----------- Below : 3.6 --------------
+
+    auto fn_getindex=[&](const string& LangMainLable)
+    {
+        int sz=langs.size();
+        for(int i=0;i<sz;i++)
+        {
+            if(langs[i][0]==LangMainLable)
+            {
+                return i;
+            }
+        }
+        return -1;
+    };
+
+    map<string,vector<cash_pack>> cashmp;
+
+    int szLoop=min(10u,vec.size());
+    for(int i=0;i<szLoop;i++)
+    {
+        int idx=fn_getindex(vec[i].first);
+
+        /// Search for salary.
+        int szAlias=langs[idx].size();
+        for(int j=0;j<szAlias;j++)
+        {
+            ostringstream ostr;
+            ostr<<"select salary from lagou_job where ((title like '%"
+            <<langs[idx][j]<<"%') or (tags like '%"
+            <<langs[idx][j]<<"%'))";
+
+            string sqlcmd=ostr.str();
+
+            conn.exec("use bigdata3",nullptr);
+            conn.exec(sqlcmd,[&](MySQLResult& res)
+                      {
+                          res.stepRow([&](char** val,unsigned long* len)
+                                      {
+                                          cash_pack pk=ParseCash(val[0]);
+                                          cashmp[vec[i].first].push_back(pk);
+                                      });
+                      });
+
+
+
+            ostr<<"select salary from pages where ((title like '%"
+            <<langs[idx][j]<<"%') or (tags like '%"
+            <<langs[idx][j]<<"%'))";
+
+            sqlcmd=ostr.str();
+            conn.exec("use bigdata5",nullptr);
+            conn.exec(sqlcmd,[&](MySQLResult& res)
+                      {
+                          res.stepRow([&](char** val,unsigned long* len)
+                                      {
+                                          cash_pack pk=ParseCash(val[0]);
+                                          cashmp[vec[i].first].push_back(pk);
+                                      });
+                      });
+        }
+    }
+
+    ofstream ofs("result/3.6.csv");
+    #define cout ofs
+    for(int i=0;i<szLoop;i++)
+    {
+        int avLow=0;
+        int avHigh=0;
+        int avMid=0;
+
+        vector<cash_pack>& vecref=cashmp[vec[i].first];
+        int szVec=vecref.size();
+        for(const auto& ky:vecref)
+        {
+            avLow+=ky.low;
+            avHigh+=ky.high;
+        }
+
+        if(szVec!=0)
+        {
+            avLow/=szVec;
+            avHigh/=szVec;
+            avMid=(avLow+avHigh)/2;
+        }
+
+        cout<<vec[i].first<<","<<avLow<<","<<avHigh<<","<<avMid<<endl;
+    }
+    #undef cout
+
+    return 0;
+}