Move packages to proejct/pkg to follow go project folder structure convention (#1190)

pkg/redis/redis.go

@@ -0,0 +1,126 @@
+/*
+Copyright 2018 The Fission Authors.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package redis
+
+import (
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"net/url"
+	"os"
+	"strings"
+
+	"github.com/golang/protobuf/proto"
+	"github.com/gomodule/redigo/redis"
+	"github.com/pkg/errors"
+	"go.uber.org/zap"
+
+	"github.com/fission/fission/pkg/redis/build/gen"
+)
+
+func NewClient() (redis.Conn, error) {
+	redisIP := os.Getenv("REDIS_SERVICE_HOST") // TODO: Do this here or somewhere earlier?
+	redisPort := os.Getenv("REDIS_SERVICE_PORT")
+
+	if len(redisIP) == 0 || len(redisPort) == 0 {
+		return nil, errors.New("redis host or port not supplied")
+	}
+
+	redisURL := fmt.Sprintf("%s:%s", redisIP, redisPort)
+
+	c, err := redis.Dial("tcp", redisURL)
+	if err != nil {
+		return nil, errors.Wrapf(err, "could not connect to Redis at url %q", redisURL)
+	}
+	return c, nil
+}
+
+func Record(logger *zap.Logger, triggerName string, recorderName string, reqUID string, request *http.Request, originalUrl url.URL, payload string, response *http.Response, namespace string, timestamp int64) {
+	// Case where the function should not have been recorded
+	if len(reqUID) == 0 {
+		return
+	}
+
+	fullPath := originalUrl.String()
+	escPayload := string(json.RawMessage(payload))
+
+	client, err := NewClient()
+	if err != nil {
+		logger.Error("could not create redis client", zap.Error(err))
+		return
+	}
+
+	url := make(map[string]string)
+	url["Host"] = request.URL.Host
+	url["Path"] = fullPath
+	url["Payload"] = escPayload
+
+	header := make(map[string]string)
+	for key, value := range request.Header {
+		header[key] = strings.Join(value, ",")
+	}
+
+	form := make(map[string]string)
+	for key, value := range request.Form {
+		form[key] = strings.Join(value, ",")
+	}
+
+	postForm := make(map[string]string)
+	for key, value := range request.PostForm {
+		postForm[key] = strings.Join(value, ",")
+	}
+
+	req := &redisCache.Request{
+		Method:   request.Method,
+		URL:      url,
+		Header:   header,
+		Host:     request.Host, // Proxied host?
+		Form:     form,
+		PostForm: postForm,
+	}
+
+	logger.Info("storing request", zap.Any("request", req))
+
+	resp := &redisCache.Response{
+		Status:     response.Status,
+		StatusCode: int32(response.StatusCode),
+	}
+
+	ureq := &redisCache.UniqueRequest{
+		Req:     req,
+		Resp:    resp,
+		Trigger: triggerName,
+	}
+
+	data, err := proto.Marshal(ureq)
+	if err != nil {
+		logger.Error("error marshalling request", zap.Error(err))
+		return
+	}
+
+	_, err = client.Do("HMSET", reqUID, "ReqResponse", data, "Timestamp", timestamp, "Trigger", triggerName)
+	if err != nil {
+		logger.Error("error saving request", zap.Error(err))
+		return
+	}
+
+	_, err = client.Do("LPUSH", recorderName, reqUID)
+	if err != nil {
+		logger.Error("error saving recorder-request pair", zap.Error(err))
+		return
+	}
+}