first commit

This commit is contained in:
Sas Andy
2024-12-09 09:51:19 +07:00
commit ecc5dfd9c0
69 changed files with 5365 additions and 0 deletions

46
utils/error.log.go Normal file
View File

@@ -0,0 +1,46 @@
package utils
import (
"encoding/json"
"math/rand"
"net/http"
"time"
)
type LogError struct {
Code string `json:"code" db:"ErrorLogCode"`
TraceID string `json:"trace_id" db:"ErrorTraceID"`
Type string `json:"type" db:"ErrorLogType"`
Params string `json:"params" db:"ErrorLogParams"`
Query string `json:"query" db:"ErrorLogQuery"`
Message string `json:"message" db:"ErrorMessage"`
TimeStamp string `json:"timestamp" db:"ErrorLogTimestamp"`
}
func (e *LogError) Error() string {
return e.Message
}
func (e *LogError) LogError() LogError {
return *e
}
func RandomTraceID(n int) string {
src := rand.NewSource(time.Now().UnixNano())
r := rand.New(src)
const letterBytes = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
b := make([]byte, n)
for i := range b {
b[i] = letterBytes[r.Intn(len(letterBytes))]
}
return string(b)
}
func WriteErrorLog(w http.ResponseWriter, status int, err LogError) {
w.Header().Add("Content-Type", "application/json")
w.WriteHeader(status)
resp := map[string]any{"status": "ERR", "error": err}
json.NewEncoder(w).Encode(resp)
}

70
utils/utils.go Normal file
View File

@@ -0,0 +1,70 @@
package utils
import (
"encoding/json"
"fmt"
"net/http"
"reflect"
"strings"
"github.com/go-playground/validator/v10"
)
var Validate = validator.New()
func WriteJSON(w http.ResponseWriter, status int, v any) error {
w.Header().Add("Content-Type", "application/json")
w.WriteHeader(status)
resp := map[string]any{"status": "OK", "data": v}
return json.NewEncoder(w).Encode(resp)
}
func WriteJSONLogin(w http.ResponseWriter, status int, v any, t any, n any) error {
w.Header().Add("Content-Type", "application/json")
w.WriteHeader(status)
resp := map[string]any{"status": "OK", "user": v, "token": t, "type": n}
return json.NewEncoder(w).Encode(resp)
}
func WriteError(w http.ResponseWriter, status int, err error) {
w.Header().Add("Content-Type", "application/json")
w.WriteHeader(status)
resp := map[string]any{"status": "ERR", "message": err.Error()}
json.NewEncoder(w).Encode(resp)
}
func ParseJSON(r *http.Request, v any) error {
if r.Body == nil {
return fmt.Errorf("missing request body")
}
return json.NewDecoder(r.Body).Decode(v)
}
func GetTokenFromRequest(r *http.Request) string {
tokenAuth := r.Header.Get("Authorization")
if strings.HasPrefix(tokenAuth, "Bearer ") {
return strings.TrimPrefix(tokenAuth, "Bearer ")
}
tokenQuery := r.URL.Query().Get("token")
if tokenQuery != "" {
return tokenQuery
}
return ""
}
func MatchStruct(src interface{}, dst interface{}) error {
srcVal := reflect.ValueOf(src)
dstVal := reflect.ValueOf(dst).Elem()
for i := 0; i < srcVal.NumField(); i++ {
srcField := srcVal.Type().Field(i)
dstField := dstVal.FieldByName(srcField.Name)
if dstField.IsValid() && dstField.CanSet() && srcField.Type == dstField.Type() {
dstField.Set(srcVal.Field(i))
}
}
return nil
}