首页 > 解决方案 > 在 Express 中使用中间件功能的正确方法

问题描述

嗨,我使用带有节点 js 的 express 创建了一个代理端点,我让它在一个 get 请求中正常工作,但是我想设置多个 get/post 请求端点,并想知道我在下面所做的是否是继续实施的正确方法其他端点,我对 NodeJS 和 express 很陌生。

我试过的是重用

app.use(validateFirebaseIdToken);

然后 app.use(new functionname) 在我调用新的 get Endpoint 之前,我需要在我的每个 bigQuery 方法中重用 decodedIDToken 变量,请记住,我将创建很多不同的 bigQuery 查询方法获取最终用户的数据

var express = require('express')`
var app = express()

const validateFirebaseIdToken = async (req, res, next) => {`
console.log('Check if request is authorized with Firebase ID token');

if ((!req.headers.authorization || !req.headers.authorization.startsWith('Bearer ')) &&!(req.cookies && req.cookies.__session)) {
    console.error('No Firebase ID token was passed as a Bearer token in the Authorization header.',
    'Make sure you authorize your request by providing the following HTTP header:',
    'Authorization: Bearer <Firebase ID Token>',
    'or by passing a "__session" cookie.');
    res.status(403).send('Unauthorized');
    return;
}

let idToken;
if (req.headers.authorization && req.headers.authorization.startsWith('Bearer ')) {
    console.log('Found "Authorization" header');
    // Read the ID Token from the Authorization header.
    idToken = req.headers.authorization.split('Bearer ')[1];
} else {
// No cookie
   res.status(403).send('Unauthorized');
   return;
}

try {
   const decodedIdToken = await admin.auth().verifyIdToken(idToken);
   console.log('ID Token correctly decoded', decodedIdToken);
   req.user = decodedIdToken;
   next();
   return;
} catch (error) {
   console.error('Error while verifying Firebase ID token:', error);
   res.status(403).send('Unauthorized');
   return;
} 
};

const runDailyCategorybigQuery = async (req, res, next) => {

const query = `select count(string_field_3) as Categories, Month(date_field_2) as Month from test.testing_api group by Month`;

const options = {
   query: query,
   // Location must match that of the dataset(s) referenced in the query.
   location: 'US',
   useLegacySql: true,
};

// Run the query as a job
const [job] = await bigquery.createQueryJob(options);
console.log(`Job ${job.id} started.`);
console.log(`ID ${req.user.user_id}`);

// Wait for the query to finish
const [rows] = await job.getQueryResults();
res.query = rows;
console.log('Rows:');
rows.forEach(row => console.log(row));
next();

};

const runCategoryMonthsbigQuery = async (req, res, next) => {

const query = `select count(string_field_3) as Categories, Month(date_field_2) as Month from test.testing_api group by Month`;

const options = {
   query: query,
   // Location must match that of the dataset(s) referenced in the query.
   location: 'US',
   useLegacySql: true,
};

// Run the query as a job
const [job] = await bigquery.createQueryJob(options);
console.log(`Job ${job.id} started.`);
console.log(`ID ${req.user.user_id}`);

// Wait for the query to finish
const [rows] = await job.getQueryResults();
res.query = rows;
console.log('Rows:');
rows.forEach(row => console.log(row));
next();

};
app.use(validateFirebaseIdToken);
app.use(runDailyCategorybigQuery);
app.get('/getCategories', (req, res) => {
res.json(res.query);
});

//what im thinking of doing
app.use(validateFirebaseIdToken);
app.use(runCategoryMonthsbigQuery);
app.get('/getCategoriesMonth', (req, res) => {
res.json(res.query);
});

标签: node.jsexpressmiddlewareendpoint

解决方案


您列为“我正在考虑做的事情”将添加两次相同的中间件(用于 validateFirebaseIdToken)。这不是必需的,因为每个请求都会调用中间件,因此您应该只添加一次。

第二个问题是您将请求处理程序视为中间件。由于所有中间件都会看到每个请求(只要它们调用 next),那么您将同时运行两个查询,第二个查询将覆盖第一个。

将您的两个查询中间件函数移动到请求处理程序中。例如:

const runDailyCategorybigQuery = async (req, res) => {
  ...
  res.json(res.query);
}

app.get('/getCategories', runDailyCategorybigQuery);


推荐阅读