mirror of
https://github.com/jimeh/go-midjourney.git
synced 2026-02-19 01:46:41 +00:00
feat(job): add support for _parsed_params job field
This commit is contained in:
37
algorithm_version.go
Normal file
37
algorithm_version.go
Normal file
@@ -0,0 +1,37 @@
|
||||
package midjourney
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
type AlgorithmVersion string
|
||||
|
||||
func (av *AlgorithmVersion) MarshalJSON() ([]byte, error) {
|
||||
n, err := strconv.Atoi(string(*av))
|
||||
if err != nil {
|
||||
return json.Marshal(string(*av))
|
||||
}
|
||||
|
||||
return json.Marshal(n)
|
||||
}
|
||||
|
||||
func (av *AlgorithmVersion) UnmarshalJSON(b []byte) error {
|
||||
var n int
|
||||
err := json.Unmarshal(b, &n)
|
||||
if err == nil {
|
||||
*av = AlgorithmVersion(strconv.Itoa(n))
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
var s string
|
||||
err = json.Unmarshal(b, &s)
|
||||
if err == nil {
|
||||
*av = AlgorithmVersion(s)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
Reference in New Issue
Block a user