38054-vm/backend/src/db/models/datasets.js
2026-02-01 08:35:56 +00:00

175 lines
2.4 KiB
JavaScript

const config = require('../../config');
const providers = config.providers;
const crypto = require('crypto');
const bcrypt = require('bcrypt');
const moment = require('moment');
module.exports = function(sequelize, DataTypes) {
const datasets = sequelize.define(
'datasets',
{
id: {
type: DataTypes.UUID,
defaultValue: DataTypes.UUIDV4,
primaryKey: true,
},
title: {
type: DataTypes.TEXT,
},
description: {
type: DataTypes.TEXT,
},
timeframe: {
type: DataTypes.ENUM,
values: [
"1m",
"5m",
"15m",
"1h",
"4h",
"1d"
],
},
source: {
type: DataTypes.TEXT,
},
candles_count: {
type: DataTypes.INTEGER,
},
importHash: {
type: DataTypes.STRING(255),
allowNull: true,
unique: true,
},
},
{
timestamps: true,
paranoid: true,
freezeTableName: true,
},
);
datasets.associate = (db) => {
/// loop through entities and it's fields, and if ref === current e[name] and create relation has many on parent entity
db.datasets.hasMany(db.candles, {
as: 'candles_dataset',
foreignKey: {
name: 'datasetId',
},
constraints: false,
});
db.datasets.hasMany(db.replay_sessions, {
as: 'replay_sessions_dataset',
foreignKey: {
name: 'datasetId',
},
constraints: false,
});
db.datasets.hasMany(db.trades, {
as: 'trades_dataset',
foreignKey: {
name: 'datasetId',
},
constraints: false,
});
//end loop
db.datasets.belongsTo(db.instruments, {
as: 'instrument',
foreignKey: {
name: 'instrumentId',
},
constraints: false,
});
db.datasets.belongsTo(db.organizations, {
as: 'organizations',
foreignKey: {
name: 'organizationsId',
},
constraints: false,
});
db.datasets.hasMany(db.file, {
as: 'file',
foreignKey: 'belongsToId',
constraints: false,
scope: {
belongsTo: db.datasets.getTableName(),
belongsToColumn: 'file',
},
});
db.datasets.belongsTo(db.users, {
as: 'createdBy',
});
db.datasets.belongsTo(db.users, {
as: 'updatedBy',
});
};
return datasets;
};