codezm 的个人博客

const fs = require('fs');
const path = require('path');
const { check: checkFilter, update: updateFilter } = require('./filter');

const MAX_LENGTH = 10;
const noop = () => {};
var mysql  = require('mysql');

var connection = mysql.createConnection({
	host     : '127.0.0.1',
	user     : 'root',
	password : '1234',
	port: '3306',
	database: 'test',
	charset: 'UTF8MB4_BIN'
});

const formatDate = function (now) {
	var year = now.getFullYear();
	var month = now.getMonth() + 1;
	var date = now.getDate();
	var hour = now.getHours();
	var minute = now.getMinutes();
	var second = now.getSeconds();

	return year + "-" + month + "-" + date + " " + hour + ":" + minute + ":" + second;
};

connection.connect();

const insertSql = 'INSERT INTO douyin(sid,name,digg,comment,share,create_time,duration,update_time) VALUES(?,?,?,?,?,?,?,?) ON DUPLICATE KEY UPDATE digg = ?, comment = ?, share = ?, create_time = ?, update_time = ?';

module.exports = (server, { storage }) => {
  let sessions = [];
  let timer;
  const writeSessions = (dir) => {
    try {
      const text = JSON.stringify(sessions.slice(), null, '  ');
      sessions = [];
      //dir = path.resolve(dir, `${Date.now()}.txt`);
      dir = path.resolve(dir, `浪涨小青岛.txt`);
      fs.writeFile(dir, text, (err) => {
        if (err) {
          fs.writeFile(dir, text, noop);
        }
      });
    } catch (e) {}
  };

  updateFilter(storage.getProperty('filterText'));
  server.on('request', (req) => {
    // filter
    const active = storage.getProperty('active');
    if (!active) {
      return;
    }
    const dir = storage.getProperty('sessionsDir');
    if (!dir || typeof dir !== 'string') {
      sessions = [];
      return;
    }
    if (!checkFilter(req.originalReq.url)) {
      return;
    }
    req.getSession((s) => {
      if (!s) {
        return;
      }

      var resBody = JSON.parse(s.res.body);
      //var currentTime = Date.now().toString().substr(0, 10);
      var currentTime = parseInt(Date.now() / 1000);
      for(var i=0;i<resBody.aweme_list.length; i++) {
	      console.log(
		      resBody.aweme_list[i].aweme_id,
		      resBody.aweme_list[i].create_time,
		      resBody.aweme_list[i].desc,
		      resBody.aweme_list[i].statistics.digg_count,
		      resBody.aweme_list[i].statistics.comment_count,
		      resBody.aweme_list[i].statistics.share_count,
		      formatDate(
		      	new Date(parseInt(resBody.aweme_list[i].create_time + "000"))
		      )
	      );

		  var  insertSqlData = [
			  resBody.aweme_list[i].aweme_id,
			  resBody.aweme_list[i].desc,
			  resBody.aweme_list[i].statistics.digg_count,
			  resBody.aweme_list[i].statistics.comment_count,
			  resBody.aweme_list[i].statistics.share_count,
			  resBody.aweme_list[i].create_time,
			  resBody.aweme_list[i].duration,
			  currentTime,
        // update
			  resBody.aweme_list[i].statistics.digg_count,
			  resBody.aweme_list[i].statistics.comment_count,
			  resBody.aweme_list[i].statistics.share_count,
			  resBody.aweme_list[i].create_time,
			  currentTime,
		  ];
		  connection.query(insertSql, insertSqlData, function (err, result) {
			  if(err){
				  console.log('[SQL ERROR]', err.message);
				  return;
			  }

			  console.log('INSERT OR UPDATE:', result.insertId, result.affectedRows);
		  });
        
    });
  });
};