Resolve "Websocket utils"

Related to #40 (closed)

handler_utils/api.go

@@ -43,6 +43,28 @@ func ValidateAPIEndpoints(endpoints map[string]map[string]interface{}) (map[stri
 	return endpoints, nil
 }
 
+// ValidateWebsocketEndpoints checks that all websocket endpoints are correctly defined using one of the supported
+// handler types and returns updated endpoints with additional information
+func ValidateWebsocketEndpoints(endpoints map[string]interface{}) (map[string]interface{}, error) {
+	for websocketAction, actionFunc := range endpoints {
+		if websocketAction == "" {
+			return nil, errors.Errorf("blank action")
+		}
+		if actionFunc == nil {
+			return nil, errors.Errorf("nil handler on %s %s", websocketAction, actionFunc)
+		}
+
+		handler, err := NewHandler(actionFunc)
+		if err != nil {
+			return nil, errors.Wrapf(err, "%s has invalid handler %T", websocketAction, actionFunc)
+		}
+		// replace the endpoint value so that we can quickly call this handler
+		endpoints[websocketAction] = handler
+
+	}
+	return endpoints, nil
+}
+
 func ValidateRequestParams(request *events.APIGatewayProxyRequest, paramsStructType reflect.Type) (reflect.Value, error) {
 	paramValues := map[string]interface{}{}
 	for n, v := range request.QueryStringParameters {