forked from GoogleCloudPlatform/nodejs-docs-samples
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathapp.js
102 lines (90 loc) · 2.49 KB
/
app.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
/**
* Copyright 2016, Google, Inc.
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
// [START app]
'use strict';
// [START setup]
const fs = require('fs');
const express = require('express');
const crypto = require('crypto');
const path = require('path');
const app = express();
app.enable('trust proxy');
const FILENAME = path.join(__dirname, 'seen.txt');
// [END setup]
// [START insertVisit]
/**
* Store a visit record on disk.
*
* @param {object} visit The visit record to insert.
*/
function insertVisit (visit) {
return new Promise((resolve, reject) => {
fs.appendFile(FILENAME, `${JSON.stringify(visit)}\n`, (err) => {
if (err) {
reject(err);
return;
}
resolve();
});
});
}
// [END insertVisit]
// [START getVisits]
/**
* Retrieve the latest 10 visit records from disk.
*/
function getVisits () {
return new Promise((resolve, reject) => {
fs.readFile(FILENAME, { encoding: 'utf8' }, (err, data) => {
if (err) {
reject(err);
return;
}
const visits = data.split('\n')
.filter((line) => line)
.map(JSON.parse)
.map((visit) => `Time: ${visit.timestamp}, AddrHash: ${visit.userIp}`);
resolve(visits);
});
});
}
// [END getVisits]
app.get('/', (req, res, next) => {
// Create a visit record to be stored on disk
const visit = {
timestamp: new Date(),
// Store a hash of the visitor's ip address
userIp: crypto.createHash('sha256').update(req.ip).digest('hex').substr(0, 7)
};
insertVisit(visit)
// Query the last 10 visits from disk.
.then(() => getVisits())
.then((visits) => {
res
.status(200)
.set('Content-Type', 'text/plain')
.send(`Last 10 visits:\n${visits.join('\n')}`);
})
.catch(next);
});
// [START listen]
const PORT = process.env.PORT || 8080;
app.listen(PORT, () => {
console.log(`App listening on port ${PORT}`);
console.log('Press Ctrl+C to quit.');
});
// [END listen]
// [END app]
module.exports = app;