sample middleware

This commit is contained in:
Kallesh B S 2025-04-08 16:03:54 +05:30
parent 32258090f5
commit e37f811d7e
6 changed files with 38 additions and 10 deletions

View File

@ -1,12 +1,19 @@
import { Module } from '@nestjs/common';
import { MiddlewareConsumer, Module } from '@nestjs/common';
import { DbModule } from './db/db.module';
import { OracleModule } from './oracle/oracle.module';
import { AuthModule } from './auth/auth.module';
import { MssqlModule } from './mssql/mssql.module';
import { OriginCheckMiddleware } from './middleware/OriginCheck.middleware';
@Module({
imports: [MssqlModule, AuthModule, DbModule, OracleModule],
controllers: [],
providers: [],
})
export class AppModule {}
export class AppModule {
configure(consumer: MiddlewareConsumer) {
consumer
.apply(OriginCheckMiddleware)
.forRoutes('*'); // Apply to all routes
}
}

View File

@ -0,0 +1,21 @@
import { Injectable, Logger, NestMiddleware } from '@nestjs/common';
import { Request, Response, NextFunction } from 'express';
@Injectable()
export class OriginCheckMiddleware implements NestMiddleware {
use(req: Request, res: Response, next: NextFunction) {
const allowedOrigin = 'http://your-specified-domain.com'; // Replace with your domain
const origin = req.headers.origin;
const logger = new Logger('OriginCheckMiddleware');
logger.log('user-agent', req.headers['user-agent']);
// console.log(req.headers['user-agent']);
// if (origin === allowedOrigin) {
next();
// } else {
// res.status(403).send('Access denied');
// }
}
}

View File

@ -17,7 +17,7 @@ export class CarnetSequenceService {
const request = new Request(connection);
request.input('P_SPID', mssql.Int, body.p_spid);
const result = await request.execute('carnetsys.GETCARNETSEQUENCE');
return { data: result.recordsets };
return { data: result.recordset };
} catch (error) {
console.error('Error executing stored procedure:', error);

View File

@ -15,7 +15,7 @@ export class RegionService {
connection = await this.mssqlDBService.getConnection();
const request = new Request(connection);
const result = await request.execute('carnetsys.GETREGIONS');
return { data: result.recordsets };
return { data: result.recordset };
} catch (error) {
console.error('Error executing stored procedure:', error);
@ -35,7 +35,7 @@ export class RegionService {
request.input('P_REGION', mssql.VarChar(30), body.p_region);
request.input('P_NAME', mssql.VarChar(30), body.p_name);
const result = await request.execute('carnetsys.INSERTNEWREGION');
return { data: result.recordsets };
return { data: result.recordset };
} catch (error) {
console.error('Error executing stored procedure:', error);

View File

@ -17,7 +17,7 @@ export class SpContactsService {
const request = new Request(connection);
request.input('P_SPID', mssql.Int, body.p_SPid);
const result = await request.execute('carnetsys.GETSPALLCONTACTS');
return { data: result.recordsets };
return { data: result.recordset };
} catch (error) {
console.error('Error executing stored procedure:', error);
@ -36,7 +36,7 @@ export class SpContactsService {
const request = new Request(connection);
request.input('P_SPID', mssql.Int, body.p_SPid);
const result = await request.execute('carnetsys.GETSPDEFAULTCONTACTS');
return { data: result.recordsets };
return { data: result.recordset };
} catch (error) {
console.error('Error executing stored procedure:', error);
@ -55,7 +55,7 @@ export class SpContactsService {
const request = new Request(connection);
request.input('P_SPCONTACTID', mssql.Int, body.p_spcontactid);
const result = await request.execute('carnetsys.INACTIVATESPCONTACTS');
return { data: result.recordsets };
return { data: result.recordset };
} catch (error) {
console.error('Error executing stored procedure:', error);

View File

@ -16,7 +16,7 @@ export class SpService {
connection = await this.mssqlDBService.getConnection();
const request = new Request(connection);
const result = await request.execute('carnetsys.GETALLSPS');
return { data: result.recordsets };
return { data: result.recordset };
} catch (error) {
console.error('Error executing stored procedure:', error);
@ -35,7 +35,7 @@ export class SpService {
const request = new Request(connection);
request.input('P_SPID', mssql.Int, body.p_spid);
const result = await request.execute('carnetsys.GETSPBYSPID');
return { data: result.recordsets };
return { data: result.recordset };
} catch (error) {
console.error('Error executing stored procedure:', error);