From df219a383a31b435f317ba6472cc87929cc052c6 Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期六, 31 八月 2019 18:01:05 +0800
Subject: [PATCH] GetAllTaskByID add !tasksingle.Enable
---
util/util.go | 100 ++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 100 insertions(+), 0 deletions(-)
diff --git a/util/util.go b/util/util.go
index d802f4c..875e0f7 100644
--- a/util/util.go
+++ b/util/util.go
@@ -1,5 +1,28 @@
package util
+import (
+ "context"
+ "errors"
+ "fmt"
+
+ "basic.com/valib/deliver.git"
+
+ // "github.com/pierrec/lz4"
+ "taskpubsub/logger"
+)
+
+const SHMMaxTryCount = 3
+
+type SocketContext struct {
+ Sock deliver.Deliver
+ Context context.Context
+ Cancel context.CancelFunc
+
+ Mode int
+ URL string
+ UseSHM bool
+}
+
// 1. oldstring element is not in new : abandon(delete)
// 2. new element is not in oldstring : add(add)
func Difference(oldstring []string, newstring []string) map[string]string {
@@ -31,3 +54,80 @@
}
return diff
}
+
+// UnCompress uncompress
+func UnCompress(in []byte) ([]byte, error) {
+ return in, nil
+
+ // out := make([]byte, 3*len(in))
+ // n, err := lz4.UncompressBlock(in, out)
+ // if err != nil {
+ // logger.Error("uncompress error: ", err)
+ // return nil, err
+ // }
+ // out = out[:n] // uncompressed data
+ // return out, nil
+}
+
+// Compress compress
+func Compress(in []byte) ([]byte, error) {
+ return in, nil
+
+ // out := make([]byte, len(in))
+ // ht := make([]int, 64<<10) // buffer for the compression table
+ // n, err := lz4.CompressBlock(in, out, ht)
+ // if err != nil {
+ // logger.Error("compress: ", err)
+ // return nil, err
+ // }
+ // if n >= len(in) {
+ // logger.Error("image is not compressible")
+ // }
+ // out = out[:n] // compressed data
+ // return out, nil
+}
+
+// create server
+func NewSocketListen(mode int, url string, shm bool) (socket SocketContext, err error) {
+ logger.Info("url is: ", url)
+ ctx, cancel := context.WithCancel(context.Background())
+
+ socket.Context = ctx
+ socket.Cancel = cancel
+
+ socket.Mode = mode
+ socket.URL = url
+ socket.UseSHM = shm
+
+ if shm {
+ var err error
+ socket.Sock, err = deliver.NewServerWithError(deliver.Mode(mode), url)
+ if err != nil {
+ fmt.Println("TASK CREATE SERVER SHM ERROR: ", err)
+ }
+ } else {
+ socket.Sock = deliver.NewServer(deliver.Mode(mode), url)
+ }
+
+ if socket.Sock == nil {
+ return socket, errors.New("create listen error")
+ }
+
+ return socket, nil
+}
+
+func NewSocketDial(mode int, url string) (socket SocketContext, err error) {
+ logger.Info("url is: ", url)
+ ctx, cancel := context.WithCancel(context.Background())
+
+ socket.Context = ctx
+ socket.Cancel = cancel
+
+ socket.Sock = deliver.NewClient(deliver.Mode(mode), url)
+
+ if socket.Sock == nil {
+ return socket, errors.New("create listen error")
+ }
+
+ return socket, nil
+}
--
Gitblit v1.8.0