rediska

nzx_tmpmail/storage/redis.go

@@ -1 +1,80 @@
 package storage
+
+import (
+	"context"
+	"encoding/json"
+	"time"
+
+	"git.uwushka.cc/nzx056/nzx056/nzx_tmpmail/loggingshit"
+	"github.com/redis/go-redis/v9"
+)
+
+var rdb *redis.Client
+var ctx = context.Background()
+
+type Mail struct {
+	From    string
+	To      string
+	RawData string
+}
+
+type MailClient struct {
+	Name string
+	Mail []Mail
+}
+
+func GetMailClient(ctx context.Context, id string) (MailClient, error) {
+	var m MailClient
+	b, err := rdb.Get(ctx, "mailclient:"+id).Bytes()
+	if err != nil {
+		return m, err
+	}
+	err = json.Unmarshal(b, &m)
+
+	return m, err
+}
+
+func NewMail(ctx context.Context, rcp string, from string, rawdata string) error {
+	m, err := GetMailClient(ctx, rcp)
+	if err != nil {
+		m = MailClient{
+			Name: rcp,
+			Mail: []Mail{},
+		}
+	}
+
+	m.Mail = append(m.Mail, Mail{
+		From:    from,
+		To:      rcp,
+		RawData: rawdata,
+	})
+
+	b, err := json.Marshal(m)
+	if err != nil {
+		return err
+	}
+
+	if err := rdb.Set(ctx, "mailclient:"+rcp, b, time.Minute*10).Err(); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func RedisInit() {
+	rdb = redis.NewClient(&redis.Options{
+		Addr:     "localhost:6379",
+		Password: "",
+		DB:       0,
+	})
+
+	if err := rdb.Ping(ctx).Err(); err != nil {
+		loggingshit.Log("redis failed on rdb.Ping() with error %v", 1, err)
+	}
+
+	loggingshit.Log("redis init", 1)
+}
+
+func Client() *redis.Client {
+	return rdb
+}