From 67d59cedfe07a93aba8a02570b63fb486f440f0e Mon Sep 17 00:00:00 2001
From: "Engelbart Kai Sven (PEA4-Fe)" <KaiSven.Engelbart@de.bosch.com>
Date: Wed, 3 Feb 2021 14:45:41 +0100
Subject: [PATCH] Add prediction model

---
 src/models/prediction.ts | 8 ++++++++
 src/models/user.ts       | 2 ++
 2 files changed, 10 insertions(+)
 create mode 100644 src/models/prediction.ts

diff --git a/src/models/prediction.ts b/src/models/prediction.ts
new file mode 100644
index 0000000..b379eaf
--- /dev/null
+++ b/src/models/prediction.ts
@@ -0,0 +1,8 @@
+import mongoose from 'mongoose';
+
+
+const PredictionSchema = new mongoose.Schema({
+	values: [{ spectrumName: String, value: Number }]
+});
+
+export default mongoose.model<any, mongoose.Model<any, any>>('prediction', PredictionSchema);
diff --git a/src/models/user.ts b/src/models/user.ts
index a70791b..51ad234 100644
--- a/src/models/user.ts
+++ b/src/models/user.ts
@@ -1,6 +1,7 @@
 import mongoose from 'mongoose';
 import db from '../db';
 import ModelModel from './model';
+import PredictionModel from './prediction';
 
 const UserSchema = new mongoose.Schema({
 	name: {type: String, index: {unique: true}},
@@ -11,6 +12,7 @@ const UserSchema = new mongoose.Schema({
 	location: String,
 	devices: [String],
 	models: [{type: mongoose.Schema.Types.ObjectId, ref: ModelModel}],
+	predictions: [{type: mongoose.Schema.Types.ObjectId, ref: PredictionModel}],
 	status: String
 });