feat(core): scheduler access token ENG-718

src/config/constants.js

@@ -27,6 +27,7 @@ module.exports = {
   'Settings:UserTokenExpirationIntervalSeconds': 3600,
   'Settings:FogTokenExpirationIntervalSeconds': 3600,
   'Settings:KubeletTokenExpirationIntervalSeconds': 3600,
+  'Settings:SchedulerTokenExpirationIntervalSeconds': 3600,
   'Settings:FogStatusUpdateIntervalSeconds': 120,
   'Settings:FogStatusFrequencySeconds': 60,
 

src/sequelize/managers/scheduler-access-token-manager.js

@@ -0,0 +1,25 @@
+/*
+ *  *******************************************************************************
+ *  * Copyright (c) 2019 Edgeworx, Inc.
+ *  *
+ *  * This program and the accompanying materials are made available under the
+ *  * terms of the Eclipse Public License v. 2.0 which is available at
+ *  * http://www.eclipse.org/legal/epl-2.0
+ *  *
+ *  * SPDX-License-Identifier: EPL-2.0
+ *  *******************************************************************************
+ *
+ */
+
+const BaseManager = require('./base-manager')
+const models = require('./../models')
+const SchedulerAccessToken = models.SchedulerAccessToken
+
+class SchedulerAccessTokenManager extends BaseManager {
+  getEntity() {
+    return SchedulerAccessToken
+  }
+}
+
+const instance = new SchedulerAccessTokenManager()
+module.exports = instance

src/sequelize/migrations/20190227154512-create-scheduler-access-token.js

@@ -0,0 +1,31 @@
+'use strict'
+module.exports = {
+  up: (queryInterface, Sequelize) => {
+    return queryInterface.createTable('SchedulerAccessTokens', {
+      id: {
+        type: Sequelize.INTEGER,
+        primaryKey: true,
+        autoIncrement: true,
+        allowNull: false,
+        field: 'id',
+      },
+      expirationTime: {
+        type: Sequelize.BIGINT,
+        field: 'expiration_time',
+      },
+      token: {
+        type: Sequelize.TEXT,
+        field: 'token',
+      },
+      userId: {
+        type: Sequelize.INTEGER,
+        field: 'user_id',
+        references: {model: 'Users', key: 'id'},
+        onDelete: 'cascade',
+      },
+    })
+  },
+  down: (queryInterface, Sequelize) => {
+    return queryInterface.dropTable('SchedulerAccessTokens')
+  },
+}

src/sequelize/models/scheduleraccesstoken.js

@@ -0,0 +1,34 @@
+'use strict'
+module.exports = (sequelize, DataTypes) => {
+  const SchedulerAccessToken = sequelize.define('SchedulerAccessToken', {
+    id: {
+      type: DataTypes.INTEGER,
+      primaryKey: true,
+      autoIncrement: true,
+      allowNull: false,
+      field: 'id',
+    },
+    expirationTime: {
+      type: DataTypes.BIGINT,
+      field: 'expiration_time',
+    },
+    token: {
+      type: DataTypes.TEXT,
+      field: 'token',
+    },
+  }, {
+    timestamps: false,
+    underscored: true,
+  })
+  SchedulerAccessToken.associate = function(models) {
+    SchedulerAccessToken.belongsTo(models.User, {
+      foreignKey: {
+        name: 'userId',
+        field: 'user_id',
+      },
+      as: 'user',
+      onDelete: 'cascade',
+    })
+  }
+  return SchedulerAccessToken
+}

src/services/kubelet-service.js

@@ -12,6 +12,7 @@
  */
 
 const KubeletAccessTokenService = require('./kubelet-access-token-service')
+const SchedulerAccessTokenService = require('./scheduler-access-token-service')
 const TransactionDecorator = require('../decorators/transaction-decorator')
 
 const kubeletCreatePod = async function(createPodData, fogNodeUuid, transaction) {
@@ -54,7 +55,13 @@ const kubeletGetVkToken = async function(userId, transaction) {
   }
 }
 const kubeletGetSchedulerToken = async function(transaction) {
-  // TODO: to implement
+  const newAccessToken = await SchedulerAccessTokenService.generateAccessToken(transaction)
+  await SchedulerAccessTokenService.updateAccessToken(userId, newAccessToken, transaction)
+
+  return {
+    userId: userId,
+    token: newAccessToken.token,
+  }
 }
 
 module.exports = {

src/services/scheduler-access-token-service.js

@@ -0,0 +1,48 @@
+/*
+ * *******************************************************************************
+ *  * Copyright (c) 2019 Edgeworx, Inc.
+ *  *
+ *  * This program and the accompanying materials are made available under the
+ *  * terms of the Eclipse Public License v. 2.0 which is available at
+ *  * http://www.eclipse.org/legal/epl-2.0
+ *  *
+ *  * SPDX-License-Identifier: EPL-2.0
+ *  *******************************************************************************
+ *
+ */
+
+const AppHelper = require('../helpers/app-helper')
+const SchedulerAccessTokenManager = require('../sequelize/managers/scheduler-access-token-manager')
+
+const Config = require('../config')
+
+const generateAccessToken = async function(transaction) {
+  while (true) {
+    const newAccessToken = AppHelper.generateAccessToken()
+    const exists = await SchedulerAccessTokenManager.findOne({
+      token: newAccessToken,
+    }, transaction)
+    if (!exists) {
+      const accessTokenExpiryTime = Date.now() + Config.get('Settings:SchedulerTokenExpirationIntervalSeconds') * 99999
+      return {
+        token: newAccessToken,
+        expirationTime: accessTokenExpiryTime,
+      }
+    }
+  }
+}
+
+async function updateAccessToken(userId, newAccessToken, transaction) {
+  return SchedulerAccessTokenManager.updateOrCreate({
+    userId: userId,
+  }, {
+    userId: userId,
+    token: newAccessToken.token,
+    expirationTime: newAccessToken.expirationTime,
+  }, transaction)
+}
+
+module.exports = {
+  generateAccessToken,
+  updateAccessToken,
+}