Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
ollama
Commits
897b2134
Unverified
Commit
897b2134
authored
Feb 20, 2024
by
Michael Yang
Committed by
GitHub
Feb 20, 2024
Browse files
use http.DefaultClient (#2530)
default client already handles proxy
parent
4613a080
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
9 additions
and
52 deletions
+9
-52
api/client.go
api/client.go
+4
-21
app/lifecycle/updater.go
app/lifecycle/updater.go
+4
-19
server/images.go
server/images.go
+1
-12
No files found.
api/client.go
View file @
897b2134
...
...
@@ -21,7 +21,7 @@ import (
type
Client
struct
{
base
*
url
.
URL
http
http
.
Client
http
*
http
.
Client
}
func
checkError
(
resp
*
http
.
Response
,
body
[]
byte
)
error
{
...
...
@@ -66,30 +66,13 @@ func ClientFromEnvironment() (*Client, error) {
}
}
client
:=
Client
{
return
&
Client
{
base
:
&
url
.
URL
{
Scheme
:
scheme
,
Host
:
net
.
JoinHostPort
(
host
,
port
),
},
}
mockRequest
,
err
:=
http
.
NewRequest
(
http
.
MethodHead
,
client
.
base
.
String
(),
nil
)
if
err
!=
nil
{
return
nil
,
err
}
proxyURL
,
err
:=
http
.
ProxyFromEnvironment
(
mockRequest
)
if
err
!=
nil
{
return
nil
,
err
}
client
.
http
=
http
.
Client
{
Transport
:
&
http
.
Transport
{
Proxy
:
http
.
ProxyURL
(
proxyURL
),
},
}
return
&
client
,
nil
http
:
http
.
DefaultClient
,
},
nil
}
func
(
c
*
Client
)
do
(
ctx
context
.
Context
,
method
,
path
string
,
reqData
,
respData
any
)
error
{
...
...
app/lifecycle/updater.go
View file @
897b2134
...
...
@@ -34,20 +34,6 @@ type UpdateResponse struct {
UpdateVersion
string
`json:"version"`
}
func
getClient
(
req
*
http
.
Request
)
http
.
Client
{
proxyURL
,
err
:=
http
.
ProxyFromEnvironment
(
req
)
if
err
!=
nil
{
slog
.
Warn
(
fmt
.
Sprintf
(
"failed to handle proxy: %s"
,
err
))
return
http
.
Client
{}
}
return
http
.
Client
{
Transport
:
&
http
.
Transport
{
Proxy
:
http
.
ProxyURL
(
proxyURL
),
},
}
}
func
IsNewReleaseAvailable
(
ctx
context
.
Context
)
(
bool
,
UpdateResponse
)
{
var
updateResp
UpdateResponse
...
...
@@ -83,10 +69,9 @@ func IsNewReleaseAvailable(ctx context.Context) (bool, UpdateResponse) {
}
req
.
Header
.
Set
(
"Authorization"
,
signature
)
req
.
Header
.
Set
(
"User-Agent"
,
fmt
.
Sprintf
(
"ollama/%s (%s %s) Go/%s"
,
version
.
Version
,
runtime
.
GOARCH
,
runtime
.
GOOS
,
runtime
.
Version
()))
client
:=
getClient
(
req
)
slog
.
Debug
(
"checking for available update"
,
"requestURL"
,
requestURL
)
resp
,
err
:=
c
lient
.
Do
(
req
)
resp
,
err
:=
http
.
DefaultC
lient
.
Do
(
req
)
if
err
!=
nil
{
slog
.
Warn
(
fmt
.
Sprintf
(
"failed to check for update: %s"
,
err
))
return
false
,
updateResp
...
...
@@ -119,8 +104,8 @@ func DownloadNewRelease(ctx context.Context, updateResp UpdateResponse) error {
if
err
!=
nil
{
return
err
}
client
:=
getClient
(
req
)
resp
,
err
:=
c
lient
.
Do
(
req
)
resp
,
err
:=
http
.
DefaultC
lient
.
Do
(
req
)
if
err
!=
nil
{
return
fmt
.
Errorf
(
"error checking update: %w"
,
err
)
}
...
...
@@ -151,7 +136,7 @@ func DownloadNewRelease(ctx context.Context, updateResp UpdateResponse) error {
cleanupOldDownloads
()
req
.
Method
=
http
.
MethodGet
resp
,
err
=
c
lient
.
Do
(
req
)
resp
,
err
=
http
.
DefaultC
lient
.
Do
(
req
)
if
err
!=
nil
{
return
fmt
.
Errorf
(
"error checking update: %w"
,
err
)
}
...
...
server/images.go
View file @
897b2134
...
...
@@ -1103,18 +1103,7 @@ func makeRequest(ctx context.Context, method string, requestURL *url.URL, header
req
.
ContentLength
=
contentLength
}
proxyURL
,
err
:=
http
.
ProxyFromEnvironment
(
req
)
if
err
!=
nil
{
return
nil
,
err
}
client
:=
http
.
Client
{
Transport
:
&
http
.
Transport
{
Proxy
:
http
.
ProxyURL
(
proxyURL
),
},
}
resp
,
err
:=
client
.
Do
(
req
)
resp
,
err
:=
http
.
DefaultClient
.
Do
(
req
)
if
err
!=
nil
{
return
nil
,
err
}
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment