91 lines
2.1 KiB
Go
91 lines
2.1 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
|
|
"git.cems.dev/cdricms/bdooc/proto"
|
|
"github.com/wailsapp/wails/v2/pkg/runtime"
|
|
)
|
|
|
|
// App struct
|
|
type App struct {
|
|
ctx context.Context
|
|
employees *proto.EmployeeList // Caching purposes
|
|
}
|
|
|
|
// NewApp creates a new App application struct
|
|
func NewApp() *App {
|
|
return &App{}
|
|
}
|
|
|
|
// startup is called when the app starts. The context is saved
|
|
// so we can call the runtime methods
|
|
func (a *App) startup(ctx context.Context) {
|
|
a.ctx = ctx
|
|
}
|
|
|
|
func (a *App) GetEmployees(path string, limit, page uint, column, query *string) ([]*proto.Employee, error) {
|
|
if a.employees == nil {
|
|
e, err := proto.LoadFromFile(path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
a.employees = e
|
|
}
|
|
|
|
if column != nil && query != nil {
|
|
e, err := a.employees.QueryEmployeesByColumn(proto.EmployeeField(proto.SnakeCaseToPascalCase(*column)), *query)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if uint(len(e)) < page*limit+limit {
|
|
return e[page*limit:], nil
|
|
}
|
|
return e[page*limit : page*limit+limit], nil
|
|
}
|
|
|
|
if uint(len(a.employees.Employees)) < page*limit+limit {
|
|
return a.employees.Employees[page*limit:], nil
|
|
}
|
|
return a.employees.Employees[page*limit : page*limit+limit], nil
|
|
}
|
|
|
|
func (a *App) GetProtoPath() string {
|
|
return a.OpenPath([]runtime.FileFilter{{
|
|
DisplayName: "Proto binary file",
|
|
Pattern: "*.bin",
|
|
}})
|
|
}
|
|
|
|
func (a *App) OpenPath(filters []runtime.FileFilter) string {
|
|
path, err := runtime.OpenFileDialog(a.ctx, runtime.OpenDialogOptions{
|
|
Filters: filters,
|
|
})
|
|
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return path
|
|
}
|
|
|
|
type FunctionAgg string
|
|
|
|
const (
|
|
AVG = FunctionAgg("AVG")
|
|
SUM = FunctionAgg("SUM")
|
|
)
|
|
|
|
func (a *App) AggregateByColumn(groupBy proto.EmployeeField, on proto.EmployeeField, function FunctionAgg) map[string]float64 {
|
|
return proto.AggregateByColumn(a.employees, proto.EmployeeField(proto.SnakeCaseToPascalCase(string(groupBy))), func(ep proto.EmployeePList) float64 {
|
|
switch function {
|
|
case AVG:
|
|
return proto.Average(ep, proto.EmployeeField(proto.SnakeCaseToPascalCase(string(on))))
|
|
case SUM:
|
|
return proto.Sum(ep, proto.EmployeeField(proto.SnakeCaseToPascalCase(string(on))))
|
|
}
|
|
|
|
return 0.0
|
|
})
|
|
}
|