| 160 | } |
| 161 | |
| 162 | func restore(rd io.Reader) (interface{}, error) { |
| 163 | sr := sds.NewReader(rd) |
| 164 | var batch leveldb.Batch |
| 165 | for { |
| 166 | key, err := sr.ReadBytes() |
| 167 | if err != nil { |
| 168 | if err == io.EOF { |
| 169 | break |
| 170 | } |
| 171 | return nil, err |
| 172 | } |
| 173 | value, err := sr.ReadBytes() |
| 174 | if err != nil { |
| 175 | return nil, err |
| 176 | } |
| 177 | batch.Put(key, value) |
| 178 | if batch.Len() == 1000 { |
| 179 | if err := db.Write(&batch, nil); err != nil { |
| 180 | return nil, err |
| 181 | } |
| 182 | batch.Reset() |
| 183 | } |
| 184 | } |
| 185 | if err := db.Write(&batch, nil); err != nil { |
| 186 | return nil, err |
| 187 | } |
| 188 | return nil, nil |
| 189 | } |
| 190 | |
| 191 | func connOpened(addr string) (context interface{}, accept bool) { |
| 192 | atomic.AddInt64(&respClientNum, 1) |