通过AWS Lambda / API Gateway 架设负载均衡API服务器 (Load Balancer)

const fetch = require('node-fetch');
 
function reverse(promise) {
    return new Promise((resolve, reject) => Promise.resolve(promise).then(reject, resolve));
}
 
function promiseAny(iterable) {
    return reverse(Promise.all([...iterable].map(reverse)));
};

function shuffle(array) {
  let currentIndex = array.length, temporaryValue, randomIndex;
 
  // While there remain elements to shuffle...
  while (0 !== currentIndex) {
 
    // Pick a remaining element...
    randomIndex = Math.floor(Math.random() * currentIndex);
    currentIndex -= 1;
 
    // And swap it with the current element.
    temporaryValue = array[currentIndex];
    array[currentIndex] = array[randomIndex];
    array[randomIndex] = temporaryValue;
  }
 
  return array;
}

let nodes = shuffle([
  "https://api.steemit.com",
  "https://api.justyy.com",
  "https://api.steemitdev.com",
]);

async function getVersion(server) {
  return new Promise((resolve, reject) => {
    fetch(server, {
      method: "POST",
      headers: {
        'Content-Type': 'application/json'
      },
      body: JSON.stringify({"id":0,"jsonrpc":"2.0","method":"call","params":["login_api","get_version",[]]})
    }).then(response => {
      resolve(response.text());
    }).catch(function(error) {
      reject(error);
    });
  });
}

async function contactServer(server) {
  return new Promise((resolve, reject) => {
    fetch(server, {
      method: "GET"
    }).then(response => {
      if (!response.ok) {
        reject({
            "server": "", 
            "error": response, 
            "host": server
        });
      } else {
        resolve({
          "server": server,
        });
      }
    }).catch(function(error) {
      reject({
          "server": "", 
          "error": error, 
          "host": server
      });
    });
  });
}

async function forwardRequestGET(apiURL) {
  return new Promise((resolve, reject) => {
    fetch(apiURL, {
      method: "GET",   
      headers: {
        'Content-Type': 'application/json'
      },
      redirect: "follow"
    }).then(response => {
      resolve(response.text());
    }).catch(function(error) {
      reject(error);
    });
  });
}

async function forwardRequestPOST(apiURL, body) {
  return new Promise((resolve, reject) => {
    fetch(apiURL, {
      method: "POST",   
      redirect: "follow",
      headers: {
      'Content-Type': 'application/json'
      },      
      body: body
    }).then(response => {
      resolve(response.text());
    }).catch(function(error) {
      reject(error);
    });
  });
}

exports.handler = async(event, context) => {
  const servers = [];
  nodes = shuffle(nodes);
  for (const server of nodes) {
    servers.push(contactServer(server));
  }
  const load = await promiseAny(servers); 
  const forwardedURL = load['server'];
  let method = "POST";
  if (event["http-method"]) {
      method = event["http-method"].toUpperCase();
  }
  let result;
  let res;
  let version = "";
  try {
    version = await getVersion(load['server']);
  } catch (e) {
    version = JSON.stringify(e);
  }
  console.log("server = ", load['server']);
  console.log("version = ", version);
  let body = {};
  if (event["body-json"]) {
    body = event["body-json"];
    body = JSON.stringify(body);
    if (body.length <= 2) {
      method = "GET";
    }
  } else {
    method = "GET";
  }
  try {    
    if (method === "POST") {
        result = await forwardRequestPOST(forwardedURL, body);
    } else if (method === "GET") {
        result = await forwardRequestGET(forwardedURL);
    } else {
        return {
            statusCode: 405,
            statusText: 'Method Not Allowed',
            headers: {
                'Access-Control-Allow-Origin': '*',
                'Cache-Control': 'max-age=3600',
            }
        };
    }
  } catch (e) {
    return {
        statusCode: 500,
        headers: {
            'Content-Type': 'application/json',
            'Access-Control-Allow-Origin': '*',
            'Cache-Control': 'max-age=3',
            'Version': version,
            'Error': JSON.stringify(e)
        }
    };
  }
  // Adding meta information before forwarding to requester.
  result = JSON.parse(result);
  try {
    result["__version__"] = JSON.parse(version);
  } catch (e) {
    result["__version__"] = version;
  }
  result["__server__"] = load['server'];
  return result;
};

##  See http://docs.aws.amazon.com/apigateway/latest/developerguide/api-gateway-mapping-template-reference.html
##  This template will pass through all parameters including path, querystring, header, stage variables, and context through to the integration endpoint via the body/payload
#set($allParams = $input.params())
{
"body-json" : $input.json('$'),
"params" : {
#foreach($type in $allParams.keySet())
    #set($params = $allParams.get($type))
"$type" : {
    #foreach($paramName in $params.keySet())
    "$paramName" : "$util.escapeJavaScript($params.get($paramName))"
        #if($foreach.hasNext),#end
    #end
}
    #if($foreach.hasNext),#end
#end
},
"stage-variables" : {
#foreach($key in $stageVariables.keySet())
"$key" : "$util.escapeJavaScript($stageVariables.get($key))"
    #if($foreach.hasNext),#end
#end
},
"context" : {
    "account-id" : "$context.identity.accountId",
    "api-id" : "$context.apiId",
    "api-key" : "$context.identity.apiKey",
    "authorizer-principal-id" : "$context.authorizer.principalId",
    "caller" : "$context.identity.caller",
    "cognito-authentication-provider" : "$context.identity.cognitoAuthenticationProvider",
    "cognito-authentication-type" : "$context.identity.cognitoAuthenticationType",
    "cognito-identity-id" : "$context.identity.cognitoIdentityId",
    "cognito-identity-pool-id" : "$context.identity.cognitoIdentityPoolId",
    "http-method" : "$context.httpMethod",
    "stage" : "$context.stage",
    "source-ip" : "$context.identity.sourceIp",
    "user" : "$context.identity.user",
    "user-agent" : "$context.identity.userAgent",
    "user-arn" : "$context.identity.userArn",
    "request-id" : "$context.requestId",
    "resource-id" : "$context.resourceId",
    "resource-path" : "$context.resourcePath"
    }
}