proxy.go 5.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248
  1. package graphdriver
  2. import (
  3. "errors"
  4. "fmt"
  5. "io"
  6. "github.com/docker/docker/pkg/archive"
  7. "github.com/docker/docker/pkg/plugingetter"
  8. )
  9. type graphDriverProxy struct {
  10. name string
  11. client pluginClient
  12. p plugingetter.CompatPlugin
  13. }
  14. type graphDriverRequest struct {
  15. ID string `json:",omitempty"`
  16. Parent string `json:",omitempty"`
  17. MountLabel string `json:",omitempty"`
  18. }
  19. type graphDriverResponse struct {
  20. Err string `json:",omitempty"`
  21. Dir string `json:",omitempty"`
  22. Exists bool `json:",omitempty"`
  23. Status [][2]string `json:",omitempty"`
  24. Changes []archive.Change `json:",omitempty"`
  25. Size int64 `json:",omitempty"`
  26. Metadata map[string]string `json:",omitempty"`
  27. }
  28. type graphDriverInitRequest struct {
  29. Home string
  30. Opts []string
  31. }
  32. func (d *graphDriverProxy) Init(home string, opts []string) error {
  33. if !d.p.IsV1() {
  34. if cp, ok := d.p.(plugingetter.CountedPlugin); ok {
  35. // always acquire here, it will be cleaned up on daemon shutdown
  36. cp.Acquire()
  37. }
  38. }
  39. args := &graphDriverInitRequest{
  40. Home: home,
  41. Opts: opts,
  42. }
  43. var ret graphDriverResponse
  44. if err := d.client.Call("GraphDriver.Init", args, &ret); err != nil {
  45. return err
  46. }
  47. if ret.Err != "" {
  48. return errors.New(ret.Err)
  49. }
  50. return nil
  51. }
  52. func (d *graphDriverProxy) String() string {
  53. return d.name
  54. }
  55. func (d *graphDriverProxy) CreateReadWrite(id, parent string, opts *CreateOpts) error {
  56. mountLabel := ""
  57. if opts != nil {
  58. mountLabel = opts.MountLabel
  59. }
  60. args := &graphDriverRequest{
  61. ID: id,
  62. Parent: parent,
  63. MountLabel: mountLabel,
  64. }
  65. var ret graphDriverResponse
  66. if err := d.client.Call("GraphDriver.CreateReadWrite", args, &ret); err != nil {
  67. return err
  68. }
  69. if ret.Err != "" {
  70. return errors.New(ret.Err)
  71. }
  72. return nil
  73. }
  74. func (d *graphDriverProxy) Create(id, parent string, opts *CreateOpts) error {
  75. mountLabel := ""
  76. if opts != nil {
  77. mountLabel = opts.MountLabel
  78. }
  79. args := &graphDriverRequest{
  80. ID: id,
  81. Parent: parent,
  82. MountLabel: mountLabel,
  83. }
  84. var ret graphDriverResponse
  85. if err := d.client.Call("GraphDriver.Create", args, &ret); err != nil {
  86. return err
  87. }
  88. if ret.Err != "" {
  89. return errors.New(ret.Err)
  90. }
  91. return nil
  92. }
  93. func (d *graphDriverProxy) Remove(id string) error {
  94. args := &graphDriverRequest{ID: id}
  95. var ret graphDriverResponse
  96. if err := d.client.Call("GraphDriver.Remove", args, &ret); err != nil {
  97. return err
  98. }
  99. if ret.Err != "" {
  100. return errors.New(ret.Err)
  101. }
  102. return nil
  103. }
  104. func (d *graphDriverProxy) Get(id, mountLabel string) (string, error) {
  105. args := &graphDriverRequest{
  106. ID: id,
  107. MountLabel: mountLabel,
  108. }
  109. var ret graphDriverResponse
  110. if err := d.client.Call("GraphDriver.Get", args, &ret); err != nil {
  111. return "", err
  112. }
  113. var err error
  114. if ret.Err != "" {
  115. err = errors.New(ret.Err)
  116. }
  117. return ret.Dir, err
  118. }
  119. func (d *graphDriverProxy) Put(id string) error {
  120. args := &graphDriverRequest{ID: id}
  121. var ret graphDriverResponse
  122. if err := d.client.Call("GraphDriver.Put", args, &ret); err != nil {
  123. return err
  124. }
  125. if ret.Err != "" {
  126. return errors.New(ret.Err)
  127. }
  128. return nil
  129. }
  130. func (d *graphDriverProxy) Exists(id string) bool {
  131. args := &graphDriverRequest{ID: id}
  132. var ret graphDriverResponse
  133. if err := d.client.Call("GraphDriver.Exists", args, &ret); err != nil {
  134. return false
  135. }
  136. return ret.Exists
  137. }
  138. func (d *graphDriverProxy) Status() [][2]string {
  139. args := &graphDriverRequest{}
  140. var ret graphDriverResponse
  141. if err := d.client.Call("GraphDriver.Status", args, &ret); err != nil {
  142. return nil
  143. }
  144. return ret.Status
  145. }
  146. func (d *graphDriverProxy) GetMetadata(id string) (map[string]string, error) {
  147. args := &graphDriverRequest{
  148. ID: id,
  149. }
  150. var ret graphDriverResponse
  151. if err := d.client.Call("GraphDriver.GetMetadata", args, &ret); err != nil {
  152. return nil, err
  153. }
  154. if ret.Err != "" {
  155. return nil, errors.New(ret.Err)
  156. }
  157. return ret.Metadata, nil
  158. }
  159. func (d *graphDriverProxy) Cleanup() error {
  160. if !d.p.IsV1() {
  161. if cp, ok := d.p.(plugingetter.CountedPlugin); ok {
  162. // always release
  163. defer cp.Release()
  164. }
  165. }
  166. args := &graphDriverRequest{}
  167. var ret graphDriverResponse
  168. if err := d.client.Call("GraphDriver.Cleanup", args, &ret); err != nil {
  169. return nil
  170. }
  171. if ret.Err != "" {
  172. return errors.New(ret.Err)
  173. }
  174. return nil
  175. }
  176. func (d *graphDriverProxy) Diff(id, parent string) (io.ReadCloser, error) {
  177. args := &graphDriverRequest{
  178. ID: id,
  179. Parent: parent,
  180. }
  181. body, err := d.client.Stream("GraphDriver.Diff", args)
  182. if err != nil {
  183. return nil, err
  184. }
  185. return body, nil
  186. }
  187. func (d *graphDriverProxy) Changes(id, parent string) ([]archive.Change, error) {
  188. args := &graphDriverRequest{
  189. ID: id,
  190. Parent: parent,
  191. }
  192. var ret graphDriverResponse
  193. if err := d.client.Call("GraphDriver.Changes", args, &ret); err != nil {
  194. return nil, err
  195. }
  196. if ret.Err != "" {
  197. return nil, errors.New(ret.Err)
  198. }
  199. return ret.Changes, nil
  200. }
  201. func (d *graphDriverProxy) ApplyDiff(id, parent string, diff io.Reader) (int64, error) {
  202. var ret graphDriverResponse
  203. if err := d.client.SendFile(fmt.Sprintf("GraphDriver.ApplyDiff?id=%s&parent=%s", id, parent), diff, &ret); err != nil {
  204. return -1, err
  205. }
  206. if ret.Err != "" {
  207. return -1, errors.New(ret.Err)
  208. }
  209. return ret.Size, nil
  210. }
  211. func (d *graphDriverProxy) DiffSize(id, parent string) (int64, error) {
  212. args := &graphDriverRequest{
  213. ID: id,
  214. Parent: parent,
  215. }
  216. var ret graphDriverResponse
  217. if err := d.client.Call("GraphDriver.DiffSize", args, &ret); err != nil {
  218. return -1, err
  219. }
  220. if ret.Err != "" {
  221. return -1, errors.New(ret.Err)
  222. }
  223. return ret.Size, nil
  224. }