├── .gitignore ├── LICENSE ├── README.md └── app.go /.gitignore: -------------------------------------------------------------------------------- 1 | # Compiled Object files, Static and Dynamic libs (Shared Objects) 2 | *.o 3 | *.a 4 | *.so 5 | 6 | # Folders 7 | _obj 8 | _test 9 | 10 | # Architecture specific extensions/prefixes 11 | *.[568vq] 12 | [568vq].out 13 | 14 | *.cgo1.go 15 | *.cgo2.c 16 | _cgo_defun.c 17 | _cgo_gotypes.go 18 | _cgo_export.* 19 | 20 | _testmain.go 21 | 22 | *.exe 23 | 24 | .DS_Store 25 | *.swp 26 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- 1 | Apache License 2 | Version 2.0, January 2004 3 | http://www.apache.org/licenses/ 4 | 5 | TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION 6 | 7 | 1. Definitions. 8 | 9 | "License" shall mean the terms and conditions for use, reproduction, and 10 | distribution as defined by Sections 1 through 9 of this document. 11 | 12 | "Licensor" shall mean the copyright owner or entity authorized by the copyright 13 | owner that is granting the License. 14 | 15 | "Legal Entity" shall mean the union of the acting entity and all other entities 16 | that control, are controlled by, or are under common control with that entity. 17 | For the purposes of this definition, "control" means (i) the power, direct or 18 | indirect, to cause the direction or management of such entity, whether by 19 | contract or otherwise, or (ii) ownership of fifty percent (50%) or more of the 20 | outstanding shares, or (iii) beneficial ownership of such entity. 21 | 22 | "You" (or "Your") shall mean an individual or Legal Entity exercising 23 | permissions granted by this License. 24 | 25 | "Source" form shall mean the preferred form for making modifications, including 26 | but not limited to software source code, documentation source, and configuration 27 | files. 28 | 29 | "Object" form shall mean any form resulting from mechanical transformation or 30 | translation of a Source form, including but not limited to compiled object code, 31 | generated documentation, and conversions to other media types. 32 | 33 | "Work" shall mean the work of authorship, whether in Source or Object form, made 34 | available under the License, as indicated by a copyright notice that is included 35 | in or attached to the work (an example is provided in the Appendix below). 36 | 37 | "Derivative Works" shall mean any work, whether in Source or Object form, that 38 | is based on (or derived from) the Work and for which the editorial revisions, 39 | annotations, elaborations, or other modifications represent, as a whole, an 40 | original work of authorship. For the purposes of this License, Derivative Works 41 | shall not include works that remain separable from, or merely link (or bind by 42 | name) to the interfaces of, the Work and Derivative Works thereof. 43 | 44 | "Contribution" shall mean any work of authorship, including the original version 45 | of the Work and any modifications or additions to that Work or Derivative Works 46 | thereof, that is intentionally submitted to Licensor for inclusion in the Work 47 | by the copyright owner or by an individual or Legal Entity authorized to submit 48 | on behalf of the copyright owner. For the purposes of this definition, 49 | "submitted" means any form of electronic, verbal, or written communication sent 50 | to the Licensor or its representatives, including but not limited to 51 | communication on electronic mailing lists, source code control systems, and 52 | issue tracking systems that are managed by, or on behalf of, the Licensor for 53 | the purpose of discussing and improving the Work, but excluding communication 54 | that is conspicuously marked or otherwise designated in writing by the copyright 55 | owner as "Not a Contribution." 56 | 57 | "Contributor" shall mean Licensor and any individual or Legal Entity on behalf 58 | of whom a Contribution has been received by Licensor and subsequently 59 | incorporated within the Work. 60 | 61 | 2. Grant of Copyright License. 62 | 63 | Subject to the terms and conditions of this License, each Contributor hereby 64 | grants to You a perpetual, worldwide, non-exclusive, no-charge, royalty-free, 65 | irrevocable copyright license to reproduce, prepare Derivative Works of, 66 | publicly display, publicly perform, sublicense, and distribute the Work and such 67 | Derivative Works in Source or Object form. 68 | 69 | 3. Grant of Patent License. 70 | 71 | Subject to the terms and conditions of this License, each Contributor hereby 72 | grants to You a perpetual, worldwide, non-exclusive, no-charge, royalty-free, 73 | irrevocable (except as stated in this section) patent license to make, have 74 | made, use, offer to sell, sell, import, and otherwise transfer the Work, where 75 | such license applies only to those patent claims licensable by such Contributor 76 | that are necessarily infringed by their Contribution(s) alone or by combination 77 | of their Contribution(s) with the Work to which such Contribution(s) was 78 | submitted. If You institute patent litigation against any entity (including a 79 | cross-claim or counterclaim in a lawsuit) alleging that the Work or a 80 | Contribution incorporated within the Work constitutes direct or contributory 81 | patent infringement, then any patent licenses granted to You under this License 82 | for that Work shall terminate as of the date such litigation is filed. 83 | 84 | 4. Redistribution. 85 | 86 | You may reproduce and distribute copies of the Work or Derivative Works thereof 87 | in any medium, with or without modifications, and in Source or Object form, 88 | provided that You meet the following conditions: 89 | 90 | You must give any other recipients of the Work or Derivative Works a copy of 91 | this License; and 92 | You must cause any modified files to carry prominent notices stating that You 93 | changed the files; and 94 | You must retain, in the Source form of any Derivative Works that You distribute, 95 | all copyright, patent, trademark, and attribution notices from the Source form 96 | of the Work, excluding those notices that do not pertain to any part of the 97 | Derivative Works; and 98 | If the Work includes a "NOTICE" text file as part of its distribution, then any 99 | Derivative Works that You distribute must include a readable copy of the 100 | attribution notices contained within such NOTICE file, excluding those notices 101 | that do not pertain to any part of the Derivative Works, in at least one of the 102 | following places: within a NOTICE text file distributed as part of the 103 | Derivative Works; within the Source form or documentation, if provided along 104 | with the Derivative Works; or, within a display generated by the Derivative 105 | Works, if and wherever such third-party notices normally appear. The contents of 106 | the NOTICE file are for informational purposes only and do not modify the 107 | License. You may add Your own attribution notices within Derivative Works that 108 | You distribute, alongside or as an addendum to the NOTICE text from the Work, 109 | provided that such additional attribution notices cannot be construed as 110 | modifying the License. 111 | You may add Your own copyright statement to Your modifications and may provide 112 | additional or different license terms and conditions for use, reproduction, or 113 | distribution of Your modifications, or for any such Derivative Works as a whole, 114 | provided Your use, reproduction, and distribution of the Work otherwise complies 115 | with the conditions stated in this License. 116 | 117 | 5. Submission of Contributions. 118 | 119 | Unless You explicitly state otherwise, any Contribution intentionally submitted 120 | for inclusion in the Work by You to the Licensor shall be under the terms and 121 | conditions of this License, without any additional terms or conditions. 122 | Notwithstanding the above, nothing herein shall supersede or modify the terms of 123 | any separate license agreement you may have executed with Licensor regarding 124 | such Contributions. 125 | 126 | 6. Trademarks. 127 | 128 | This License does not grant permission to use the trade names, trademarks, 129 | service marks, or product names of the Licensor, except as required for 130 | reasonable and customary use in describing the origin of the Work and 131 | reproducing the content of the NOTICE file. 132 | 133 | 7. Disclaimer of Warranty. 134 | 135 | Unless required by applicable law or agreed to in writing, Licensor provides the 136 | Work (and each Contributor provides its Contributions) on an "AS IS" BASIS, 137 | WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied, 138 | including, without limitation, any warranties or conditions of TITLE, 139 | NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A PARTICULAR PURPOSE. You are 140 | solely responsible for determining the appropriateness of using or 141 | redistributing the Work and assume any risks associated with Your exercise of 142 | permissions under this License. 143 | 144 | 8. Limitation of Liability. 145 | 146 | In no event and under no legal theory, whether in tort (including negligence), 147 | contract, or otherwise, unless required by applicable law (such as deliberate 148 | and grossly negligent acts) or agreed to in writing, shall any Contributor be 149 | liable to You for damages, including any direct, indirect, special, incidental, 150 | or consequential damages of any character arising as a result of this License or 151 | out of the use or inability to use the Work (including but not limited to 152 | damages for loss of goodwill, work stoppage, computer failure or malfunction, or 153 | any and all other commercial damages or losses), even if such Contributor has 154 | been advised of the possibility of such damages. 155 | 156 | 9. Accepting Warranty or Additional Liability. 157 | 158 | While redistributing the Work or Derivative Works thereof, You may choose to 159 | offer, and charge a fee for, acceptance of support, warranty, indemnity, or 160 | other liability obligations and/or rights consistent with this License. However, 161 | in accepting such obligations, You may act only on Your own behalf and on Your 162 | sole responsibility, not on behalf of any other Contributor, and only if You 163 | agree to indemnify, defend, and hold each Contributor harmless for any liability 164 | incurred by, or claims asserted against, such Contributor by reason of your 165 | accepting any such warranty or additional liability. 166 | 167 | END OF TERMS AND CONDITIONS 168 | 169 | APPENDIX: How to apply the Apache License to your work 170 | 171 | To apply the Apache License to your work, attach the following boilerplate 172 | notice, with the fields enclosed by brackets "[]" replaced with your own 173 | identifying information. (Don't include the brackets!) The text should be 174 | enclosed in the appropriate comment syntax for the file format. We also 175 | recommend that a file or class name and description of purpose be included on 176 | the same "printed page" as the copyright notice for easier identification within 177 | third-party archives. 178 | 179 | Copyright [yyyy] [name of copyright owner] 180 | 181 | Licensed under the Apache License, Version 2.0 (the "License"); 182 | you may not use this file except in compliance with the License. 183 | You may obtain a copy of the License at 184 | 185 | http://www.apache.org/licenses/LICENSE-2.0 186 | 187 | Unless required by applicable law or agreed to in writing, software 188 | distributed under the License is distributed on an "AS IS" BASIS, 189 | WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 190 | See the License for the specific language governing permissions and 191 | limitations under the License. 192 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- 1 | rtb-server-go 2 | ============= 3 | 4 | Implement golang RTB Server. 5 | 6 | # Run 7 | 8 | go run app.go -port 8080 9 | -------------------------------------------------------------------------------- /app.go: -------------------------------------------------------------------------------- 1 | package main 2 | 3 | import ( 4 | "log" 5 | "fmt" 6 | "net/http" 7 | "strconv" 8 | "flag" 9 | "time" 10 | "io/ioutil" 11 | "encoding/json" 12 | "runtime" 13 | ) 14 | 15 | type Dsp struct { 16 | DspId int 17 | ReqId string 18 | SleepMs int 19 | Status int 20 | Price int 21 | } 22 | 23 | // {"id":"gZtaC8svEsDMyXYUukU3GM62gRPZm4mAT7MNINjrnTB3l47gBFbRxmeI4djNejm1","status":1,"price":1000} 24 | type BidResponse struct { 25 | Id string 26 | Status int 27 | Price int 28 | } 29 | 30 | type Result struct { 31 | DspId int 32 | ReqId string 33 | Status int 34 | Price int 35 | } 36 | 37 | type Win struct { 38 | DspId int 39 | ReqId string 40 | Price int 41 | } 42 | 43 | type Response struct { 44 | Id string `json:"id"` 45 | DspId int `json:"dsp_id"` 46 | Price int `json:"price"` 47 | } 48 | 49 | func makeClient() http.Client { 50 | transport := http.Transport{ 51 | ResponseHeaderTimeout: time.Millisecond * 120, 52 | MaxIdleConnsPerHost: 200, 53 | } 54 | 55 | client := http.Client{ 56 | Transport: &transport, 57 | } 58 | 59 | return client 60 | } 61 | 62 | func doRequests(dsps []Dsp) <-chan Result { 63 | receiver := make(chan Result, len(dsps)) 64 | 65 | for _, dsp := range dsps { 66 | go doRequest(dsp, receiver) 67 | } 68 | 69 | return receiver 70 | } 71 | 72 | var client = makeClient() 73 | func doRequest(dsp Dsp, receiver chan Result) { 74 | url := fmt.Sprintf("http://%s/ad?id=%s&t=%d&s=%d&p=%d", *host, dsp.ReqId, dsp.SleepMs, dsp.Status, dsp.Price) 75 | //log.Println(url) 76 | resp, err := client.Get(url) 77 | 78 | result := Result{} 79 | result.DspId = dsp.DspId 80 | if err != nil { 81 | //log.Println("error") 82 | } else if resp != nil { 83 | defer resp.Body.Close() 84 | 85 | body, errRead := ioutil.ReadAll(resp.Body) 86 | if errRead != nil { 87 | log.Println("response read error") 88 | } else { 89 | //log.Println(string(body)) 90 | var bidResp BidResponse 91 | errJson := json.Unmarshal(body, &bidResp) 92 | if errJson != nil { 93 | log.Println("json parse error") 94 | } 95 | 96 | result.ReqId = bidResp.Id 97 | result.Status = bidResp.Status 98 | result.Price = bidResp.Price 99 | } 100 | } 101 | 102 | receiver <- result 103 | } 104 | 105 | func doAuction(results []Result) Win { 106 | win := Win{} 107 | win.DspId = -1 108 | 109 | var maxResult Result 110 | for _, result := range results { 111 | if result.Status != 1 { continue } 112 | 113 | if maxResult.Price == 0 { 114 | maxResult = result 115 | } else if result.Price > maxResult.Price { 116 | maxResult = result 117 | } 118 | } 119 | 120 | win.DspId = maxResult.DspId 121 | win.ReqId = maxResult.ReqId 122 | win.Price = maxResult.Price 123 | 124 | return win 125 | } 126 | 127 | func handler(w http.ResponseWriter, r *http.Request) { 128 | params := r.URL.Query() 129 | id := params["id"][0] 130 | dspNum, _ := strconv.Atoi( params["dsp"][0] ) 131 | dsps := make([]Dsp, dspNum) 132 | 133 | // parse dsp parameters 134 | for i := 0; i < dspNum; i++ { 135 | sleepMs, _ := strconv.Atoi( params[fmt.Sprintf("d%d_t", i)][0] ) 136 | status, _ := strconv.Atoi( params[fmt.Sprintf("d%d_s", i)][0] ) 137 | price := 0 138 | price_key := fmt.Sprintf("d%d_p", i) 139 | if params[price_key] != nil { 140 | price, _ = strconv.Atoi( params[price_key][0] ) 141 | } 142 | 143 | dsp := &Dsp{ i, id, sleepMs, status, price } 144 | dsps[i] = *dsp 145 | } 146 | 147 | // do request to dsps 148 | results := make([]Result, len(dsps)) 149 | receiver := doRequests(dsps) 150 | 151 | // receive result 152 | resultNum := 0 153 | for { 154 | result := <-receiver 155 | //log.Println(result) 156 | results[result.DspId] = result 157 | 158 | resultNum++ 159 | if len(results) == resultNum { break } 160 | } 161 | //log.Println(results) 162 | 163 | win := doAuction(results) 164 | //log.Println(win) 165 | 166 | w.Header().Set("Content-Type", "application/json") 167 | response := Response{ 168 | Id: win.ReqId, 169 | DspId: win.DspId, 170 | Price: win.Price, 171 | } 172 | bytes, _ := json.Marshal(response) 173 | jsonStr := string(bytes) 174 | // log.Println(jsonStr) 175 | fmt.Fprint(w, jsonStr) 176 | } 177 | 178 | var host *string 179 | func main() { 180 | runtime.GOMAXPROCS(runtime.NumCPU()) 181 | 182 | port := flag.Int("port", 5000, "PORT") 183 | host = flag.String("host", "dsp", "HOST") 184 | flag.Parse() 185 | 186 | http.HandleFunc("/ad", handler) 187 | log.Fatal( http.ListenAndServe(fmt.Sprintf(":%d", *port), nil) ) 188 | } 189 | --------------------------------------------------------------------------------