Compare commits
5 Commits
Author | SHA1 | Date | |
---|---|---|---|
c28bc086b2
|
|||
d44e971325
|
|||
fe4cdc48af
|
|||
631006a4e1
|
|||
567ead8697
|
254
dataext/casMutex.go
Normal file
254
dataext/casMutex.go
Normal file
@@ -0,0 +1,254 @@
|
||||
package dataext
|
||||
|
||||
import (
|
||||
"context"
|
||||
"golang.org/x/sync/semaphore"
|
||||
"runtime"
|
||||
"sync"
|
||||
"sync/atomic"
|
||||
"time"
|
||||
"unsafe"
|
||||
)
|
||||
|
||||
// from https://github.com/viney-shih/go-lock/blob/2f19fd8ce335e33e0ab9dccb1ff2ce820c3da332/cas.go
|
||||
|
||||
// CASMutex is the struct implementing RWMutex with CAS mechanism.
|
||||
type CASMutex struct {
|
||||
state casState
|
||||
turnstile *semaphore.Weighted
|
||||
|
||||
broadcastChan chan struct{}
|
||||
broadcastMut sync.RWMutex
|
||||
}
|
||||
|
||||
func NewCASMutex() *CASMutex {
|
||||
return &CASMutex{
|
||||
state: casStateNoLock,
|
||||
turnstile: semaphore.NewWeighted(1),
|
||||
broadcastChan: make(chan struct{}),
|
||||
}
|
||||
}
|
||||
|
||||
type casState int32
|
||||
|
||||
const (
|
||||
casStateUndefined casState = iota - 2 // -2
|
||||
casStateWriteLock // -1
|
||||
casStateNoLock // 0
|
||||
casStateReadLock // >= 1
|
||||
)
|
||||
|
||||
func (m *CASMutex) getState(n int32) casState {
|
||||
switch st := casState(n); {
|
||||
case st == casStateWriteLock:
|
||||
fallthrough
|
||||
case st == casStateNoLock:
|
||||
return st
|
||||
case st >= casStateReadLock:
|
||||
return casStateReadLock
|
||||
default:
|
||||
// actually, it should not happened.
|
||||
return casStateUndefined
|
||||
}
|
||||
}
|
||||
|
||||
func (m *CASMutex) listen() <-chan struct{} {
|
||||
m.broadcastMut.RLock()
|
||||
defer m.broadcastMut.RUnlock()
|
||||
|
||||
return m.broadcastChan
|
||||
}
|
||||
|
||||
func (m *CASMutex) broadcast() {
|
||||
newCh := make(chan struct{})
|
||||
|
||||
m.broadcastMut.Lock()
|
||||
ch := m.broadcastChan
|
||||
m.broadcastChan = newCh
|
||||
m.broadcastMut.Unlock()
|
||||
|
||||
close(ch)
|
||||
}
|
||||
|
||||
func (m *CASMutex) tryLock(ctx context.Context) bool {
|
||||
for {
|
||||
broker := m.listen()
|
||||
if atomic.CompareAndSwapInt32(
|
||||
(*int32)(unsafe.Pointer(&m.state)),
|
||||
int32(casStateNoLock),
|
||||
int32(casStateWriteLock),
|
||||
) {
|
||||
return true
|
||||
}
|
||||
|
||||
if ctx == nil {
|
||||
return false
|
||||
}
|
||||
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
// timeout or cancellation
|
||||
return false
|
||||
case <-broker:
|
||||
// waiting for signal triggered by m.broadcast() and trying again.
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// TryLockWithContext attempts to acquire the lock, blocking until resources
|
||||
// are available or ctx is done (timeout or cancellation).
|
||||
func (m *CASMutex) TryLockWithContext(ctx context.Context) bool {
|
||||
if err := m.turnstile.Acquire(ctx, 1); err != nil {
|
||||
// Acquire failed due to timeout or cancellation
|
||||
return false
|
||||
}
|
||||
|
||||
defer m.turnstile.Release(1)
|
||||
|
||||
return m.tryLock(ctx)
|
||||
}
|
||||
|
||||
// Lock acquires the lock.
|
||||
// If it is currently held by others, Lock will wait until it has a chance to acquire it.
|
||||
func (m *CASMutex) Lock() {
|
||||
ctx := context.Background()
|
||||
|
||||
m.TryLockWithContext(ctx)
|
||||
}
|
||||
|
||||
// TryLock attempts to acquire the lock without blocking.
|
||||
// Return false if someone is holding it now.
|
||||
func (m *CASMutex) TryLock() bool {
|
||||
if !m.turnstile.TryAcquire(1) {
|
||||
return false
|
||||
}
|
||||
|
||||
defer m.turnstile.Release(1)
|
||||
|
||||
return m.tryLock(nil)
|
||||
}
|
||||
|
||||
// TryLockWithTimeout attempts to acquire the lock within a period of time.
|
||||
// Return false if spending time is more than duration and no chance to acquire it.
|
||||
func (m *CASMutex) TryLockWithTimeout(duration time.Duration) bool {
|
||||
ctx, cancel := context.WithTimeout(context.Background(), duration)
|
||||
defer cancel()
|
||||
|
||||
return m.TryLockWithContext(ctx)
|
||||
}
|
||||
|
||||
// Unlock releases the lock.
|
||||
func (m *CASMutex) Unlock() {
|
||||
if ok := atomic.CompareAndSwapInt32(
|
||||
(*int32)(unsafe.Pointer(&m.state)),
|
||||
int32(casStateWriteLock),
|
||||
int32(casStateNoLock),
|
||||
); !ok {
|
||||
panic("Unlock failed")
|
||||
}
|
||||
|
||||
m.broadcast()
|
||||
}
|
||||
|
||||
func (m *CASMutex) rTryLock(ctx context.Context) bool {
|
||||
for {
|
||||
broker := m.listen()
|
||||
n := atomic.LoadInt32((*int32)(unsafe.Pointer(&m.state)))
|
||||
st := m.getState(n)
|
||||
switch st {
|
||||
case casStateNoLock, casStateReadLock:
|
||||
if atomic.CompareAndSwapInt32((*int32)(unsafe.Pointer(&m.state)), n, n+1) {
|
||||
return true
|
||||
}
|
||||
}
|
||||
|
||||
if ctx == nil {
|
||||
return false
|
||||
}
|
||||
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
// timeout or cancellation
|
||||
return false
|
||||
default:
|
||||
switch st {
|
||||
// read-lock failed due to concurrence issue, try again immediately
|
||||
case casStateNoLock, casStateReadLock:
|
||||
runtime.Gosched() // allow other goroutines to do stuff.
|
||||
continue
|
||||
}
|
||||
}
|
||||
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
// timeout or cancellation
|
||||
return false
|
||||
case <-broker:
|
||||
// waiting for signal triggered by m.broadcast() and trying again.
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// RTryLockWithContext attempts to acquire the read lock, blocking until resources
|
||||
// are available or ctx is done (timeout or cancellation).
|
||||
func (m *CASMutex) RTryLockWithContext(ctx context.Context) bool {
|
||||
if err := m.turnstile.Acquire(ctx, 1); err != nil {
|
||||
// Acquire failed due to timeout or cancellation
|
||||
return false
|
||||
}
|
||||
|
||||
m.turnstile.Release(1)
|
||||
|
||||
return m.rTryLock(ctx)
|
||||
}
|
||||
|
||||
// RLock acquires the read lock.
|
||||
// If it is currently held by others writing, RLock will wait until it has a chance to acquire it.
|
||||
func (m *CASMutex) RLock() {
|
||||
ctx := context.Background()
|
||||
|
||||
m.RTryLockWithContext(ctx)
|
||||
}
|
||||
|
||||
// RTryLock attempts to acquire the read lock without blocking.
|
||||
// Return false if someone is writing it now.
|
||||
func (m *CASMutex) RTryLock() bool {
|
||||
if !m.turnstile.TryAcquire(1) {
|
||||
return false
|
||||
}
|
||||
|
||||
m.turnstile.Release(1)
|
||||
|
||||
return m.rTryLock(nil)
|
||||
}
|
||||
|
||||
// RTryLockWithTimeout attempts to acquire the read lock within a period of time.
|
||||
// Return false if spending time is more than duration and no chance to acquire it.
|
||||
func (m *CASMutex) RTryLockWithTimeout(duration time.Duration) bool {
|
||||
ctx, cancel := context.WithTimeout(context.Background(), duration)
|
||||
defer cancel()
|
||||
|
||||
return m.RTryLockWithContext(ctx)
|
||||
}
|
||||
|
||||
// RUnlock releases the read lock.
|
||||
func (m *CASMutex) RUnlock() {
|
||||
n := atomic.AddInt32((*int32)(unsafe.Pointer(&m.state)), -1)
|
||||
switch m.getState(n) {
|
||||
case casStateUndefined, casStateWriteLock:
|
||||
panic("RUnlock failed")
|
||||
case casStateNoLock:
|
||||
m.broadcast()
|
||||
}
|
||||
}
|
||||
|
||||
// RLocker returns a Locker interface that implements the Lock and Unlock methods
|
||||
// by calling CASMutex.RLock and CASMutex.RUnlock.
|
||||
func (m *CASMutex) RLocker() sync.Locker {
|
||||
return (*rlocker)(m)
|
||||
}
|
||||
|
||||
type rlocker CASMutex
|
||||
|
||||
func (r *rlocker) Lock() { (*CASMutex)(r).RLock() }
|
||||
func (r *rlocker) Unlock() { (*CASMutex)(r).RUnlock() }
|
@@ -35,6 +35,23 @@ func (s *SyncMap[TKey, TData]) SetIfNotContains(key TKey, data TData) bool {
|
||||
return true
|
||||
}
|
||||
|
||||
func (s *SyncMap[TKey, TData]) SetIfNotContainsFunc(key TKey, data func() TData) bool {
|
||||
s.lock.Lock()
|
||||
defer s.lock.Unlock()
|
||||
|
||||
if s.data == nil {
|
||||
s.data = make(map[TKey]TData)
|
||||
}
|
||||
|
||||
if _, existsInPreState := s.data[key]; existsInPreState {
|
||||
return false
|
||||
}
|
||||
|
||||
s.data[key] = data()
|
||||
|
||||
return true
|
||||
}
|
||||
|
||||
func (s *SyncMap[TKey, TData]) Get(key TKey) (TData, bool) {
|
||||
s.lock.Lock()
|
||||
defer s.lock.Unlock()
|
||||
@@ -50,6 +67,39 @@ func (s *SyncMap[TKey, TData]) Get(key TKey) (TData, bool) {
|
||||
}
|
||||
}
|
||||
|
||||
func (s *SyncMap[TKey, TData]) GetAndSetIfNotContains(key TKey, data TData) TData {
|
||||
s.lock.Lock()
|
||||
defer s.lock.Unlock()
|
||||
|
||||
if s.data == nil {
|
||||
s.data = make(map[TKey]TData)
|
||||
}
|
||||
|
||||
if v, ok := s.data[key]; ok {
|
||||
return v
|
||||
} else {
|
||||
s.data[key] = data
|
||||
return data
|
||||
}
|
||||
}
|
||||
|
||||
func (s *SyncMap[TKey, TData]) GetAndSetIfNotContainsFunc(key TKey, data func() TData) TData {
|
||||
s.lock.Lock()
|
||||
defer s.lock.Unlock()
|
||||
|
||||
if s.data == nil {
|
||||
s.data = make(map[TKey]TData)
|
||||
}
|
||||
|
||||
if v, ok := s.data[key]; ok {
|
||||
return v
|
||||
} else {
|
||||
dataObj := data()
|
||||
s.data[key] = dataObj
|
||||
return dataObj
|
||||
}
|
||||
}
|
||||
|
||||
func (s *SyncMap[TKey, TData]) Delete(key TKey) bool {
|
||||
s.lock.Lock()
|
||||
defer s.lock.Unlock()
|
||||
|
22
go.mod
22
go.mod
@@ -3,15 +3,20 @@ module gogs.mikescher.com/BlackForestBytes/goext
|
||||
go 1.22
|
||||
|
||||
require (
|
||||
github.com/gin-gonic/gin v1.9.1
|
||||
github.com/gin-gonic/gin v1.10.0
|
||||
github.com/glebarez/go-sqlite v1.22.0 // only needed for tests -.-
|
||||
github.com/jmoiron/sqlx v1.4.0
|
||||
github.com/rs/xid v1.5.0
|
||||
github.com/rs/zerolog v1.32.0
|
||||
go.mongodb.org/mongo-driver v1.15.0
|
||||
golang.org/x/crypto v0.22.0
|
||||
golang.org/x/sys v0.19.0
|
||||
golang.org/x/term v0.19.0
|
||||
golang.org/x/crypto v0.23.0
|
||||
golang.org/x/sys v0.20.0
|
||||
golang.org/x/term v0.20.0
|
||||
)
|
||||
|
||||
require (
|
||||
github.com/jung-kurt/gofpdf v1.16.2
|
||||
golang.org/x/sync v0.7.0
|
||||
)
|
||||
|
||||
require (
|
||||
@@ -47,11 +52,10 @@ require (
|
||||
github.com/xdg-go/scram v1.1.2 // indirect
|
||||
github.com/xdg-go/stringprep v1.0.4 // indirect
|
||||
github.com/youmark/pkcs8 v0.0.0-20240424034433-3c2c7870ae76 // indirect
|
||||
golang.org/x/arch v0.7.0 // indirect
|
||||
golang.org/x/net v0.24.0 // indirect
|
||||
golang.org/x/sync v0.7.0 // indirect
|
||||
golang.org/x/text v0.14.0 // indirect
|
||||
google.golang.org/protobuf v1.34.0 // indirect
|
||||
golang.org/x/arch v0.8.0 // indirect
|
||||
golang.org/x/net v0.25.0 // indirect
|
||||
golang.org/x/text v0.15.0 // indirect
|
||||
google.golang.org/protobuf v1.34.1 // indirect
|
||||
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||
modernc.org/libc v1.37.6 // indirect
|
||||
modernc.org/mathutil v1.6.0 // indirect
|
||||
|
25
go.sum
25
go.sum
@@ -1,4 +1,5 @@
|
||||
filippo.io/edwards25519 v1.1.0/go.mod h1:BxyFTGdWcka3PhytdK4V28tE5sGfRvvvRV7EaN4VDT4=
|
||||
github.com/boombuler/barcode v1.0.0/go.mod h1:paBWMcWSl3LHKBqUq+rly7CNSldXjb2rDl3JlRe0mD8=
|
||||
github.com/bytedance/sonic v1.5.0/go.mod h1:ED5hyg4y6t3/9Ku1R6dU/4KyJ48DZ4jPhfY1O2AihPM=
|
||||
github.com/bytedance/sonic v1.10.0-rc/go.mod h1:ElCzW+ufi8qKqNW0FY314xriJhyJhuoJ3gFZdAHF7NM=
|
||||
github.com/bytedance/sonic v1.10.2 h1:GQebETVBxYB7JGWJtLBi07OVzWwt+8dWA00gEVW2ZFE=
|
||||
@@ -50,6 +51,8 @@ github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE
|
||||
github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI=
|
||||
github.com/gin-gonic/gin v1.9.1 h1:4idEAncQnU5cB7BeOkPtxjfCSye0AAm1R0RVIqJ+Jmg=
|
||||
github.com/gin-gonic/gin v1.9.1/go.mod h1:hPrL7YrpYKXt5YId3A/Tnip5kqbEAP+KLuI3SUcPTeU=
|
||||
github.com/gin-gonic/gin v1.10.0 h1:nTuyha1TYqgedzytsKYqna+DfLos46nTv2ygFy86HFU=
|
||||
github.com/gin-gonic/gin v1.10.0/go.mod h1:4PMNQiOhvDRa013RKVbsiNwoyezlm2rm0uX/T7kzp5Y=
|
||||
github.com/glebarez/go-sqlite v1.22.0 h1:uAcMJhaA6r3LHMTFgP0SifzgXg46yJkgxqyuyec+ruQ=
|
||||
github.com/glebarez/go-sqlite v1.22.0/go.mod h1:PlBIdHe0+aUEFn+r2/uthrWq4FxbzugL0L8Li6yQJbc=
|
||||
github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s=
|
||||
@@ -93,6 +96,9 @@ github.com/jmoiron/sqlx v1.4.0 h1:1PLqN7S1UYp5t4SrVVnt4nUVNemrDAtxlulVe+Qgm3o=
|
||||
github.com/jmoiron/sqlx v1.4.0/go.mod h1:ZrZ7UsYB/weZdl2Bxg6jCRO9c3YHl8r3ahlKmRT4JLY=
|
||||
github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=
|
||||
github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=
|
||||
github.com/jung-kurt/gofpdf v1.0.0/go.mod h1:7Id9E/uU8ce6rXgefFLlgrJj/GYY22cpxn+r32jIOes=
|
||||
github.com/jung-kurt/gofpdf v1.16.2 h1:jgbatWHfRlPYiK85qgevsZTHviWXKwB1TTiKdz5PtRc=
|
||||
github.com/jung-kurt/gofpdf v1.16.2/go.mod h1:1hl7y57EsiPAkLbOwzpzqgx1A30nQCk/YmFV8S2vmK0=
|
||||
github.com/klauspost/compress v1.13.6/go.mod h1:/3/Vjq9QcHkK5uEr5lBEmyoZ1iFhe47etQ6QUkpK6sk=
|
||||
github.com/klauspost/compress v1.17.4 h1:Ej5ixsIri7BrIjBkRZLTo6ghwrEtHFk7ijlczPW4fZ4=
|
||||
github.com/klauspost/compress v1.17.4/go.mod h1:/dCuZOvVtNoHsyb+cuJD3itjs3NbnF6KH9zAO4BDxPM=
|
||||
@@ -141,6 +147,8 @@ github.com/pelletier/go-toml/v2 v2.2.1 h1:9TA9+T8+8CUCO2+WYnDLCgrYi9+omqKXyjDtos
|
||||
github.com/pelletier/go-toml/v2 v2.2.1/go.mod h1:1t835xjRzz80PqgE6HHgN2JOsmgYu/h4qDAS4n929Rs=
|
||||
github.com/pelletier/go-toml/v2 v2.2.2 h1:aYUidT7k73Pcl9nb2gScu7NSrKCSHIDE89b3+6Wq+LM=
|
||||
github.com/pelletier/go-toml/v2 v2.2.2/go.mod h1:1t835xjRzz80PqgE6HHgN2JOsmgYu/h4qDAS4n929Rs=
|
||||
github.com/phpdave11/gofpdi v1.0.7/go.mod h1:vBmVV0Do6hSBHC8uKUQ71JGW+ZGQq74llk/7bXwjDoI=
|
||||
github.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
||||
github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||
@@ -152,10 +160,12 @@ github.com/rs/zerolog v1.31.0 h1:FcTR3NnLWW+NnTwwhFWiJSZr4ECLpqCm6QsEnyvbV4A=
|
||||
github.com/rs/zerolog v1.31.0/go.mod h1:/7mN4D5sKwJLZQ2b/znpjC3/GQWY/xaDXUM0kKWRHss=
|
||||
github.com/rs/zerolog v1.32.0 h1:keLypqrlIjaFsbmJOBdB/qvyF8KEtCWHwobLp5l/mQ0=
|
||||
github.com/rs/zerolog v1.32.0/go.mod h1:/7mN4D5sKwJLZQ2b/znpjC3/GQWY/xaDXUM0kKWRHss=
|
||||
github.com/ruudk/golang-pdf417 v0.0.0-20181029194003-1af4ab5afa58/go.mod h1:6lfFZQK844Gfx8o5WFuvpxWRwnSoipWe/p622j1v06w=
|
||||
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||
github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
|
||||
github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
|
||||
github.com/stretchr/objx v0.5.2/go.mod h1:FRsXN1f5AsAjCGJKqEizvkpNtU+EGNCLh3NxZ/8L+MA=
|
||||
github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
|
||||
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
|
||||
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||
github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||
@@ -191,6 +201,8 @@ go.mongodb.org/mongo-driver v1.15.0/go.mod h1:Vzb0Mk/pa7e6cWw85R4F/endUC3u0U9jGc
|
||||
golang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8=
|
||||
golang.org/x/arch v0.7.0 h1:pskyeJh/3AmoQ8CPE95vxHLqp1G1GfGNXTmcl9NEKTc=
|
||||
golang.org/x/arch v0.7.0/go.mod h1:FEVrYAQjsQXMVJ1nsMoVVXPZg6p2JE2mx8psSWTDQys=
|
||||
golang.org/x/arch v0.8.0 h1:3wRIsP3pM4yUptoR96otTUOXI367OS0+c9eeRi9doIc=
|
||||
golang.org/x/arch v0.8.0/go.mod h1:FEVrYAQjsQXMVJ1nsMoVVXPZg6p2JE2mx8psSWTDQys=
|
||||
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
||||
golang.org/x/crypto v0.0.0-20200302210943-78000ba7a073/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
|
||||
golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
|
||||
@@ -207,6 +219,9 @@ golang.org/x/crypto v0.21.0 h1:X31++rzVUdKhX5sWmSOFZxx8UW/ldWx55cbf08iNAMA=
|
||||
golang.org/x/crypto v0.21.0/go.mod h1:0BP7YvVV9gBbVKyeTG0Gyn+gZm94bibOW5BjDEYAOMs=
|
||||
golang.org/x/crypto v0.22.0 h1:g1v0xeRhjcugydODzvb3mEM9SQ0HGp9s/nh3COQ/C30=
|
||||
golang.org/x/crypto v0.22.0/go.mod h1:vr6Su+7cTlO45qkww3VDJlzDn0ctJvRgYbC2NvXHt+M=
|
||||
golang.org/x/crypto v0.23.0 h1:dIJU/v2J8Mdglj/8rJ6UUOM3Zc9zLZxVZwwxMooUSAI=
|
||||
golang.org/x/crypto v0.23.0/go.mod h1:CKFgDieR+mRhux2Lsu27y0fO304Db0wZe70UKqHu0v8=
|
||||
golang.org/x/image v0.0.0-20190910094157-69e4b8554b2a/go.mod h1:FeLwcggjj3mMvU+oOTbSwawSJRM1uh48EjtB4UJZlP0=
|
||||
golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
|
||||
golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
|
||||
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
|
||||
@@ -223,6 +238,8 @@ golang.org/x/net v0.22.0 h1:9sGLhx7iRIHEiX0oAJ3MRZMUCElJgy7Br1nO+AMN3Tc=
|
||||
golang.org/x/net v0.22.0/go.mod h1:JKghWKKOSdJwpW2GEx0Ja7fmaKnMsbu+MWVZTokSYmg=
|
||||
golang.org/x/net v0.24.0 h1:1PcaxkF854Fu3+lvBIx5SYn9wRlBzzcnHZSiaFFAb0w=
|
||||
golang.org/x/net v0.24.0/go.mod h1:2Q7sJY5mzlzWjKtYUEXSlBWCdyaioyXzRB2RtU8KVE8=
|
||||
golang.org/x/net v0.25.0 h1:d/OCCoBEUq33pjydKrGQhw7IlUPI2Oylr+8qLx49kac=
|
||||
golang.org/x/net v0.25.0/go.mod h1:JkAGAh7GEvH74S6FOH42FLoXpXbE/aqXSrIQjXgsiwM=
|
||||
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||
golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||
golang.org/x/sync v0.6.0 h1:5BMeUDZ7vkXGfEr1x9B4bRcTH4lpkTkpdh0T/J+qjbQ=
|
||||
@@ -248,6 +265,8 @@ golang.org/x/sys v0.18.0 h1:DBdB3niSjOA/O0blCZBqDefyWNYveAYMNF1Wum0DYQ4=
|
||||
golang.org/x/sys v0.18.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
|
||||
golang.org/x/sys v0.19.0 h1:q5f1RH2jigJ1MoAWp2KTp3gm5zAGFUTarQZ5U386+4o=
|
||||
golang.org/x/sys v0.19.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
|
||||
golang.org/x/sys v0.20.0 h1:Od9JTbYCk261bKm4M/mw7AklTlFYIa0bIp9BgSm1S8Y=
|
||||
golang.org/x/sys v0.20.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
|
||||
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
|
||||
golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
|
||||
golang.org/x/term v0.16.0 h1:m+B6fahuftsE9qjo0VWp2FW0mB3MTJvR0BaMQrq0pmE=
|
||||
@@ -258,6 +277,8 @@ golang.org/x/term v0.18.0 h1:FcHjZXDMxI8mM3nwhX9HlKop4C0YQvCVCdwYl2wOtE8=
|
||||
golang.org/x/term v0.18.0/go.mod h1:ILwASektA3OnRv7amZ1xhE/KTR+u50pbXfZ03+6Nx58=
|
||||
golang.org/x/term v0.19.0 h1:+ThwsDv+tYfnJFhF4L8jITxu1tdTWRTZpdsWgEgjL6Q=
|
||||
golang.org/x/term v0.19.0/go.mod h1:2CuTdWZ7KHSQwUzKva0cbMg6q2DMI3Mmxp+gKJbskEk=
|
||||
golang.org/x/term v0.20.0 h1:VnkxpohqXaOBYJtBmEppKUG6mXpi+4O6purfc2+sMhw=
|
||||
golang.org/x/term v0.20.0/go.mod h1:8UkIAJTvZgivsXaD6/pH6U9ecQzZ45awqEOzuCvwpFY=
|
||||
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
||||
golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
|
||||
golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
|
||||
@@ -266,6 +287,8 @@ golang.org/x/text v0.3.8/go.mod h1:E6s5w1FMmriuDzIBO73fBruAKo1PCIq6d2Q6DHfQ8WQ=
|
||||
golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=
|
||||
golang.org/x/text v0.14.0 h1:ScX5w1eTa3QqT8oi6+ziP7dTV1S2+ALU0bI+0zXKWiQ=
|
||||
golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
|
||||
golang.org/x/text v0.15.0 h1:h1V/4gjBv8v9cjcR6+AR5+/cIYK5N/WAgiv4xlsEtAk=
|
||||
golang.org/x/text v0.15.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
|
||||
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||
golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
|
||||
golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=
|
||||
@@ -279,6 +302,8 @@ google.golang.org/protobuf v1.33.0 h1:uNO2rsAINq/JlFpSdYEKIZ0uKD/R9cpdv0T+yoGwGm
|
||||
google.golang.org/protobuf v1.33.0/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos=
|
||||
google.golang.org/protobuf v1.34.0 h1:Qo/qEd2RZPCf2nKuorzksSknv0d3ERwp1vFG38gSmH4=
|
||||
google.golang.org/protobuf v1.34.0/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos=
|
||||
google.golang.org/protobuf v1.34.1 h1:9ddQBjfCyZPOHPUiPxpYESBLc+T8P3E+Vo4IbKZgFWg=
|
||||
google.golang.org/protobuf v1.34.1/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
|
@@ -1,5 +1,5 @@
|
||||
package goext
|
||||
|
||||
const GoextVersion = "0.0.445"
|
||||
const GoextVersion = "0.0.450"
|
||||
|
||||
const GoextVersionTimestamp = "2024-05-03T15:28:53+0200"
|
||||
const GoextVersionTimestamp = "2024-05-14T11:52:56+0200"
|
||||
|
@@ -52,6 +52,7 @@ type Coll[TData any] struct {
|
||||
customDecoder *func(ctx context.Context, dec Decodable) (TData, error) // custom decoding function (useful if TData is an interface)
|
||||
isInterfaceDataType bool // true if TData is an interface (not a struct)
|
||||
unmarshalHooks []func(d TData) TData // called for every object after unmarshalling
|
||||
marshalHooks []func(d TData) TData // called for every object before marshalling
|
||||
extraModPipeline []func(ctx context.Context) mongo.Pipeline // appended to pipelines after filter/limit/skip/sort, used for $lookup, $set, $unset, $project, etc
|
||||
}
|
||||
|
||||
@@ -83,18 +84,32 @@ func (c *Coll[TData]) WithDecodeFunc(cdf func(ctx context.Context, dec Decodable
|
||||
return c
|
||||
}
|
||||
|
||||
// WithUnmarshalHook
|
||||
// function that is called for every object after reading from DB
|
||||
func (c *Coll[TData]) WithUnmarshalHook(fn func(d TData) TData) *Coll[TData] {
|
||||
c.unmarshalHooks = append(c.unmarshalHooks, fn)
|
||||
|
||||
return c
|
||||
}
|
||||
|
||||
// WithMarshalHook
|
||||
// function that is called for every object before writing to DB
|
||||
func (c *Coll[TData]) WithMarshalHook(fn func(d TData) TData) *Coll[TData] {
|
||||
c.marshalHooks = append(c.marshalHooks, fn)
|
||||
|
||||
return c
|
||||
}
|
||||
|
||||
// WithModifyingPipeline
|
||||
// pipeline that is appended to all read operations (after filtering)
|
||||
func (c *Coll[TData]) WithModifyingPipeline(p mongo.Pipeline) *Coll[TData] {
|
||||
c.extraModPipeline = append(c.extraModPipeline, func(ctx context.Context) mongo.Pipeline { return p })
|
||||
|
||||
return c
|
||||
}
|
||||
|
||||
// WithModifyingPipelineFunc
|
||||
// pipeline that is appended to all read operations (after filtering)
|
||||
func (c *Coll[TData]) WithModifyingPipelineFunc(fn func(ctx context.Context) mongo.Pipeline) *Coll[TData] {
|
||||
c.extraModPipeline = append(c.extraModPipeline, fn)
|
||||
|
||||
|
@@ -9,6 +9,10 @@ import (
|
||||
)
|
||||
|
||||
func (c *Coll[TData]) InsertOne(ctx context.Context, valueIn TData) (TData, error) {
|
||||
for _, hook := range c.marshalHooks {
|
||||
valueIn = hook(valueIn)
|
||||
}
|
||||
|
||||
insRes, err := c.coll.InsertOne(ctx, valueIn)
|
||||
if err != nil {
|
||||
return *new(TData), exerr.Wrap(err, "mongo-query[insert-one] failed").Str("collection", c.Name()).Build()
|
||||
@@ -36,6 +40,12 @@ func (c *Coll[TData]) InsertOneUnchecked(ctx context.Context, valueIn any) (TDat
|
||||
}
|
||||
|
||||
func (c *Coll[TData]) InsertMany(ctx context.Context, valueIn []TData) (*mongo.InsertManyResult, error) {
|
||||
for _, hook := range c.marshalHooks {
|
||||
for i := 0; i < len(valueIn); i++ {
|
||||
valueIn[i] = hook(valueIn[i])
|
||||
}
|
||||
}
|
||||
|
||||
insRes, err := c.coll.InsertMany(ctx, langext.ArrayToInterface(valueIn))
|
||||
if err != nil {
|
||||
return nil, exerr.Wrap(err, "mongo-query[insert-many] failed").Int("len(valueIn)", len(valueIn)).Str("collection", c.Name()).Build()
|
||||
|
@@ -61,6 +61,10 @@ func (c *Coll[TData]) UpdateMany(ctx context.Context, filterQuery bson.M, update
|
||||
}
|
||||
|
||||
func (c *Coll[TData]) ReplaceOne(ctx context.Context, filterQuery bson.M, value TData) error {
|
||||
for _, hook := range c.marshalHooks {
|
||||
value = hook(value)
|
||||
}
|
||||
|
||||
_, err := c.coll.UpdateOne(ctx, filterQuery, bson.M{"$set": value})
|
||||
if err != nil {
|
||||
return exerr.Wrap(err, "mongo-query[replace-one] failed").
|
||||
@@ -73,6 +77,10 @@ func (c *Coll[TData]) ReplaceOne(ctx context.Context, filterQuery bson.M, value
|
||||
}
|
||||
|
||||
func (c *Coll[TData]) FindOneAndReplace(ctx context.Context, filterQuery bson.M, value TData) (TData, error) {
|
||||
for _, hook := range c.marshalHooks {
|
||||
value = hook(value)
|
||||
}
|
||||
|
||||
mongoRes := c.coll.FindOneAndReplace(ctx, filterQuery, value, options.FindOneAndReplace().SetReturnDocument(options.After))
|
||||
if err := mongoRes.Err(); err != nil {
|
||||
return *new(TData), exerr.Wrap(err, "mongo-query[find-one-and-update] failed").
|
||||
|
183
wpdf/wpdf.go
Normal file
183
wpdf/wpdf.go
Normal file
@@ -0,0 +1,183 @@
|
||||
package wpdf
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"github.com/jung-kurt/gofpdf"
|
||||
)
|
||||
|
||||
type WPDFBuilder struct {
|
||||
b *gofpdf.Fpdf
|
||||
tr func(string) string
|
||||
cellHeight float64
|
||||
fontName PDFFontFamily
|
||||
fontStyle PDFFontStyle
|
||||
fontSize float64
|
||||
}
|
||||
|
||||
type PDFMargins struct {
|
||||
Left float64
|
||||
Top float64
|
||||
Right float64
|
||||
}
|
||||
|
||||
type PDFColor struct {
|
||||
R int
|
||||
G int
|
||||
B int
|
||||
}
|
||||
|
||||
func NewPDFBuilder(orientation PDFOrientation, size PDFSize, unicode bool) *WPDFBuilder {
|
||||
|
||||
fpdfbuilder := gofpdf.New(string(orientation), "mm", string(size), "")
|
||||
|
||||
var tr func(string) string
|
||||
if unicode {
|
||||
tr = fpdfbuilder.UnicodeTranslatorFromDescriptor("")
|
||||
} else {
|
||||
tr = func(s string) string { return s }
|
||||
}
|
||||
|
||||
b := &WPDFBuilder{
|
||||
b: fpdfbuilder,
|
||||
tr: tr,
|
||||
cellHeight: 5,
|
||||
}
|
||||
|
||||
b.SetMargins(PDFMargins{Left: 15, Top: 25, Right: 15}) // default values
|
||||
b.SetFont(FontHelvetica, Normal, 12) // ensures font is set
|
||||
|
||||
return b
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) SetMargins(v PDFMargins) {
|
||||
b.b.SetMargins(v.Left, v.Top, v.Right)
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) AddPage() {
|
||||
b.b.AddPage()
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) SetTextColor(cr, cg, cb int) {
|
||||
b.b.SetTextColor(cr, cg, cb)
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) GetTextColor() (cr, cg, cb int) {
|
||||
return b.b.GetTextColor()
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) SetDrawColor(cr, cg, cb int) {
|
||||
b.b.SetDrawColor(cr, cg, cb)
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) GetDrawColor() (cr, cg, cb int) {
|
||||
return b.b.GetDrawColor()
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) SetFillColor(cr, cg, cb int) {
|
||||
b.b.SetFillColor(cr, cg, cb)
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) GetFillColor() (cr, cg, cb int) {
|
||||
return b.b.GetFillColor()
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) SetLineWidth(w float64) {
|
||||
b.b.SetLineWidth(w)
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) GetLineWidth() float64 {
|
||||
return b.b.GetLineWidth()
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) SetFont(fontName PDFFontFamily, fontStyle PDFFontStyle, fontSize float64) {
|
||||
b.b.SetFont(string(fontName), string(fontStyle), fontSize)
|
||||
|
||||
b.fontName = fontName
|
||||
b.fontStyle = fontStyle
|
||||
b.fontSize = fontSize
|
||||
|
||||
b.cellHeight = b.b.PointConvert(fontSize)
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) Ln(h float64) {
|
||||
b.b.Ln(h)
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) Build() ([]byte, error) {
|
||||
buf := new(bytes.Buffer)
|
||||
err := b.b.Output(buf)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return buf.Bytes(), nil
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) SetX(x float64) {
|
||||
b.b.SetX(x)
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) GetX() float64 {
|
||||
return b.b.GetX()
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) SetY(y float64) {
|
||||
b.b.SetY(y)
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) GetY() float64 {
|
||||
return b.b.GetY()
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) SetXY(x float64, y float64) {
|
||||
b.b.SetXY(x, y)
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) GetXY() (x float64, y float64) {
|
||||
return b.b.GetXY()
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) GetMargins() (left, top, right, bottom float64) {
|
||||
return b.b.GetMargins()
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) GetMarginLeft() float64 {
|
||||
v, _, _, _ := b.b.GetMargins()
|
||||
return v
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) GetMarginTop() float64 {
|
||||
_, v, _, _ := b.b.GetMargins()
|
||||
return v
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) GetMarginRight() float64 {
|
||||
_, _, v, _ := b.b.GetMargins()
|
||||
return v
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) GetMarginBottom() float64 {
|
||||
_, _, _, v := b.b.GetMargins()
|
||||
return v
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) GetPageSize() (width, height float64) {
|
||||
return b.b.GetPageSize()
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) GetPageWidth() float64 {
|
||||
v, _ := b.b.GetPageSize()
|
||||
return v
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) GetPageHeight() float64 {
|
||||
_, v := b.b.GetPageSize()
|
||||
return v
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) GetWorkAreaWidth() float64 {
|
||||
return b.GetPageWidth() - b.GetMarginLeft() - b.GetMarginRight()
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) GetStringWidth(str string) float64 {
|
||||
return b.b.GetStringWidth(str)
|
||||
}
|
173
wpdf/wpdfCell.go
Normal file
173
wpdf/wpdfCell.go
Normal file
@@ -0,0 +1,173 @@
|
||||
package wpdf
|
||||
|
||||
import "gogs.mikescher.com/BlackForestBytes/goext/langext"
|
||||
|
||||
type PDFCellOpt struct {
|
||||
width *float64
|
||||
height *float64
|
||||
border *PDFBorder
|
||||
ln *PDFTextBreak
|
||||
align *PDFTextAlign
|
||||
fill *bool
|
||||
link *int
|
||||
linkStr *string
|
||||
fontNameOverride *PDFFontFamily
|
||||
fontStyleOverride *PDFFontStyle
|
||||
fontSizeOverride *float64
|
||||
extraLn *float64
|
||||
x *float64
|
||||
autoWidth *bool
|
||||
}
|
||||
|
||||
func NewPDFCellOpt() *PDFCellOpt {
|
||||
return &PDFCellOpt{}
|
||||
}
|
||||
|
||||
func (opt *PDFCellOpt) Width(v float64) *PDFCellOpt {
|
||||
opt.width = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFCellOpt) Height(v float64) *PDFCellOpt {
|
||||
opt.height = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFCellOpt) Border(v PDFBorder) *PDFCellOpt {
|
||||
opt.border = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFCellOpt) LnPos(v PDFTextBreak) *PDFCellOpt {
|
||||
opt.ln = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFCellOpt) Align(v PDFTextAlign) *PDFCellOpt {
|
||||
opt.align = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFCellOpt) FillBackground(v bool) *PDFCellOpt {
|
||||
opt.fill = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFCellOpt) Link(v int) *PDFCellOpt {
|
||||
opt.link = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFCellOpt) LinkStr(v string) *PDFCellOpt {
|
||||
opt.linkStr = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFCellOpt) Font(fontName PDFFontFamily, fontStyle PDFFontStyle, fontSize float64) *PDFCellOpt {
|
||||
opt.fontNameOverride = &fontName
|
||||
opt.fontStyleOverride = &fontStyle
|
||||
opt.fontSizeOverride = &fontSize
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFCellOpt) FontName(v PDFFontFamily) *PDFCellOpt {
|
||||
opt.fontNameOverride = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFCellOpt) FontStyle(v PDFFontStyle) *PDFCellOpt {
|
||||
opt.fontStyleOverride = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFCellOpt) FontSize(v float64) *PDFCellOpt {
|
||||
opt.fontSizeOverride = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFCellOpt) Bold() *PDFCellOpt {
|
||||
opt.fontStyleOverride = langext.Ptr(Bold)
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFCellOpt) Italic() *PDFCellOpt {
|
||||
opt.fontStyleOverride = langext.Ptr(Italic)
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFCellOpt) LnAfter(v float64) *PDFCellOpt {
|
||||
opt.extraLn = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFCellOpt) X(v float64) *PDFCellOpt {
|
||||
opt.x = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFCellOpt) AutoWidth() *PDFCellOpt {
|
||||
opt.autoWidth = langext.PTrue
|
||||
return opt
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) Cell(txt string, opts ...*PDFCellOpt) {
|
||||
|
||||
txtTR := b.tr(txt)
|
||||
|
||||
width := float64(0)
|
||||
height := b.cellHeight
|
||||
border := BorderNone
|
||||
ln := BreakToNextLine
|
||||
align := AlignLeft
|
||||
fill := false
|
||||
link := 0
|
||||
linkStr := ""
|
||||
var fontNameOverride *PDFFontFamily
|
||||
var fontStyleOverride *PDFFontStyle
|
||||
var fontSizeOverride *float64
|
||||
extraLn := float64(0)
|
||||
var x *float64
|
||||
autoWidth := false
|
||||
|
||||
for _, opt := range opts {
|
||||
width = langext.Coalesce(opt.width, width)
|
||||
height = langext.Coalesce(opt.height, height)
|
||||
border = langext.Coalesce(opt.border, border)
|
||||
ln = langext.Coalesce(opt.ln, ln)
|
||||
align = langext.Coalesce(opt.align, align)
|
||||
fill = langext.Coalesce(opt.fill, fill)
|
||||
link = langext.Coalesce(opt.link, link)
|
||||
linkStr = langext.Coalesce(opt.linkStr, linkStr)
|
||||
fontNameOverride = langext.CoalesceOpt(opt.fontNameOverride, fontNameOverride)
|
||||
fontStyleOverride = langext.CoalesceOpt(opt.fontStyleOverride, fontStyleOverride)
|
||||
fontSizeOverride = langext.CoalesceOpt(opt.fontSizeOverride, fontSizeOverride)
|
||||
extraLn = langext.Coalesce(opt.extraLn, extraLn)
|
||||
x = langext.CoalesceOpt(opt.x, x)
|
||||
autoWidth = langext.Coalesce(opt.autoWidth, autoWidth)
|
||||
}
|
||||
|
||||
if fontNameOverride != nil || fontStyleOverride != nil || fontSizeOverride != nil {
|
||||
oldFontName := b.fontName
|
||||
oldFontStyle := b.fontStyle
|
||||
oldFontSize := b.fontSize
|
||||
newFontName := langext.Coalesce(fontNameOverride, oldFontName)
|
||||
newFontStyle := langext.Coalesce(fontStyleOverride, oldFontStyle)
|
||||
newFontSize := langext.Coalesce(fontSizeOverride, oldFontSize)
|
||||
b.SetFont(newFontName, newFontStyle, newFontSize)
|
||||
defer func() { b.SetFont(oldFontName, oldFontStyle, oldFontSize) }()
|
||||
}
|
||||
|
||||
if x != nil {
|
||||
b.b.SetX(*x)
|
||||
}
|
||||
|
||||
if autoWidth {
|
||||
width = b.b.GetStringWidth(txtTR)
|
||||
}
|
||||
|
||||
b.b.CellFormat(width, height, txtTR, string(border), int(ln), string(align), fill, link, linkStr)
|
||||
|
||||
if extraLn != 0 {
|
||||
b.b.Ln(extraLn)
|
||||
}
|
||||
}
|
80
wpdf/wpdfConstants.go
Normal file
80
wpdf/wpdfConstants.go
Normal file
@@ -0,0 +1,80 @@
|
||||
package wpdf
|
||||
|
||||
type PDFOrientation string
|
||||
|
||||
const (
|
||||
Portrait PDFOrientation = "P"
|
||||
Landscape PDFOrientation = "L"
|
||||
)
|
||||
|
||||
type PDFSize string
|
||||
|
||||
const (
|
||||
SizeA3 PDFSize = "A3"
|
||||
SizeA4 PDFSize = "A4"
|
||||
SizeA5 PDFSize = "A4"
|
||||
SizeLetter PDFSize = "Letter"
|
||||
SizeLegal PDFSize = "Legal"
|
||||
SizeTabloid PDFSize = "Tabloid"
|
||||
)
|
||||
|
||||
type PDFFontFamily string
|
||||
|
||||
const (
|
||||
FontCourier PDFFontFamily = "courier"
|
||||
FontHelvetica PDFFontFamily = "helvetica"
|
||||
FontTimes PDFFontFamily = "times"
|
||||
FontZapfDingbats PDFFontFamily = "zapfdingbats"
|
||||
FontSymbol PDFFontFamily = "symbol"
|
||||
)
|
||||
|
||||
type PDFFontStyle string
|
||||
|
||||
const (
|
||||
Normal PDFFontStyle = ""
|
||||
Bold PDFFontStyle = "B"
|
||||
Italic PDFFontStyle = "I"
|
||||
BoldItalic PDFFontStyle = "IB"
|
||||
)
|
||||
|
||||
type PDFBorder string
|
||||
|
||||
const (
|
||||
BorderNone PDFBorder = ""
|
||||
BorderFull PDFBorder = "1"
|
||||
BorderLeft PDFBorder = "L"
|
||||
BorderTop PDFBorder = "T"
|
||||
BorderRight PDFBorder = "R"
|
||||
BorderBottom PDFBorder = "B"
|
||||
BorderTLR PDFBorder = "TLR"
|
||||
BorderLR PDFBorder = "LR"
|
||||
)
|
||||
|
||||
type PDFTextBreak int
|
||||
|
||||
const (
|
||||
BreakToRight PDFTextBreak = 0
|
||||
BreakToNextLine PDFTextBreak = 1
|
||||
BreakToBelow PDFTextBreak = 2
|
||||
)
|
||||
|
||||
type PDFTextAlign string
|
||||
|
||||
const (
|
||||
AlignLeft PDFTextAlign = "L"
|
||||
AlignHorzCenter PDFTextAlign = "C"
|
||||
AlignRight PDFTextAlign = "R"
|
||||
)
|
||||
|
||||
type PDFRectStyle string
|
||||
|
||||
const (
|
||||
RectFill PDFRectStyle = "F"
|
||||
RectOutline PDFRectStyle = "D"
|
||||
RectFillOutline PDFRectStyle = "FD"
|
||||
)
|
||||
|
||||
const (
|
||||
BackgroundFill = true
|
||||
BackgroundTransparent = false
|
||||
)
|
186
wpdf/wpdfImage.go
Normal file
186
wpdf/wpdfImage.go
Normal file
@@ -0,0 +1,186 @@
|
||||
package wpdf
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"github.com/jung-kurt/gofpdf"
|
||||
"gogs.mikescher.com/BlackForestBytes/goext/langext"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
type PDFImageRef struct {
|
||||
Info *gofpdf.ImageInfoType
|
||||
Name string
|
||||
}
|
||||
|
||||
type PDFImageRegisterOpt struct {
|
||||
imageType *string
|
||||
readDpi *bool
|
||||
allowNegativePosition *bool
|
||||
name *string
|
||||
}
|
||||
|
||||
func NewPDFImageRegisterOpt() *PDFImageRegisterOpt {
|
||||
return &PDFImageRegisterOpt{}
|
||||
}
|
||||
|
||||
func (opt *PDFImageRegisterOpt) ImageType(v string) *PDFImageRegisterOpt {
|
||||
opt.imageType = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFImageRegisterOpt) ReadDpi(v bool) *PDFImageRegisterOpt {
|
||||
opt.readDpi = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFImageRegisterOpt) AllowNegativePosition(v bool) *PDFImageRegisterOpt {
|
||||
opt.allowNegativePosition = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFImageRegisterOpt) Name(v string) *PDFImageRegisterOpt {
|
||||
opt.name = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) RegisterImage(bin []byte, opts ...*PDFImageRegisterOpt) *PDFImageRef {
|
||||
imgName := "fpdf_img_" + langext.MustRawHexUUID()
|
||||
imageType := ""
|
||||
readDpi := false
|
||||
allowNegativePosition := false
|
||||
|
||||
for _, opt := range opts {
|
||||
imageType = langext.Coalesce(opt.imageType, imageType)
|
||||
readDpi = langext.Coalesce(opt.readDpi, readDpi)
|
||||
allowNegativePosition = langext.Coalesce(opt.allowNegativePosition, allowNegativePosition)
|
||||
imgName = langext.Coalesce(opt.name, imgName)
|
||||
}
|
||||
|
||||
if imageType == "" {
|
||||
ct := http.DetectContentType(bin[:512])
|
||||
switch ct {
|
||||
case "image/jpg":
|
||||
imageType = "JPG"
|
||||
case "image/jpeg":
|
||||
imageType = "JPEG"
|
||||
case "image/png":
|
||||
imageType = "PNG"
|
||||
case "image/gif":
|
||||
imageType = "GIF"
|
||||
}
|
||||
}
|
||||
|
||||
options := gofpdf.ImageOptions{
|
||||
ImageType: imageType,
|
||||
ReadDpi: readDpi,
|
||||
AllowNegativePosition: allowNegativePosition,
|
||||
}
|
||||
|
||||
info := b.b.RegisterImageOptionsReader(imgName, options, bytes.NewReader(bin))
|
||||
|
||||
return &PDFImageRef{
|
||||
Name: imgName,
|
||||
Info: info,
|
||||
}
|
||||
}
|
||||
|
||||
type PDFImageOpt struct {
|
||||
x *float64
|
||||
y *float64
|
||||
width *float64
|
||||
height *float64
|
||||
flow *bool
|
||||
link *int
|
||||
linkStr *string
|
||||
imageType *string
|
||||
readDpi *bool
|
||||
allowNegativePosition *bool
|
||||
}
|
||||
|
||||
func NewPDFImageOpt() *PDFImageOpt {
|
||||
return &PDFImageOpt{}
|
||||
}
|
||||
|
||||
func (opt *PDFImageOpt) X(v float64) *PDFImageOpt {
|
||||
opt.x = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFImageOpt) Y(v float64) *PDFImageOpt {
|
||||
opt.y = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFImageOpt) Width(v float64) *PDFImageOpt {
|
||||
opt.width = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFImageOpt) Height(v float64) *PDFImageOpt {
|
||||
opt.height = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFImageOpt) Flow(v bool) *PDFImageOpt {
|
||||
opt.flow = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFImageOpt) Link(v int) *PDFImageOpt {
|
||||
opt.link = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFImageOpt) LinkStr(v string) *PDFImageOpt {
|
||||
opt.linkStr = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFImageOpt) ImageType(v string) *PDFImageOpt {
|
||||
opt.imageType = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFImageOpt) ReadDpi(v bool) *PDFImageOpt {
|
||||
opt.readDpi = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFImageOpt) AllowNegativePosition(v bool) *PDFImageOpt {
|
||||
opt.allowNegativePosition = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) Image(img *PDFImageRef, opts ...*PDFImageOpt) {
|
||||
x := b.GetX()
|
||||
y := b.GetY()
|
||||
w := img.Info.Width()
|
||||
h := img.Info.Height()
|
||||
flow := true
|
||||
link := 0
|
||||
linkStr := ""
|
||||
imageType := ""
|
||||
readDpi := false
|
||||
allowNegativePosition := false
|
||||
|
||||
for _, opt := range opts {
|
||||
x = langext.Coalesce(opt.x, x)
|
||||
y = langext.Coalesce(opt.y, y)
|
||||
w = langext.Coalesce(opt.width, w)
|
||||
h = langext.Coalesce(opt.height, h)
|
||||
flow = langext.Coalesce(opt.flow, flow)
|
||||
link = langext.Coalesce(opt.link, link)
|
||||
linkStr = langext.Coalesce(opt.linkStr, linkStr)
|
||||
imageType = langext.Coalesce(opt.imageType, imageType)
|
||||
readDpi = langext.Coalesce(opt.readDpi, readDpi)
|
||||
allowNegativePosition = langext.Coalesce(opt.allowNegativePosition, allowNegativePosition)
|
||||
}
|
||||
|
||||
fpdfOpt := gofpdf.ImageOptions{
|
||||
ImageType: imageType,
|
||||
ReadDpi: readDpi,
|
||||
AllowNegativePosition: allowNegativePosition,
|
||||
}
|
||||
|
||||
b.b.ImageOptions(img.Name, x, y, w, h, flow, fpdfOpt, link, linkStr)
|
||||
}
|
137
wpdf/wpdfMultiCell.go
Normal file
137
wpdf/wpdfMultiCell.go
Normal file
@@ -0,0 +1,137 @@
|
||||
package wpdf
|
||||
|
||||
import "gogs.mikescher.com/BlackForestBytes/goext/langext"
|
||||
|
||||
type PDFMultiCellOpt struct {
|
||||
width *float64
|
||||
height *float64
|
||||
border *PDFBorder
|
||||
align *PDFTextAlign
|
||||
fill *bool
|
||||
fontNameOverride *PDFFontFamily
|
||||
fontStyleOverride *PDFFontStyle
|
||||
fontSizeOverride *float64
|
||||
extraLn *float64
|
||||
x *float64
|
||||
}
|
||||
|
||||
func NewPDFMultiCellOpt() *PDFMultiCellOpt {
|
||||
return &PDFMultiCellOpt{}
|
||||
}
|
||||
|
||||
func (opt *PDFMultiCellOpt) Width(v float64) *PDFMultiCellOpt {
|
||||
opt.width = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFMultiCellOpt) Height(v float64) *PDFMultiCellOpt {
|
||||
opt.height = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFMultiCellOpt) Border(v PDFBorder) *PDFMultiCellOpt {
|
||||
opt.border = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFMultiCellOpt) Align(v PDFTextAlign) *PDFMultiCellOpt {
|
||||
opt.align = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFMultiCellOpt) FillBackground(v bool) *PDFMultiCellOpt {
|
||||
opt.fill = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFMultiCellOpt) Font(fontName PDFFontFamily, fontStyle PDFFontStyle, fontSize float64) *PDFMultiCellOpt {
|
||||
opt.fontNameOverride = &fontName
|
||||
opt.fontStyleOverride = &fontStyle
|
||||
opt.fontSizeOverride = &fontSize
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFMultiCellOpt) FontName(v PDFFontFamily) *PDFMultiCellOpt {
|
||||
opt.fontNameOverride = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFMultiCellOpt) FontStyle(v PDFFontStyle) *PDFMultiCellOpt {
|
||||
opt.fontStyleOverride = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFMultiCellOpt) FontSize(v float64) *PDFMultiCellOpt {
|
||||
opt.fontSizeOverride = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFMultiCellOpt) Bold() *PDFMultiCellOpt {
|
||||
opt.fontStyleOverride = langext.Ptr(Bold)
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFMultiCellOpt) Italic() *PDFMultiCellOpt {
|
||||
opt.fontStyleOverride = langext.Ptr(Italic)
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFMultiCellOpt) LnAfter(v float64) *PDFMultiCellOpt {
|
||||
opt.extraLn = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFMultiCellOpt) X(v float64) *PDFMultiCellOpt {
|
||||
opt.x = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) MultiCell(txt string, opts ...*PDFMultiCellOpt) {
|
||||
|
||||
txtTR := b.tr(txt)
|
||||
|
||||
width := float64(0)
|
||||
height := b.cellHeight
|
||||
border := BorderNone
|
||||
align := AlignLeft
|
||||
fill := false
|
||||
var fontNameOverride *PDFFontFamily
|
||||
var fontStyleOverride *PDFFontStyle
|
||||
var fontSizeOverride *float64
|
||||
extraLn := float64(0)
|
||||
var x *float64
|
||||
|
||||
for _, opt := range opts {
|
||||
width = langext.Coalesce(opt.width, width)
|
||||
height = langext.Coalesce(opt.height, height)
|
||||
border = langext.Coalesce(opt.border, border)
|
||||
align = langext.Coalesce(opt.align, align)
|
||||
fill = langext.Coalesce(opt.fill, fill)
|
||||
fontNameOverride = langext.CoalesceOpt(opt.fontNameOverride, fontNameOverride)
|
||||
fontStyleOverride = langext.CoalesceOpt(opt.fontStyleOverride, fontStyleOverride)
|
||||
fontSizeOverride = langext.CoalesceOpt(opt.fontSizeOverride, fontSizeOverride)
|
||||
extraLn = langext.Coalesce(opt.extraLn, extraLn)
|
||||
x = langext.CoalesceOpt(opt.x, x)
|
||||
}
|
||||
|
||||
if fontNameOverride != nil || fontStyleOverride != nil || fontSizeOverride != nil {
|
||||
oldFontName := b.fontName
|
||||
oldFontStyle := b.fontStyle
|
||||
oldFontSize := b.fontSize
|
||||
newFontName := langext.Coalesce(fontNameOverride, oldFontName)
|
||||
newFontStyle := langext.Coalesce(fontStyleOverride, oldFontStyle)
|
||||
newFontSize := langext.Coalesce(fontSizeOverride, oldFontSize)
|
||||
b.SetFont(newFontName, newFontStyle, newFontSize)
|
||||
defer func() { b.SetFont(oldFontName, oldFontStyle, oldFontSize) }()
|
||||
}
|
||||
|
||||
if x != nil {
|
||||
b.b.SetX(*x)
|
||||
}
|
||||
|
||||
b.b.MultiCell(width, height, txtTR, string(border), string(align), fill)
|
||||
|
||||
if extraLn != 0 {
|
||||
b.b.Ln(extraLn)
|
||||
}
|
||||
}
|
126
wpdf/wpdfRect.go
Normal file
126
wpdf/wpdfRect.go
Normal file
@@ -0,0 +1,126 @@
|
||||
package wpdf
|
||||
|
||||
import "gogs.mikescher.com/BlackForestBytes/goext/langext"
|
||||
|
||||
type PDFRectOpt struct {
|
||||
x *float64
|
||||
y *float64
|
||||
lineWidth *float64
|
||||
drawColor *PDFColor
|
||||
fillColor *PDFColor
|
||||
radiusTL *float64
|
||||
radiusTR *float64
|
||||
radiusBR *float64
|
||||
radiusBL *float64
|
||||
}
|
||||
|
||||
func NewPDFRectOpt() *PDFRectOpt {
|
||||
return &PDFRectOpt{}
|
||||
}
|
||||
|
||||
func (opt *PDFRectOpt) X(v float64) *PDFRectOpt {
|
||||
opt.x = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFRectOpt) Y(v float64) *PDFRectOpt {
|
||||
opt.y = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFRectOpt) LineWidth(v float64) *PDFRectOpt {
|
||||
opt.lineWidth = &v
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFRectOpt) DrawColor(cr, cg, cb int) *PDFRectOpt {
|
||||
opt.drawColor = &PDFColor{R: cr, G: cg, B: cb}
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFRectOpt) DrawColorHex(c uint32) *PDFRectOpt {
|
||||
opt.drawColor = &PDFColor{R: int((c >> 16) & 0xFF), G: int((c >> 8) & 0xFF), B: int((c >> 0) & 0xFF)}
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFRectOpt) FillColor(cr, cg, cb int) *PDFRectOpt {
|
||||
opt.fillColor = &PDFColor{R: cr, G: cg, B: cb}
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFRectOpt) FillColorHex(c uint32) *PDFRectOpt {
|
||||
opt.fillColor = &PDFColor{R: int((c >> 16) & 0xFF), G: int((c >> 8) & 0xFF), B: int((c >> 0) & 0xFF)}
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFRectOpt) Rounded(radius float64) *PDFRectOpt {
|
||||
opt.radiusTL = &radius
|
||||
opt.radiusTR = &radius
|
||||
opt.radiusBR = &radius
|
||||
opt.radiusBL = &radius
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFRectOpt) RadiusTL(radius float64) *PDFRectOpt {
|
||||
opt.radiusTL = &radius
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFRectOpt) RadiusTR(radius float64) *PDFRectOpt {
|
||||
opt.radiusTR = &radius
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFRectOpt) RadiusBL(radius float64) *PDFRectOpt {
|
||||
opt.radiusBL = &radius
|
||||
return opt
|
||||
}
|
||||
|
||||
func (opt *PDFRectOpt) RadiusBR(radius float64) *PDFRectOpt {
|
||||
opt.radiusBR = &radius
|
||||
return opt
|
||||
}
|
||||
|
||||
func (b *WPDFBuilder) Rect(w float64, h float64, styleStr PDFRectStyle, opts ...*PDFRectOpt) {
|
||||
x := b.GetX()
|
||||
y := b.GetY()
|
||||
var lineWidth *float64
|
||||
var drawColor *PDFColor
|
||||
var fillColor *PDFColor
|
||||
radiusTL := float64(0)
|
||||
radiusTR := float64(0)
|
||||
radiusBR := float64(0)
|
||||
radiusBL := float64(0)
|
||||
|
||||
for _, opt := range opts {
|
||||
x = langext.Coalesce(opt.x, x)
|
||||
y = langext.Coalesce(opt.y, y)
|
||||
lineWidth = langext.CoalesceOpt(opt.lineWidth, lineWidth)
|
||||
drawColor = langext.CoalesceOpt(opt.drawColor, drawColor)
|
||||
fillColor = langext.CoalesceOpt(opt.fillColor, fillColor)
|
||||
radiusTL = langext.Coalesce(opt.radiusTL, radiusTL)
|
||||
radiusTR = langext.Coalesce(opt.radiusTR, radiusTR)
|
||||
radiusBR = langext.Coalesce(opt.radiusBR, radiusBR)
|
||||
radiusBL = langext.Coalesce(opt.radiusBL, radiusBL)
|
||||
}
|
||||
|
||||
if lineWidth != nil {
|
||||
old := b.GetLineWidth()
|
||||
b.SetLineWidth(*lineWidth)
|
||||
defer func() { b.SetLineWidth(old) }()
|
||||
}
|
||||
|
||||
if drawColor != nil {
|
||||
oldR, oldG, oldB := b.GetDrawColor()
|
||||
b.SetDrawColor(drawColor.R, drawColor.G, drawColor.B)
|
||||
defer func() { b.SetDrawColor(oldR, oldG, oldB) }()
|
||||
}
|
||||
|
||||
if fillColor != nil {
|
||||
oldR, oldG, oldB := b.GetFillColor()
|
||||
b.SetFillColor(fillColor.R, fillColor.G, fillColor.B)
|
||||
defer func() { b.SetFillColor(oldR, oldG, oldB) }()
|
||||
}
|
||||
|
||||
b.b.RoundedRectExt(x, y, w, h, radiusTL, radiusTR, radiusBR, radiusBL, string(styleStr))
|
||||
}
|
Reference in New Issue
Block a user