Silverfish/proton-bridge
1
0
Fork 1
mirror of https://github.com/ProtonMail/proton-bridge.git synced 2025-12-10 04:36:43 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Other: Make user-agent implementation threadsafe

Browse Source
This commit is contained in:
James Houlahan
2022-10-24 16:12:37 +02:00
parent 798cd5caf3
commit d0fb3509cc
1 changed files with 15 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
internal/useragent/useragent.go
View File

@ -21,10 +21,13 @@ import (
"fmt"
"regexp"
"runtime"
"sync"
)
type UserAgent struct {
client, platform string
lock sync.RWMutex
}
func New() *UserAgent {
@ -35,18 +38,30 @@ func New() *UserAgent {
}
func (ua *UserAgent) SetClient(name, version string) {
ua.lock.Lock()
defer ua.lock.Unlock()
ua.client = fmt.Sprintf("%v/%v", name, regexp.MustCompile(`(.*) \((.*)\)`).ReplaceAllString(version, "$1-$2"))
}
func (ua *UserAgent) HasClient() bool {
ua.lock.RLock()
defer ua.lock.RUnlock()
return ua.client != ""
}
func (ua *UserAgent) SetPlatform(platform string) {
ua.lock.Lock()
defer ua.lock.Unlock()
ua.platform = platform
}
func (ua *UserAgent) GetUserAgent() string {
ua.lock.RLock()
defer ua.lock.RUnlock()
var client string
if ua.client != "" {