Add file

Author: WingGao

array.go

@@ -0,0 +1,27 @@
+package utils
+
+import (
+	"github.com/thoas/go-funk"
+	"reflect"
+)
+
+func GetItemString(arr []string, index int) (out string, ok bool) {
+	if len(arr) > index {
+		out = arr[index]
+		ok = true
+	}
+	return
+}
+
+//
+func ArrJoin(arrL interface{}, arrR interface{}, keyL, keyR string, joinFunc interface{}) interface{} {
+	rmap := ArrayToHashmap(arrR, keyR)
+	funcVal := reflect.ValueOf(joinFunc)
+	funk.ForEach(arrL, func(vl interface{}) {
+		keyV := funk.Get(vl, keyL)
+		if vr, ok := rmap.Get(keyV); ok {
+			funcVal.Call([]reflect.Value{reflect.ValueOf(vl), reflect.ValueOf(vr)})
+		}
+	})
+	return arrL
+}

config.go

@@ -0,0 +1,68 @@
+package utils
+
+import (
+	"io/ioutil"
+	"gopkg.in/yaml.v2"
+	"fmt"
+	tconfig "github.com/RichardKnop/machinery/v1/config"
+)
+
+//main config
+type MConfig struct {
+	Debug           bool
+	Addr            string //服务地址
+	Host            string
+	Mysql           string
+	DefaultPassword string `yaml:"default_password"` //默认密码
+	MediaPath       string
+	WebApps         string
+	MaxMediaSize    string `yaml:"max_media_size"`
+	CookieExpires   int64  `yaml:"cookie_expires"`
+	SiteCreator struct {
+		Mysql string
+		WpDir string
+	}
+	WxConfig WxConfig
+	Redis    RedisConf
+	Task     tconfig.Config
+
+	Cms struct {
+	}
+
+	LibreOfficePath  string `yaml:"libreoffice_path"`
+	TestAdminSession string
+	TestStudentSession string
+	TestTeacherSession string
+}
+
+type WxConfig struct {
+	AppId     string
+	MchId     string
+	ApiKey    string
+	NotifyUrl string
+}
+
+type RedisConf struct {
+	Addr        string
+	Password    string
+	Database    string
+	UniqueIdKey string
+}
+
+func LoadConfig(confPath string) (MConfig, error) {
+	conf := MConfig{
+		Debug: false,
+	}
+	if confPath == "" {
+
+	} else {
+		confd, _ := ioutil.ReadFile(confPath)
+		err := yaml.Unmarshal(confd, &conf)
+		if err != nil {
+			return conf, err
+		} else {
+			fmt.Println("MConfig loaded")
+		}
+	}
+	return conf, nil
+}

consts.go

@@ -0,0 +1,9 @@
+package utils
+
+const (
+	XSESSION_KEY = "xsession"
+)
+
+var (
+	DEBUG = true
+)

convert.go

@@ -0,0 +1,17 @@
+package utils
+
+import "github.com/thoas/go-funk"
+
+func ToInterfaceArrayString(t []string) []interface{} {
+	s := make([]interface{}, len(t))
+	for i, v := range t {
+		s[i] = v
+	}
+	return s
+}
+
+func ToInterfaceArray(m interface{}) []interface{} {
+	return funk.Map(m, func(x interface{}) interface{} {
+		return x
+	}).([]interface{})
+}

crypto.go

@@ -0,0 +1,10 @@
+package utils
+
+import (
+	"crypto/md5"
+	"fmt"
+)
+
+func Md5Sum(s string) string {
+	return fmt.Sprintf("%x", md5.Sum([]byte(s)))
+}

errors.go

@@ -0,0 +1,65 @@
+package utils
+
+import (
+	"github.com/go-errors/errors"
+	sll "github.com/emirpasic/gods/lists/singlylinkedlist"
+)
+
+var (
+	ERR_REQUIRE_LOGIN = errors.New("require login")
+	ERR_REQUIRE_ADMIN = errors.New("require admin")
+	ERR_NO_ACCOUNT    = errors.New("no such account")
+	ERR_PARAMS        = errors.New("params error")
+	ERR_NO_PERMISSION = errors.New("no permission")
+	ERR_CANNOT_MODIFY = errors.New("cannot modify")
+)
+
+func Nothing(...interface{}) {
+
+}
+
+// 获取第一个错误
+func FirstError(es ...error) error {
+	for _, e := range es {
+		if e != nil {
+			return e
+		}
+	}
+	return nil
+}
+
+type ErrorList struct {
+	List *sll.List
+}
+
+func NewErrorList() *ErrorList {
+	l := &ErrorList{}
+	l.List = sll.New()
+	return l
+}
+
+//只添加非nil的error
+func (l *ErrorList) AppendE(errs ...error) {
+	for _, v := range errs {
+		if v != nil {
+			//l.List.Add(errors.Wrap(v, 1))
+			l.List.Add(v)
+		}
+	}
+}
+
+func (l *ErrorList) FirstError() error {
+	_, err := l.List.Find(func(index int, value interface{}) bool {
+		return value != nil
+	})
+	if err == nil {
+		return nil
+	}
+	return err.(error)
+}
+
+func (l *ErrorList) Panic() {
+	if err := l.FirstError(); err != nil {
+		panic(err)
+	}
+}

handler.go

@@ -0,0 +1,49 @@
+package utils
+
+import (
+	"github.com/kataras/iris"
+	"github.com/kataras/iris/context"
+	"encoding/json"
+	"github.com/go-playground/form"
+	"github.com/json-iterator/go"
+)
+
+type ErrJson struct {
+	Err string `json:"err_msg"`
+}
+
+func AfterHandler(ictx context.Context, o interface{}, err error) {
+	if err != nil {
+		ictx.StatusCode(iris.StatusBadRequest)
+		ictx.JSON(ErrJson{Err: err.Error()})
+	} else {
+		var buf []byte
+		offj, isffj := o.(json.Marshaler)
+		if isffj {
+			buf, err = offj.MarshalJSON()
+		} else {
+			buf, err = jsoniter.Marshal(o)
+		}
+
+		if err != nil {
+			ictx.StatusCode(iris.StatusBadRequest)
+			ictx.JSON(ErrJson{Err: err.Error()})
+		} else {
+			ictx.StatusCode(iris.StatusOK)
+			ictx.ContentType("application/json")
+			ictx.Write(buf)
+		}
+	}
+	//ictx.Next()
+}
+
+//params必须是ptr
+func ParseForm(ictx context.Context, params interface{}) (err error) {
+	dec := form.NewDecoder()
+	err = dec.Decode(params, ictx.FormValues())
+	if err != nil {
+		ictx.Application().Logger().Warnf("Error when reading form: " + err.Error())
+		err = ERR_PARAMS
+	}
+	return
+}

math.go

@@ -0,0 +1,9 @@
+package utils
+
+import (
+	"math"
+)
+
+func MinInt(a, b int) int {
+	return int(math.Min(float64(a), float64(b)))
+}